python - 根据 SQLAlchemy 中提交的 id 自动修改相关对象值

Question

首先，示例代码:

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy import event


Base = declarative_base()


class Foo(Base):

    __tablename__ = 'foo'

    id = Column(Integer, primary_key=True)
    label = Column(String)


class Bar(Base):

    __tablename__ = 'bar'

    id = Column(Integer, primary_key=True)
    foo_id = Column(Integer, ForeignKey('foo.id'))
    foo = relationship(Foo)

    def __init__(self, foo=None):

        self.foo = foo if foo else Foo()
        print 'init', self.id, self, self.foo.id, self.foo


def _adjust_label(target, value, old_value, initiator):

    print 'adjust', target, value, old_value, initiator
    if value and not target.foo.label:
        target.foo.label = 'autostring %d' % value
        print 'adjust', target.id, target, target.foo.id, target.foo, target.foo.label

event.listen(Bar.id, 'set', _adjust_label)

engine = create_engine('sqlite:////path/to/some.db')
Base.metadata.create_all(engine)
session = sessionmaker(bind=engine)()
bar = Bar()
print 'pre-add', bar.id, bar, bar.foo.id, bar.foo
session.add(bar)
print 'added', bar.id, bar, bar.foo.id, bar.foo
session.commit()
print 'commited', bar.id, bar, bar.foo.id, bar.foo, bar.foo.label

我得到的是:

pre-add None <__main__.Bar object at 0x2929f50>
pre-add None <__main__.Foo object at 0x292e310>
added None <__main__.Bar object at 0x2929f50>
added None <__main__.Foo object at 0x292e310>
adjust <__main__.Bar object at 0x2929f50> 14 None Bar.id
adjust None <__main__.Bar object at 0x2929f50>
adjust 14 <__main__.Foo object at 0x292e310> autostring 14
commited 14 <__main__.Bar object at 0x2929f50>
commited 14 <__main__.Foo object at 0x2932e10> None

令我惊讶的是committed中的内容, bar.foo与提交之前的实例不同，显然，结果是我成功更改为 bar.foo.label在事件监听器被抛出窗外。我尝试这样做是因为我需要一个自动生成的唯一字符串 bar.foo.label ，我想要比纯随机字符串更有意义的东西。是否可以自动执行此操作(无论有或没有事件监听器)，还是我必须在比 ORM 模型更高的级别处理此问题？

Answer 1

您的方法存在以下问题:Bar.id 是主键。因此，它是从数据库自动创建的。反过来，这意味着 set 事件在不寻常的地方被调用:在更改(即 Foo 和 Bar 的对象创建)之后已推送到数据库。

所以 SQLAlchemy 认为它已经完成了推送更改。现在您继续进行更改，但 SQLAlchemy 没有注意到这一点(看起来) - 相反，此更改存在于 Python 中，但不存在于数据库中。

可以通过打开查询日志来查看最后一步(提交)的更多详细信息:create_engine(..., echo=True)。现在您将看到执行的查询。 echo="debug" 然后您就会看到返回结果。如果您现在再次运行它，您会注意到数据库被再次查询。

现在发生的情况是，SQLAlchemy 取回所有值，包括 ID，并且(对于 Bar)看到“哦，具有该 ID 的对象已存在于我的 session 中”，并且它返回的正是该内容目的。现在我不能确定为什么这会在一种情况下返回旧对象，而在另一种情况下返回新对象。这对 SQLAlchemy 来说太深奥了，我无法回答，但这里有一个有趣的享受:

如果你这样做 Bar() 身份将会改变，但如果你这样做 Bar(foo=Foo()) 它不会(至少当我测试时)它不会)。但在这两种情况下，您的更改都会丢失，因此您可以看到这不是由于身份问题，而是由于您何时进行更改而导致的问题。

想出一个解决方案并不那么容易。最大的问题来自于你对身份的依赖。我尝试了多种解决方案，但没有一个是真正好的。然而，我至少能够想出这个查询的想法:

UPDATE foo SET label="autostring " || (SELECT bar.id 
    FROM bar 
    WHERE bar.foo_id = foo.id) 
WHERE foo.id=1

这将为服务器端相应的 foo 条目设置标签。您可以将其设置为对所有人执行此操作的查询:

UPDATE foo SET label="autostring " || (SELECT bar.id 
    FROM bar 
    WHERE bar.foo_id = foo.id) 
WHERE foo.label IS NULL

您可能还想询问 SQLAlchemy list有关此问题的帮助:在那里您通常可以找到比这里更有效的帮助。