flask_sqlalchemy: 重写查询方法例子(重写filter_by)

本文介绍如何在SQLAlchemy中重写filter_by方法以简化查询操作。通过定义自定义的Query类并在实例化SQLAlchemy对象时指定该类作为query_class参数,可以实现自动添加逻辑删除标志status=1的过滤条件。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

http://blog.wktadmin.com
1. 引出

比如某个项目中, status是逻辑删除的标志, 每次查询都需要加上status=1,

Gifts.query.filter_by(uid=current_user.id, isbn=isbn, launched=False, status=1)

非常的不方便, 重写filter_by是一个不错的选择

  1. 重写

filter_by等查询函数定义在sqlalchemy.orm.Query中, flask_sqlalchemy.BaseQuery继承了orm的Query, 因此我们重写的时候可是直接继承BaseQuery.

自定义filter_by的代码为:

class Query(BaseQuery):
    def filter_by(self, **kwargs):
        if 'status' not in kwargs.keys():
            kwargs['status'] = 1
        return super(Query, self).filter_by(**kwargs)
  1. 指定query类

但是这时候, 系统并不知道要使用我们重写的filter_by.
SQLAlchemy类的构造函数中为我们提供了可指定的query_class参数:

def __init__(self, app=None, use_native_unicode=True, session_options=None,
                 metadata=None, query_class=BaseQuery, model_class=Model):
    ...

因此可以在实例化SQLALchemy对象时候直接使用参数:


db = SQLALchemy(query_class=Query)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值