写3个不同的函数txt_to_mysql、csv_to_mysql、excel_to_mysql分别完成txt文件、csv文件、xlsx文件向Mysql数据库的存储
函数txt_to_mysql:
import mysql.connector
def txt_to_mysql(file_path):
try:
conn = mysql.connector.connect(
host="localhost",
user="root",
password="password",
database="database_name"
)
cursor = conn.cursor()
with open(file_path, 'r') as file:
lines = file.readlines()
for line in lines:
data = line.strip().split(',')
query = "INSERT INTO table_name (col1, col2, col3) VALUES (%s, %s, %s)"
params = (data[0], data[1], data[2])
cursor.execute(query, params)
conn.commit()
print("Data inserted successfully")
except mysql.connector.Error as error:
print("Failed to insert data into MySQL table", error)
finally:
if conn.is_connected():
cursor.close()
conn.close()
print("MySQL connection closed")
# Example usage
txt_to_mysql("data.txt")
函数csv_to_mysql:
import csv
import mysql.connector
def csv_to_mysql(file_path):
try:
conn = mysql.connector.connect(
host="localhost",
user="root",
password="password",
database="database_name"
)
cursor = conn.cursor()
with open(file_path, 'r') as file:
csv_data = csv.reader(file)
next(csv_data) # Skip the header row
for row in csv_data:
query = "INSERT INTO table_name (col1, col2, col3) VALUES (%s, %s, %s)"
params = (row[0], row[1], row[2])
cursor.execute(query, params)
conn.commit()
print("Data inserted successfully")
except mysql.connector.Error as error:
print("Failed to insert data into MySQL table", error)
finally:
if conn.is_connected():
cursor.close()
conn.close()
print("MySQL connection closed")
# Example usage
csv_to_mysql("data.csv")
函数excel_to_mysql:
import openpyxl
import mysql.connector
def excel_to_mysql(file_path):
try:
conn = mysql.connector.connect(
host="localhost",
user="root",
password="password",
database="database_name"
)
cursor = conn.cursor()
workbook = openpyxl.load_workbook(file_path)
sheet = workbook.active
for row in sheet.iter_rows(min_row=2, values_only=True):
query = "INSERT INTO table_name (col1, col2, col3) VALUES (%s, %s, %s)"
params = (row[0], row[1], row[2])
cursor.execute(query, params)
conn.commit()
print("Data inserted successfully")
except mysql.connector.Error as error:
print("Failed to insert data into MySQL table", error)
finally:
if conn.is_connected():
cursor.close()
conn.close()
print("MySQL connection closed")
# Example usage
excel_to_mysql("data.xlsx")
``
原文地址: https://www.cveoy.top/t/topic/hLKZ 著作权归作者所有。请勿转载和采集!