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

时间:2014-12-16 13:27:21

标签: python flask sqlalchemy flask-sqlalchemy

我正在使用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)?我不想继续将此块传递给整个应用程序中的所有查询。我想为查询创建类似默认范围的内容。

1 个答案:

答案 0 :(得分:4)

许多方法之一是添加类方法。使用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' s BaseQuery

的文档