黄兵的个人博客

Python SQLAlchemy 用户支付查询相关总结

假设我们需要查询用户的支付记录，在 SQLAlchemy 模型中，我们查询方式如下：

from sqlalchemy.orm import joinedload
from models import User, Payment

# 查询已经支付的用户的支付记录
paid_payments = (
    Payment.query
    .options(joinedload(Payment.user))
    .filter(Payment.status == 'paid')
    .all()
)

# 输出每个用户的支付记录
for payment in paid_payments:
    print(f"User {payment.user.name} paid {payment.amount} on {payment.date}")

上述代码使用 SQLAlchemy ORM 的查询语法，使用 joinedload 来指定一次性加载相关的用户信息，从而避免了 N+1 查询的问题。然后使用 filter 方法来过滤出支付状态为 'paid' 的记录。最后，将结果保存在 paid_payments 变量中，并输出每个用户的支付记录。

转换成 SQL 语句如下：

SELECT *
FROM users u
JOIN payments p ON u.id = p.user_id
WHERE p.status = 'paid';

上述查询使用 JOIN 操作将 users 表和 payments 表连接起来，然后使用 WHERE 子句过滤出 payments 表中支付状态为 'paid' 的记录。最后，查询结果将只包含已经支付过的用户的支付记录。

如果我们需要统计用户的合计支付金额，我们可以使用如下的 SQL 来查询：

SELECT u.id, u.name, SUM(p.amount) AS total_amount
FROM users u
JOIN payments p ON u.id = p.user_id
WHERE p.status = 'paid'
GROUP BY u.id, u.name;

上述查询使用 JOIN 操作将 users 表和 payments 表连接起来，然后使用 WHERE 子句过滤出支付状态为 'paid' 的记录。接着使用 GROUP BY 子句按照用户的 ID 和姓名对结果进行分组，最后使用 SUM 函数计算每个用户的总支付金额。 在 SQLAlchemy ORM，可以使用以下代码来实现按用户合并计算支付金额的功能：

from sqlalchemy import func
from sqlalchemy.orm import joinedload
from models import User, Payment

# 按用户合并计算支付金额
paid_payments = (
    Payment.query
    .options(joinedload(Payment.user))
    .filter(Payment.status == 'paid')
    .group_by(User.id, User.name)
    .with_entities(
        User.id,
        User.name,
        func.sum(Payment.amount).label('total_amount')
    )
    .all()
)

# 输出每个用户的总支付金额
for payment in paid_payments:
    print(f"User {payment.name} paid {payment.total_amount}")

使用 filter 方法来过滤出支付状态为 'paid' 的记录。接着使用 group_by 子句按照用户的 ID 和姓名对结果进行分组，使用 with_entities 方法来选择需要的字段，最后使用 func.sum 函数计算每个用户的总支付金额，并将结果保存在 paid_payments 变量中。

最后，输出每个用户的总支付金额。

其它相关参考：

1、SQLAlchemy级联查询
2、SQLAlchemy 多条件查询

3、SQLAlchemy insert 和 update

4、SQLALCHEMY中的自引用

5、SQLAlchemy 如何更新数据库