概述
在Flask开发中需要批量导入数据,数据是用Excel保存的。所以在网上搜索了下资料,提取Excel里的数据可以使用xlrd包。这篇文章主要是记录Flask从Excel批量导入数据到数据库的方法。
开发环境
Python版本:3.5.1
操作系统:win10 x64
xlrd
首先是安装命令:
pip install xlrd
代码清单
然后下面放上代码清单,以后做项目的时候可以直接移植。
目录结构
- app
- data-dev.sqlite
- demo.py
- model.py
- people.xlsx
- init.py
- venv
app是放置flask应用的地方,venv是Python的虚拟环境。
__init__.py
import os
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
basedir = os.path.abspath(os.path.dirname(__file__))
app = Flask(__name__)
app.config['SQLALCHEMY_TRACK_MODIFICATIONS']=True
app.config['SQLALCHEMY_DATABASE_URI'] = \
'sqlite:///' + os.path.join(basedir, 'data-dev.sqlite')
db = SQLAlchemy(app)
model.py
存放数据表的地方
from app import db
class People(db.Model):
__tablename__ = 'peoples'
def __repr__(self):
return "<%r %r %r>" % (self.id, self.name, self.age)
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String)
age= db.Column(db.Integer)
demo.py
示例程序,这里只是作为一个演示,根据项目需要可以把导入数据的代码放在flask的manage命令里。
import os
from app import db
from xlrd import open_workbook
from app.model import People
excelName = "people.xlsx"
bk = open_workbook(excelName, encoding_override="utf-8")
# for s in bk.sheets():
# print "sheet:",s.name
# 创建数据表
db.create_all()
try:
sh = bk.sheets()[0] # 因为Excel里只有sheet1有数据,如果都有可以使用注释掉的语句
# sh = bk.sheet_by_name("benke.xlsx")
except:
print("no sheet in %s named sheet1" % excelName)
else:
nrows = sh.nrows
ncols = sh.ncols
print(os.getcwd(), "行数 %d,列数 %d" % (nrows, ncols))
row_list = []
for i in range(1, nrows):
row_date = sh.row_values(i)
print(row_date)
n = i - 1
people = People()
people.id = row_date[0]
people.name = row_date[1]
people.age = row_date[2]
db.session.add(people)
db.session.commit()
db.session.close()
peoples = People.query.all()
print(peoples)
如果之前没有创建数据表,则首先要db.create_all()
语句创建数据表。
打印测试
F:\Python_Demo\importDataToDBFromExcel\app 行数 5,列数 3
[1.0, ‘小明’, 12.0]
[2.0, ‘小红’, 13.0]
[3.0, ‘小刚’, 15.0]
[4.0, ‘小乖’, 16.0]
[<1 ‘小明’ 12>, <2 ‘小红’ 13>, <3 ‘小刚’ 15>, <4 ‘小乖’ 16>]
前面四行是print(row_date)
的打印,说明提取数据成功;
最后一行是print(peoples)
的打印,说已经成功入库。