SQLAlchemy —— query

2023年8月24日 349次阅读 来源: SingleDiego

在 SQLAlchemy 中执行查询是通过 session 对象的 query 方法完成的。

遍历

for user in session.query(User).order_by(User.id):
    print(user.name, user.fullname)

# 执行结果：
ed Ed Jones
wendy Wendy Williams
mary Mary Contrary
fred Fred Flinstone

或者：

for name, fullname in session.query(User.name, User.fullname):
    print(name, fullname)

# 执行结果：
ed Ed Jones
wendy Wendy Williams
mary Mary Contrary
fred Fred Flinstone

再看一个例子：

for name in session.query(User.name):
    print(name)

# 执行结果：
('ed',)
('wendy',)
('mary',)
('fred',)

这时，返回结果为元组。

设置别名

给返回的结果起一个别名（label）：

for row in session.query(User.name.label('name_label')).all():
    print(row.name_label)

# 执行结果：
ed
wendy
mary
fred

给数据表对象设置别名（aliased）：

from sqlalchemy.orm import aliased

# 把 User 别名设置为：user_alias
user_alias = aliased(User, name='user_alias')

for row in session.query(user_alias):
    print(row)

# 执行结果：
ed
wendy
mary
fred

切片

for u in session.query(User).order_by(User.id)[1:3]:
    print(u)

# 执行结果：
wendy
mary

查询

查询，也就是 SQL 的 where 子句。在 SQLAlchemy 中，query 的查询可以通过 filter_by 或 filter 来实现。

1.使用 filter_by：

for name, in session.query(User.name).\
    filter_by(fullname='Ed Jones'):
    print(name)

# 执行结果：
ed

2.使用 filter：

for name, in session.query(User.name).\
    filter(User.fullname=='Ed Jones'):
    print(name)

# 执行结果：
ed

运算符

1. 等于：

query.filter(User.name == 'ed')

2. 不等于

query.filter(User.name != 'ed')

3. LIKE 运算符：

# 查询所有 "e" 开头的 name
for user in session.query(User).\
    filter(User.name.like("e%")):
    print(user)

# 执行结果：
ed

4. IN 运算符：

for user in session.query(User).\
    filter(User.name.in_(['ed', 'wendy', 'jack'])):
    print(user)

# 执行结果：
ed
wendy

5. NOT IN 运算符：

for user in session.query(User).\
    filter(~User.name.in_(['ed', 'wendy', 'jack'])):
    print(user)

# 执行结果：
mary
fred

6. IS NULL 运算符：

query.filter(User.name == None)

query.filter(User.name.is_(None))

7. IS NOT NULL 运算符：

query.filter(User.name != None)

query.filter(User.name.isnot(None))

8. AND 运算符：

方法1：

from sqlalchemy import and_

for user in session.query(User).\
    filter(and_(User.name == 'ed', User.fullname == 'Ed Jones')):
    print(user)

# 执行结果：
ed

方法2：

for user in session.query(User).\
    filter(User.name == 'ed', User.fullname == 'Ed Jones'):
    print(user)

方法3：

for user in session.query(User).\
    filter(User.name == 'ed').\
    filter(User.fullname == 'Ed Jones'):
    print(user)

9. OR 运算符：

from sqlalchemy import or_

for user in session.query(User).\
    filter(or_(User.name == 'ed', User.name == 'wendy')):
    print(user)

# 执行结果：
ed
wendy

query 包含的方法

1. all() ：
返回一个包含所有对象的 list

u = session.query(User).all()
print(u)

# 执行结果：
[<__main__.User object at 0x033DA410>, 
<__main__.User object at 0x033DA470>, 
<__main__.User object at 0x033DA4B0>, 
<__main__.User object at 0x033DA510>]

2. first() ：
返回结果的第一个对象

u = session.query(User).first()
print(u)

# 执行结果：
ed

3. one() ：
返回结果的实例对象

u = session.query(User).filter(User.id == 1).one()
print(u)
print(u.fullname)

# 执行结果：
ed
Ed Jones

如果有多个结果返回，使用 one() 将会报错：

u = session.query(User).one()

# 错误信息
Traceback (most recent call last):
...
MultipleResultsFound: Multiple rows were found for one()

如果查找的对象不存在，也会报错：

u = session.query(User).filter(User.id == 99).one()

# 错误信息
Traceback (most recent call last):
...
NoResultFound: No row was found for one()

我们可以用 one_or_none() 来处理对象未必存在的情况：

u = session.query(User).filter(User.id == 99).one_or_none()
print(u)

# 执行结果：
None

使用 scalar() 方法，返回结果的第一列的数值：

u = session.query(User.id).filter(User.name == 'wendy').scalar()
print(u)

# 执行结果：
2

文本化 SQL

使用 text() 方法可以用文本化的方式执行查询，使得语法更灵活，多数方法都支持这种文本化的方式，比如：filter() 和 order_by()。

from sqlalchemy import text

# 筛选 id<3 再根据 id 排序
for user in session.query(User).\
    filter(text("id<3")).\
    order_by(text("id")).all():
    print(user.name)

# 执行结果：
ed
wendy

需要用冒号传递参数时，可以用 params() 方法：

from sqlalchemy import text

# 筛选出 id<5 和 name='fred'
u = session.query(User).\
    filter(text("id<:value and name=:name")).\
    params(value=5, name='fred').order_by(User.id).one()

print(u)

# 执行结果：
fred

使用 from_statement() 方法，支持直接写原生 SQL 语句：

from sqlalchemy import text

u = session.query(User).\
    from_statement(text(" SELECT * FROM users where name='ed' ")).one()

print(u)

# 执行结果：
ed

计数

返回查询结果的对象的数目。

u = session.query(User).\
    filter(User.name.like('%ed')).count()

print(u)

# 执行结果：
2

需要单独计算某一类对象的数量，可以用 func() 和 group_by() 配合查询：

from sqlalchemy import func

u = session.query(User.name, func.count(User.name)).group_by(User.name).all()
print(u)

# 执行结果：
[('ed', 1), ('fred', 1), ('mary', 1), ('wendy', 1)]

返回查询得到的的数值，我们使用 scalar() 方法：

u = session.query(func.count(User.id)).scalar()
print(u)

# 执行结果：
4
    原文作者：SingleDiego
    原文地址: https://www.jianshu.com/p/b03e20cde341
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞