数据库是一个网站的基础,在Flask中可以自由的使用MySQL、PostgreSQL、SQLite、Redis、MongoDB来写原生的语句实现功能,也可以使用更高级别的数据库抽象方式,如SQLAlchemy或MongoEngine这样的OR(D)M。本教程以MySQL+SQLAlchemy的组合来进行讲解。
在讲解Flask中的数据库操作之前,先确保你已经安装了以下软件:
mysql:如果是在windows上,到官网下载。如果是ubuntu,通过命令sudo apt-get install mysql-server libmysqlclient-dev -yq进行下载安装。
MySQLdb:MySQLdb是用Python来操作mysql的包,因此通过pip来安装,命令如下:pip install mysql-python。
SQLAlchemy:SQLAlchemy是一个数据库的 ORM 框架,我们在后面会用到。安装命令为:pip install SQLAlchemy。
通过SQLAlchemy连接数据库:
首先来看一段代码:
from sqlalchemy import create_engine
# 数据库的配置变量
HOSTNAME = '127.0.0.1'
PORT = '3306'
DATABASE = 'xt_flask'
USERNAME = 'root'
PASSWORD = 'root'
DB_URI = 'mysql+mysqldb://{}:{}@{}:{}/{}'.format(USERNAME,PASSWORD,HOSTNAME,PORT,DATABASE)
# 创建数据库引擎
engine = create_engine(DB_URI)
#创建连接
with engine.connect() as con:
rs = con.execute('SELECT 1')
print rs.fetchone()
首先从sqlalchemy中导入create_engine,用这个函数来创建引擎,然后用engine.connect()来连接数据库。其中一个比较重要的一点是,通过create_engine函数的时候,需要传递一个满足某种格式的字符串,对这个字符串的格式来进行解释:
dialect+driver://username:[email protected]:port/database
dialect是数据库的实现,比如MySQL、PostgreSQL、SQLite,并且转换成小写。driver是Python对应的驱动,如果不指定,会选择默认的驱动,比如 MySQL 的默认驱动是MySQLdb。username是连接数据库的用户名,password是连接数据库的密码,host是连接数据库的域名,port是数据库监听的端口号,database是连接哪个数据库的名字。
如果以上输出了1,说明SQLAlchemy能成功连接到数据库。
用 SQLAlchemy 执行原生 SQL:
我们将上一个例子中的数据库配置选项单独放在一个
from sqlalchemy import create_engine
from constants import DB_URI
#连接数据库
engine = create_engine(DB_URI,echo=True)
# 使用 with 语句连接数据库,如果发生异常会被捕获
with engine.connect() as con:
# 先删除 users 表
con.execute('drop table if exists users')
# 创建一个 users 表,有自增长的 id 和 name
con.execute('create table users(id int primary key auto_increment,'
'name varchar(25))')
# 插入两条数据到表中
con.execute('insert into users(name) values("xiaoming")')
con.execute('insert into users(name) values("xiaotuo")')
# 执行查询操作
rs = con.execute('select * from users')
# 从查找的结果中遍历
for row in rs:
print row
使用 ORM:
随着项目越来越大,采用写原生 SQL 的方式在代码中会出现大量的 SQL 语句,那么问题就出现了:
SQL 语句重复利用率不高,越复杂的 SQL 语句条件越多,代码越长。会出现很多相近的 SQL 语句。
很多 SQL 语句是在业务逻辑中拼出来的,如果有数据库需要更改,就要去修改这些逻辑,这会很容易漏掉对某些 SQL 语句的修改。
写 SQL 时容易忽略 web 安全问题,给未来造成隐患。
ORM,全称Object Relational Mapping,中文叫做对象关系映射,通过ORM我们可以通过类的方式去操作数据库,而不用再写原生的 SQL 语句。通过把表映射成类,把行作为示例,把字段作为属性,ORM在执行对象操作的时候最终还是会把对应的操作转换为数据库原生语句。使用ORM有许多优点:
易用性:使用ORM做数据库的开发可以有效的减少重复 SQL 语句的概率,写出来的模型也更加直观、清晰。
性能损耗小:ORM转换成底层数据库操作指令确实会有一些开销。但从实际的情况来看,这种性能损耗很少(不足 5%),只要不是对性能有严苛的要求,综合考虑开发效率、代码的阅读性,带来的好处要远远大于性能损耗,而且项目越大作用越明显。
设计灵活:可以轻松的写出复杂的查询。
可移植性:SQLAlchemy封装了底层的数据库实现,支持多个关系数据库引擎,包括流行的MySQL、PostgreSQL和SQLite。可以非常轻松的切换数据库。
使用 SQLAlchemy:要使用ORM来操作数据库,首先需要创建一个类来与对应的表进行映射。现在以User 表来做为例子,它有自增长的 id、name、fullname、password这些字段,那么对应的类为:
from sqlalchemy import Column,Integer,String
from constants import DB_URI
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
engine = create_engine(DB_URI,echo=True)
# 所有的类都要继承自`declarative_base`这个函数生成的基类
Base = declarative_base(engine)
class User(Base):
# 定义表名为 users
__tablename__ = 'users'
# 将 id 设置为主键,并且默认是自增长的
id = Column(Integer,primary_key=True)
# name 字段,字符类型,最大的长度是 50 个字符
name = Column(String(50))
fullname = Column(String(50))
password = Column(String(100))
# 让打印出来的数据更好看,可选的
def __repr__(self):
return "" % (self.id,self.name,self.fullname,self.password)
SQLAlchemy会自动的设置第一个Integer的主键并且没有被标记为外键的字段添加自增长的属性。因此以上例子中id自动的变成自增长的。以上创建完和表映射的类后,还没有真正的映射到数据库当中,执行以下代码将类映射到数据库中:
Base.metadata.create_all()
在创建完数据表,并且做完和数据库的映射后,接下来让我们添加数据进去:
ed_user = User(name='ed',fullname='Ed