读取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)
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.


写入数据库

这里使用了mysqlclient

安装mysqlclient

pip mysqlclient
  • 1.

连接数据库

import MySQLdb
# 建立连接
conn = MySQLdb.connect(
    host=DATABASE_HOST,  # 数据库主机地址
    user=DATABASE_USER,   # 数据库用户名
    password=DATABASE_PASSWORD,  # 数据库密码
    db=DATABASE_NAME  # 数据库名称
)
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.

查询数据库

cursor.execute('select id from asset_category where name=%s', (name,))
    res = cursor.fetchone()
  • 1.
  • 2.

插入数据库

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)
  • 1.
  • 2.
  • 3.
  • 4.