读取Excel文件
我们使用openpyxl从Excel文件读取数据
import openpyxl
workbook = openpyxl.load_workbook('./import.xlsx')
table = workbook[workbook.sheetnames[0]]
tableValues = table.values
for index, row in enumerate(tableValues):
if index == 0:
continue
number = row[0]
name = row[1]
parent_name = row[2]
sort = row[3]
print(row)
写入数据库
这里使用了mysqlclient
安装mysqlclient
pip mysqlclient连接数据库
import MySQLdb
# 建立连接
conn = MySQLdb.connect(
host=DATABASE_HOST, # 数据库主机地址
user=DATABASE_USER, # 数据库用户名
password=DATABASE_PASSWORD, # 数据库密码
db=DATABASE_NAME # 数据库名称
)查询数据库
cursor.execute('select id from asset_category where name=%s', (name,))
res = cursor.fetchone()插入数据库
cursor.execute("""
insert into asset_category(number, name, parent_id, sort, status, creator_id, update_datetime, create_datetime, modifier)
values(%s, %s, %s, %s, %s, %s, now(), now(), 1)
""", param)
