首页 文章

SQLAlchemy:返回现有对象,而不是在调用构造函数时创建新的on

提问于
浏览
6

我想以这样的方式使用sqlalchemy:

email1 = EmailModel(email="user@domain.com", account=AccountModel(name="username"))
email2 = EmailModel(email="otheruser@domain.com", account=AccountModel(name="username"))

通常sqlalchemy将为该帐户创建两个条目,并将每个电子邮件地址链接到此 . 如果我将帐户名设置为唯一的sqlalchemy,则抛出一个异常,告诉我有一个具有相同值的条目已经存在 . 这一切都有意义,并按预期工作 .

现在我想出了一个我自己的方法,允许提到的代码,只需通过覆盖AccountModel类的 new 构造函数创建一个帐户:

def __new__(*cls, **kw):
    if len(kw) and "name" in kw:
        x = session.query(cls.__class__).filter(cls[0].name==kw["name"]).first()
        if x: return x
    return object.__new__(*cls, **kw)

这对我来说非常合适 . 但问题是:

  • 这是正确的方法吗?

  • 有没有一种方法可以实现同样的目标?

我使用的是最新的0.8.x SQLAlchemy版本和Python 2.7.x.

谢谢你的帮助 :)

2 回答

  • 1

    我认为这不是实现这一目标的最佳方法,因为它会创建构造函数与全局变量 session 的依赖关系,并且还会以意外的方式修改构造函数的行为( new 预计会返回一个新对象) . 例如,如果有人在两个会话中并行使用您的类,则代码将失败,他将不得不深入研究代码以查找错误 .

    我不知道任何"sqlalchemy"实现这个的方法,但我建议创建一个函数 createOrGetAccountModel like

    def createOrGetAccountModel(**kw):
        if len(kw) and "name" in kw:
          x = session.query(AccountModel).filter_by(name=kw["name"]).first()
          if x: return x
        return AccountModel(**kw)
    
  • 6

    http://www.sqlalchemy.org/trac/wiki/UsageRecipes/UniqueObject的维基上有这个例子 .

    虽然最近我更倾向于使用@classmethod而不是重新定义构造函数,因为显式优于隐式,也更简单:

    user.email = Email.as_unique('foo@bar.com')
    

    (我实际上现在要更新维基以更全面地表示这里的使用选项)

相关问题