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的对象”并且它返回该对象。现在我无法确定为什么这会为一个案例返回旧对象而另一个案例返回一个新对象。这在SQLAlchemy中远比我能回答的要深，但这是一个有趣的享受：

如果你做Bar()，身份会发生变化，但如果你做Bar(foo=Foo())则不会（至少我测试时不会）。但在这两种情况下，您的更改都会丢失，因此您可以看到这不是由于身份问题，而是您在何时进行更改时出现问题。

想出一个解决方案并不容易。最大的问题来自于你对id的依赖。我尝试了多种解决方案但没有一种方法真的很好。但是，我至少能够为查询提出这个想法：

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询问有关此问题的帮助：在那里，您通常可以找到比此处更有能力的帮助。

Answer 2

感谢@javex让我重新思考我的问题;我对他的回答的评论背后的想法导致了我的解决方案。我从set上的Bar.id事件切换到after_insert上的Bar事件，然后直接通过连接发送查询以更新触发的函数中的foo.label事件。

def _adjust_label(mapper, connection, target):

    print 'adjust', mapper, connection, target
    if not target.foo.label:
        connection.execute('UPDATE foo SET label="autostring %d" WHERE id=%d' % (target.id, target.foo.id))

event.listen(Bar, 'after_insert', _adjust_label)

我必须通过使用正确的插入对象和语句构建器来清理它，但这个概念有效。

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

2 个答案: