ORM和SQLAlchemy简介
对象关系映射(Object Relational Mapping,简称ORM),简单的来说,ORM是将数据库中的表与面向对象语言中的类建立了一种对应的关系。然后我们操作数据库的时候,数据库中的表或者表中的某一行,直接通过操作类或者类的实例就可以完成了。
SQLAlchemy是Python社区最知名的ORM工具之一,为高效和高性能的数据库访问设计,实现了完整的企业级持久模型。
SQLAlchemy是建立在数据库API之上,使用关系对象映射进行数据库操作,简单的来说,就是将对象转换成数据行,类转换成数据表,然后使用数据库API执行SQL并获取执行结果。
SQLAlchemy本身是无法是操作数据库的,必须借助第三方插件,比如mysqldb,pymysql。
SQLAlchemy的安装
python3.x自带pip3,可以直接安装:
C:\Windows\system32>pip3 install sqlalchemy
连接和创建
在MySQL服务器上,首先需要自己创建数据库:
mysql> create database db6;
Query OK, 1 row affected (0.00 sec)
连接数据库:
from sqlalchemy import create_engine
engine = create_engine('mysql+pymysql://root:LBLB1212@@@localhost:3306/db6')
print(engine)
#运行结果
Engine(mysql+pymysql://root:***@localhost:3306/db6)
如果出现以上的结果,说明已经连接成功了,这里使用的pymysql,不同的数据库或者第三方插件,create_engine的字符串是不同的:
mysql+mysqldb://:@[:]/
mysql+pymysql://:@/[?]
mysql+mysqlconnector://:@[:]/
oracle+cx_oracle://user:pass@host:port/dbname[?key=value&key=value...]
更多可见官方文档:http://docs.sqlalchemy.org/en/latest/dialects/index.html
现在连接已经成功了,那么现在需要将数据表的结构用ORM的语言描述出来,SQLAlchemy提供了一套Declarative系统来完成这个任务,比如这里我们需要创建一个userinfo表,看看它是怎么实现的:
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,CHAR,Integer
engine = create_engine('mysql+pymysql://root:LBLB1212@@@localhost:3306/db6?charset=utf8',max_overflow=5)
Base = declarative_base()
class userinfo(Base):
__tablename__ = 'userinfo'
id = Column(Integer,primary_key=True) #primary_key=True代表创建主键
username = Column(CHAR(20),nullable=False,index=True) #nullable=False代表不能为空,index=True代表创建索引
password = Column(CHAR(20),nullable=False)
email = Column(CHAR(64),nullable=False,index=True)
def init_db():
Base.metadata.create_all(engine)
if __name__ == '__main__':
init_db()
__tablename__指定表名,使用Column来创建列,可以在sqlalchemy中导入各种数据类型(CHAR,VARCHAR,Integer,String等),使用方法create_all创建文件中所有的表。在MySQL服务器上查看上面创建的表:
mysql> show create table userinfo \G;
*************************** 1. row ***************************
Table: userinfo
Create Table: CREATE TABLE `userinfo` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`username` char(20) NOT NULL,
`password` char(20) NOT NULL,
`email` char(64) NOT NULL,
PRIMARY KEY (`id`),
KEY `ix_userinfo_username` (`username`),
KEY `ix_userinfo_email` (`email`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8
1 row in set (0.00 sec)
本质上来说类相当于是表,类实例化后的对象相当于是数据行。
删除表使用:
Base.metadata.drop_all(engine)
数据行的增删改查操作
现在创建两个表,用户表(user)和用户类型(usertype)表:
from sqlalchemy import create_engine,ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,CHAR,Integer
engine = create_engine('mysql+pymysql://root:LBLB1212@@@localhost:3306/db6?charset=utf8',max_overflow=5)
Base = declarative_base()
class Usertype(Base):
__tablename__ = 'usertype'
type_id = Column(Integer, primary_key=True, autoincrement=True)
type = Column(CHAR(20),nullable=False)
class User(Base):
__tablename__ = 'user'
id = Column(Integer,primary_key=True,autoincrement=True)
username = Column(CHAR(20),nullable=False,index=True)
password = Column(CHAR(20),nullable=False)
user_type_id = Column(Integer,ForeignKey('usertype.type_id')) #ForeignKey创建外键
def init_db():
Base.metadata.create_all(engine)
def drop_db():
Base.metadata.drop_all(engine)
if __name__ == '__main__':
init_db()
执行之后,到数据库上查看是否已经创建:
1 mysql>desc user;2 +--------------+----------+------+-----+---------+----------------+
3 | Field | Type | Null | Key | Default | Extra |
4 +--------------+----------+------+-----+---------+----------------+
5 | id | int(11) | NO | PRI | NULL | auto_increment |
6 | username | char(20) | NO | MUL | NULL | |
7 | password | char(20) | NO | | NULL | |
8 | user_type_id | int(11) | YES | MUL | NULL | |
9 +--------------+----------+------+-----+---------+----------------+
10 4 rows in set (0.00sec)11
12 mysql>desc usertype;13 +---------+----------+------+-----+---------+----------------+
14 | Field | Type | Null | Key | Default | Extra |
15 +---------+----------+------+-----+---------+----------------+
16 | type_id | int(11) | NO | PRI | NULL | auto_increment |
17 | type | char(20) | NO | | NULL | |
18 +---------+----------+------+-----+---------+----------------+
19 2 rows in set (0.00 sec)
在对数据表操作之前,这里必须要建立session:
from sqlalchemy.orm import sessionmaker
Session = sessionmaker(bind=engine)
session = Session()
1.新增操作
现在要向表中插入数据,上面说过实例化类就可以得到一个对象而这个对象也就是一个数据行:
#插入单个数据行
obj1=Usertype(type="普通用户")
session.add(obj1)
session.commit() #必须记得要提交
也可以使用下面的方法插入多个数据行:
objs = [
Usertype(type="超级用户"),
Usertype(type="白金用户"),
Usertype(type="黄金用户"),
]
session.add_all(objs)
session.commit()
再来在数据库上查看用户类型表:
mysql> select * from usertype;
+---------+--------------+
| type_id | type |
+---------+--------------+
| 1 | 普通用户 |
| 2 | 超级用户 |
| 3 | 白金用户 |
| 4 | 黄金用户 |
+---------+--------------+
4 rows in set (0.02 sec)
最后在用户表中插入用户:
objs = [
User(username='frank',password='111123',user_type_id=2),
User(username='rose',password='312231',user_type_id=1),
User(username='jack',password='112311',user_type_id=1),
User(username='tom',password='123231',user_type_id=3),
User(username='jeff',password='121213',user_type_id=3),
User(username='alex',password='123323',user_type_id=4),
User(username='christ',password='122123',user_type_id=1),
User(username='duke',password='123111',user_type_id=4),
]
session.add_all(objs)
session.commit()
2.查询操作
使用query()方法进行查询,先来看一下下面的例子:
res = session.query(User)
print(res)
#运行结果
SELECT user.id AS user_id, user.username AS user_username, user.password AS user_password, user.user_type_id AS user_user_type_id
FROM user
根据以上例子会发现返回值为SQL语句,再来看一下下面的例子:
res = session.query(User).all()
print(res)
#运行结果
[<__main__.user object at>, <__main__.user object at>, <__main__.user object at>, <__main__.user object at>, <__main__.user object at>, <__main__.user object at>, <__main__.user object at>, <__main__.user object at>]
返回的User的对象列表,这也就是每一个数据行,一个数据行对应一个User对象,可以使用fro循环遍历:
for row in res:
print(row.id,row.username,row.password,row.user_type_id)
#运行结果
1 frank 111123 2
2 rose 312231 1
3 jack 112311 1
4 tom 123231 3
5 jeff 121213 3
6 alex 123323 4
7 christ 122123 1
8 duke 123111 4
下面来看一下SQL中各种查询在SQLAlchemy是怎么实现的:
a.普通的条件查询
1 1.SQL中的多个条件查询2 res = session.query(User).filter(User.id > 5,User.username == 'duke').all() #使用filter,里面为条件表达式,默认多个表达式之间为与关系3 print(res[0].id,res[0].username) #res ==> [<__main__.user object at>]4 #运行结果5 duke6
7 2.SQL中的BETWEEN语句8 res = session.query(User).filter(User.id.between(3,5)).all()9 for row inres:10 print(row.id,row.username,row.password,row.user_type_id)11 #运行结果12 jack 112311 1
13 tom 123231 3
14 jeff 121213 3
15
16 3.SQL中的IN17 res = session.query(User).filter(User.id.in_([1,5])).all()18 for row inres:19 print(row.id,row.username,row.password,row.user_type_id)20 #运行结果21 frank 111123 2
22 jeff 121213 3
23 #取反没在User前面加一个波浪号24 res = session.query(User).filter(~User.id.in_([1,2,3,4,5])).all()25 for row inres:26 print(row.id,row.username,row.password,row.user_type_id)27
28 4.子查询的3中类型29 4.1类似SQL中select * from table_name where id in (select id from tablename where name='frank')30 res = session.query(User).filter(User.id.in_(session.query(User.id).filter(User.username == 'frank'))).all()31 print(res[0].username)32 #运行结果33 frank34 4.2类似SQL中select * from (select * from tablename where id >4) asB,相当于在临时表中查询35 q1=session.query(User).filter(User.id>4).subquery()36 res =session.query(q1).all()37 print(res)38 #运行结果39 [(5, 'jeff', '121213', 3), (6, 'alex', '123323', 4), (7, 'christ', '122123', 1), (8, 'duke', '123111', 4)]40 4.3类似于SQL中的select id,(select type from usertype where usertype.type_id=user.user_type_id) from user where id > 5,子查询在select中41 res = session.query(User.id,session.query(Usertype.type).filter(User.user_type_id==Usertype.type_id).as_scalar()).filter(User.id>5).all()42 print(res)43 #运行结果44 [(6, '黄金用户'), (7, '普通用户'), (8, '黄金用户')]45
46 5.默认情况下,多个条件之间是与关系,那么如何改成或呢?47 fromsqlalchemy import or_,and_48 res = session.query(User.id,User.username).filter (or_(User.id == 3, User.username =='frank')).all()49 print(res)50 #运行结果51 [(1, 'frank'), (3, 'jack')]52
53 最后这里的filter也可以使用filter_by,但是里面表达式却不是判断的表达式:54 res = session.query(User.id,User.username).filter_by(username='frank').all()55 print(res)56 #运行结果57 [(1, 'frank')]58
59 条件查询
b.通配符
1 SQL中的LIKE操作2 1.%:代表任何数目任意字符,比如这里要查找名字中含有'e'字符的:3 res = session.query(User.id,User.username).filter(User.username.like('%e%')).all()4 print(res)5 #运行结果6 [(6, 'alex'), (8, 'duke'), (5, 'jeff'), (2, 'rose')]7 2._:代表任意单个字符,比如这里要查找第二个字符为'r'的用户信息:8 res = session.query(User.id,User.username).filter(User.username.like('_r%')).all()9 print(res)10 #运行结果11 [(1, 'frank')]
c.限制行数
1 SQL中的limit操作2 res=session.query(User)[1:3]3 for row inres:4 print(row.id,row.username,row.password,row.user_type_id)5 #运行结果6 rose 312231 1
7 jack 112311 1
d.排序
1 SQL中的排序操作2 res = session.query(User.id,User.username).filter(User.id>4).order_by(User.id.desc()).all()3 print(res)4 res = session.query(User.id,User.username,User.user_type_id).filter(User.id>4).order_by(User.id.desc(),User.user_type_id.asc()).all()5 print(res)6 #运行结果7 [(8, 'duke'), (7, 'christ'), (6, 'alex'), (5, 'jeff')]8 [(8, 'duke', 4), (7, 'christ', 1), (6, 'alex', 4), (5, 'jeff', 3)]
e.分组
1 在SQL中,可以使用很多的内置函数,在SQLAlchemy中可以使用以下方式使用内置函数2 fromsqlalchemy.sql import func3 现在来看一下SQL中分组是怎么实现的:4 res =session.query(User.user_type_id,func.sum(User.id)).group_by(User.user_type_id).all()5 print(res)6 #运行结果7 [(1, Decimal('12')), (2, Decimal('1')), (3, Decimal('9')), (4, Decimal('14'))]8 使用having进行筛选9 res = session.query(User.user_type_id,func.sum(User.id)).group_by(User.user_type_id).having(func.sum(User.id)>10).all()10 print(res)11 #运行结果12 [(1, Decimal('12')), (4, Decimal('14'))]
f.连表
1 1.普通的连表2 res = session.query(User.username,Usertype.type).filter(User.user_type_id ==Usertype.type_id)3 print(res)4 #运行结果5 SELECT user.username AS user_username, usertype.type AS usertype_type6 FROM user, usertype7 WHERE user.user_type_id =usertype.type_id8 2.SQL的INNER方式连表9 res =session.query(User).join(Usertype)10 print(res)11 #运行结果12 SELECT user.id AS user_id, user.username AS user_username, user.password AS user_password, user.user_type_id AS user_user_type_id13 FROM user INNER JOIN usertype ON usertype.type_id =user.user_type_id14 3.SQL的左连接(右连接直接将User和Usertype换个位置即可)15 res = session.query(User).join(Usertype,isouter=True)16 print(res)17 #运行结果18 SELECT user.id AS user_id, user.username AS user_username, user.password AS user_password, user.user_type_id AS user_user_type_id19 FROM user LEFT OUTER JOIN usertype ON usertype.type_id = user.user_type_id
g.组合
1 SQL中的UNION2 1.去重3 q1 = session.query(User.id,User.user_type_id).filter(User.id>4)4 q2 = session.query(Usertype.type_id,Usertype.type).filter(Usertype.type_id<3)5 res =q1.union(q2).all()6 print(res)7 #运行结果8 [(5, '3'), (6, '4'), (7, '1'), (8, '4'), (1, '普通用户'), (2, '超级用户')]9 2.不去重10 q1 = session.query(User.id,User.user_type_id).filter(User.id>4)11 q2 = session.query(Usertype.type_id,Usertype.type).filter(Usertype.type_id<3)12 res =q1.union_all(q2).all()13 print(res)14 #运行结果15 [(5, '3'), (6, '4'), (7, '1'), (8, '4'), (1, '普通用户'), (2, '超级用户')]16
17 组合
3.删除操作
删除动作是基于查询操作的,例如删除用户ID为8的用户信息
res=session.query(User).filter(User.id == 8).delete()
print(res) #返回1,代表有数据可删除
session.commit() #一定记得要commit,否则删除不会成功
在数据库上查看,已经被删除:
mysql> select * from user;
+----+----------+----------+--------------+
| id | username | password | user_type_id |
+----+----------+----------+--------------+
| 1 | frank | 111123 | 2 |
| 2 | rose | 312231 | 1 |
| 3 | jack | 112311 | 1 |
| 4 | tom | 123231 | 3 |
| 5 | jeff | 121213 | 3 |
| 6 | alex | 123323 | 4 |
| 7 | christ | 122123 | 1 |
+----+----------+----------+--------------+
7 rows in set (0.00 sec)
4.修改操作
1.直接修改条件筛选后的内容,如修改用户id为7的用户的用户类型为2
res = session.query(User).filter(User.id == 7).update({"user_type_id":2})
print(res) #返回1代表执行成功
session.commit() #记得commit
2.在字符串后拼接字符串
session.query(User).filter(User.id > 2).update({User.username: User.username + "GG"}, synchronize_session=False)
session.commit()
3.累加数值
session.query(User).filter(User.id > 1).update({"user_type_id": User.user_type_id + 1}, synchronize_session="evaluate")
synchronize_session=False,synchronize_session="evaluate"为固定用法
relationship建立关系
还是引用上面的两个类来阐述relationship的使用。
engine = create_engine('mysql+pymysql://root:LBLB1212@@@localhost:3306/db6?charset=utf8',max_overflow=5)
Base = declarative_base()
class Usertype(Base):
__tablename__ = 'usertype'
type_id = Column(Integer, primary_key=True, autoincrement=True)
type = Column(CHAR(20),nullable=False)
class User(Base):
__tablename__ = 'user'
id = Column(Integer,primary_key=True,autoincrement=True)
username = Column(CHAR(20),nullable=False,index=True)
password = Column(CHAR(20),nullable=False)
user_type_id = Column(Integer,ForeignKey('usertype.type_id'))
如果想取得用户的用户名和用户类型,按照之前学过的,可以使用join操作来完成:
res = session.query(User.username,Usertype.type).join(Usertype,isouter=True).all()
print(res)
#运行结果
[('rose', '普通用户'), ('jack', '普通用户'), ('frank', '超级用户'), ('christ', '超级用户'), ('tom', '白金用户'), ('jeff', '白金用户'), ('alex', '黄金用户')]
SQLAlchemy为用户提供了relationship方法,将两个表建立联系,使用方法如下:
from sqlalchemy.orm import sessionmaker,relationships #需要导入relationship
class Usertype(Base):
__tablename__ = 'usertype'
type_id = Column(Integer, primary_key=True, autoincrement=True)
type = Column(CHAR(20),nullable=False)
class User(Base):
__tablename__ = 'user'
id = Column(Integer,primary_key=True,autoincrement=True)
username = Column(CHAR(20),nullable=False,index=True)
password = Column(CHAR(20),nullable=False)
user_type_id = Column(Integer,ForeignKey('usertype.type_id'))
user_type = relationship("Usertype") #不会生成列,括号内为要建立关系的类名
现在User类下多了一个属性,来打印看一下是什么?
res = session.query(User).all()
for row in res:
print(row.username,row.user_type)
#运行结果
frank <__main__.usertype object at>
rose <__main__.usertype object at>
jack <__main__.usertype object at>
tom <__main__.usertype object at>
jeff <__main__.usertype object at>
alex <__main__.usertype object at>
christ <__main__.usertype object at>
duke <__main__.usertype object at>
由上面的打印的结果可知,user_type为Usertype的对象,而Usertype的对象即表usertype的行,所以这里建立了relationship之后,可以直接在User对象中通过user_type属性获得指定用户的用户类型了,实现如下:
res = session.query(User).all()
for row in res:
print(row.username,row.user_type.type)
#运行结果
frank 超级用户
rose 普通用户
jack 普通用户
tom 白金用户
jeff 白金用户
alex 黄金用户
christ 普通用户
duke 黄金用户
在上面的例子中,可以通过User对象获取其用户类型,但是不能通过Usertype对象获取用户的信息。下面举另外一个例子,获取不同类型用户下面的用户有哪些?
首先根据之前学的知识,可以使用子查询来实现:
res = session.query(Usertype)
for row in res:
print(row.type_id,row.type,session.query(User.username).filter(User.user_type_id == row.type_id).all())
#运行结果
1 普通用户 [('rose',), ('jack',), ('christ',)]
2 超级用户 [('frank',)]
3 白金用户 [('tom',), ('jeff',)]
4 黄金用户 [('alex',), ('duke',)]
使用子查询固然可以实现,但是实现起来颇为麻烦,所以这里引入了backref属性:
class Usertype(Base):
__tablename__ = 'usertype'
type_id = Column(Integer, primary_key=True, autoincrement=True)
type = Column(CHAR(20),nullable=False)
class User(Base):
__tablename__ = 'user'
id = Column(Integer,primary_key=True,autoincrement=True)
username = Column(CHAR(20),nullable=False,index=True)
password = Column(CHAR(20),nullable=False)
user_type_id = Column(Integer,ForeignKey('usertype.type_id'))
user_type = relationship("Usertype",backref='use') #定义backref给Usertype类使用的
下面来看看Usertype如何使用backref:
res = session.query(Usertype).all()
for row in res:
print(row.type_id,row.type,row.use)
#运行结果
1 普通用户 [<__main__.user object at>, <__main__.user object at>, <__main__.user object at>]
2 超级用户 [<__main__.user object at>]
3 白金用户 [<__main__.user object at>, <__main__.user object at>]
4 黄金用户 [<__main__.user object at>, <__main__.user object at>]
所以在User中定义了backref就相当于在Usertype中定义了use属性,那么上面的需求就很好实现了:
res = session.query(Usertype)
for row in res:
print(row.type_id,row.type,[user.username for user in row.use])
#运行结果
1 普通用户 ['rose', 'jack', 'christ']
2 超级用户 ['frank']
3 白金用户 ['tom', 'jeff']
4 黄金用户 ['alex', 'duke']