智能解密TXT数据存储MySQL数据库

2023-11-05 17:45:43

从TXT文件到MySQL数据库：简易51行代码实现数据存储

在数据处理和分析领域，文本文件（TXT）是一种常用的数据存储格式。然而，当我们需要将TXT数据导入关系型数据库（如MySQL）时，手工操作往往耗时费力。本文将介绍一种利用Python语言编写的简洁工具，仅需51行代码即可实现TXT数据向MySQL数据库的自动导入，从而大幅提升数据处理效率。

第一步：准备工作

在开始之前，我们需要导入必要的库文件，获取用户输入，并连接到MySQL数据库。

import os
import csv
import MySQLdb

file_path = input("请输入TXT文件路径：")
db_host = input("请输入数据库主机名：")
db_user = input("请输入数据库用户名：")
db_password = input("请输入数据库密码：")
db_name = input("请输入数据库名称：")
db_table = input("请输入数据表名称：")

conn = MySQLdb.connect(host=db_host, user=db_user, passwd=db_password, db=db_name, charset='utf8')

第二步：创建数据表

接下来，我们需要在MySQL数据库中创建数据表来存储TXT数据。

cursor = conn.cursor()
try:
    cursor.execute("CREATE TABLE IF NOT EXISTS `%s` (id INT NOT NULL AUTO_INCREMENT, PRIMARY KEY (`id`), `column_name` VARCHAR(255))" % db_table)
except:
    print("数据表已存在！")

第三步：读取TXT文件数据

现在，我们可以读取TXT文件中的数据。

with open(file_path, 'r') as f:
    reader = csv.reader(f)
    data = [row for row in reader]

第四步：插入数据到数据库

最后，我们将TXT文件中的数据插入到MySQL数据库中。

for row in data:
    try:
        cursor.execute("INSERT INTO `%s` (`column_name`) VALUES ('%s')" % (db_table, "', '".join(row)))
    except:
        print("数据插入失败！")

第五步：关闭数据库连接

完成数据导入后，我们需要关闭数据库连接。