这里将不介绍openpyxl模块的详细操作。
主要就是记录一个使用openpyxl模块将Excel表格的数据导入数据库中的实例。
from openpyxl import load_workbookimport os,django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "项目名称.settings")
django.setup()from backend.models import User # 封装成一个函数，调用是传入表格路径即可
def xlsxHandler(localpath):
wb = load_workbook(localpath) # 打开已存在的execl文件，格式xlsx ws = wb[wb.sheetnames[0]] # 选择第一张sheet表
rows = ws.max_row # 获取表的最大行数
columns = ws.max_column # 获取表的最大列数 column_heading = [ws.cell(row=1,column=x).value for x in range(1,columns+1)] # 读取excel第一行的值，写入list column_name = ['用户名','公司名称','电话'] # 数据库必需字段 # print(column_heading) # 文件第一行title if len([name for name in column_name if name not in column_heading]) == 0: # 返回字段组成的list为空，则说明文件列标题包含MySQL需要的字段
print(' - 检查完成，执行写入')
# 判断Excel中各字段所在列号
username = column_heading.index(column_name[0]) # 用户名 - 位置
company = column_heading.index(column_name[1]) # 公司 - 位置
phone = column_heading.index(column_name[2]) # 电话 - 位置 if ws.cell(row=2,column=1).value == None:
table_start_line = 3
else:
table_start_line = 2
data = []
userList = []
for row in range(table_start_line, rows + 1):
for column in range(1, columns + 1): # 因为从第1列开始，所以此处从1开始
data.append(str(ws.cell(row=row, column=column).value)) # 以字符串形式保存数据到MySQL
print(data)
print(data[username],data[company],data[phone])
userList.append(User(username=data[username],company=data[company],phone=data[phone]))
data = []
print('userList ',userList ) try:
User.objects.bulk_create(userList ) # 使用bulk_create批量导入
msg = '导入成功'
except Exception as e:
print('异常',e)
msg = '导入失败'
else:
print('文件列标题不完全包含数据库需要的字段，请检查文件。')
msg = '文件列标题不完全包含数据库需要的字段，请检查文件。'
wb.close() # 关闭excel return msg
