1

首先,示例代码:

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

令我惊讶的是, in与提交之前的实例不同committedbar.foo显然结果是,我bar.foo.label对事件侦听器的成功更改被抛出了窗口。我正在尝试这样做,因为我需要一个自动生成的唯一字符串bar.foo.label,并且我想要比纯随机字符串更有意义的东西。是否可以在有或没有事件监听器的情况下自动执行此操作,或者我是否必须在比 ORM 模型更高的级别上处理它?

4

2 回答 2

1

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

所以 SQLAlchemy 认为它已经完成了推送更改。现在您继续进行更改,但 SQLAlchemy 并没有注意到(似乎) - 相反,此更改存在于 Python 中,但不存在于数据库中。

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

现在发生的事情是 SQLAlchemy 取回所有值,包括 ID,并且 (for 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 列表寻求有关此方面的帮助:通常,您可以在此处找到比此处更有效的帮助。

于 2013-08-29T11:35:55.020 回答
1

感谢@javex 让我重新思考我的问题;我对他的回答的评论背后的想法导致了我的解决方案。我从setevent on切换Bar.idafter_insertevent on 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)

显然,我必须通过使用适当的插入对象和语句构建器来清理它,但这个概念是有效的。

于 2013-08-29T14:23:44.423 回答