Question

我正在使用sqlalchemy的挂架。我有几个模型，发现自己一次又一次地编写了这样的代码：

question = Session.query(Question).filter_by(id=question_id).one()
answer = Session.query(Answer).fileter_by(id=answer_id).one()
...
user = Session.query(User).filter_by(id=user_id).one()

由于模型都是扩展类Base，有没有办法定义一个通用的get_by_id()方法？

所以我可以用它作为：

quesiton = Question.get_by_id(question_id)
answer = Answer.get_by_id(answer_id)
...
user = User.get_by_id(user_id)

Answer 1

如果id是您的主键列，您只需：

session.query(Foo).get(id)

，如果该实例已在会话中，则具有不查询数据库的优点。

Answer 2

不幸的是，SQLAlchemy不允许您在没有相应表声明的情况下继承Base。您可以使用get_by_id作为classmethod定义mixin class，但是您需要为每个类指定它。

更简单，更脏的解决方案就是将其修补为Base：

def get_by_id(cls, id, session=session):
    return session.query(cls).filter_by(id=id).one()

Base.get_by_id = classmethod(get_by_id)

这假设您在定义时有一个session对象，否则您每次都需要将其作为参数传递。

Answer 3

class Base(object):
    @classmethod
    def get_by_id(cls, session, id):
        q = session.query(cls).filter_by(id=id)
        return q.one()

Question.get_by_id(Session, question_id)

如何在Sqlalchemy中为各种模型编写一个常见的get_by_id（）方法？

3 个答案: