python - 使用 Python + Flask-SqlAlchemy 的默认查询范围

Question

我正在使用 Python + Flask-SqlAlchemy 扩展编写一个应用程序。

我有以下代码:

class Resource(db.Model):
    __tablename__ = 'resources'
    id = db.Column(db.Integer, primary_key=True)
    created_at = db.Column(db.DateTime)
    updated_at = db.Column(db.DateTime)
    is_active = db.Column(db.Boolean)
    key = db.Column(db.String(64))
    title = db.Column(db.String(64))

    def __init__(self):
        self.created_at = datetime.now()
        self.updated_at = datetime.now()
        self.is_active = 1

@app.route('/resources', methods=['GET'])
def resources_get_all():
    get_limits()
    resources = []
    data = Resource.query.filter(Resource.is_active == 1).limit(LIMIT).offset(OFFSET).all()
    for row in data:
        resources.append(
            resources_show(row)
        )
    return envelope(resources)

有没有办法让所有对数据库的查询自动实现.filter(ModelName.is_active == 1).limit(LIMIT).offset(OFFSET)？我不想继续将此 block 传递给整个应用程序中的所有查询。我想创建类似于查询的默认范围之类的内容。

Answer 1

多种方法之一是添加类方法。顺便说一句，通过 python 类定义，您可以自由添加自己的方法。

class Resource(db.Model):
    # .... your existing code
    @staticmethod
    def customq(limit, offset):
        """This method returns an instance of flask_sqlalchemy.BaseQuery"""
        return Resource.query.filter_by(is_active=1).limit(limit).offset(offset)

然后你的示例代码变成:

@app.route('/resources', methods=['GET'])
def resources_get_all():
    get_limits()
    resources = []
    data = Resource.customq(LIMIT, OFFSET).all()

这是关于flask_sqlalchemy的BaseQuery的文档