python - 挂钩到 sqlalchemy 模型

Question

我正在寻找 sqlalchemy 模型的模型创建周期。例如在创建或保存时（如在 Ruby ORM ActiveRecord 中，实际上我正在将模型从 ActiveRecord 移动到 SqlAlchemy）。

事件看起来像我需要的： http: //docs.sqlalchemy.org/en/rel_0_7/core/event.html，但我还没有找到更详细的例子。我想听听有人对此的经验。

sqlalchemy 中是否有类似的工具可以根据某些线索（例如 after_create）使用模型/实例进行操作？

Answer 1

一旦你掌握了它的窍门，事件就非常简单。这是一个使用事件的快速示例

import uuid    

from sqlalchemy.event import listen

from mypackage.models import Base


def generate_license(mapper, connect, target):
    target.generate_license()

class User(Base):
    __tablename__ = "users"
    id = Column(String(36))
    license = Column(String(256))

    def generate_license(self):
        if not self.license:
            self.license = str(uuid.uuid4())
        return self.license

listen(User, 'before_insert', generate_license)

或者，您可以使用装饰器：

from sqlalchemy.event import listens_for
…
class User(Base):
    …

@listens_for(User, 'before_insert')
def generate_license(mapper, connect, self):
    …

Answer 2

from sqlalchemy.event import listen_for
…
class User(Base):
…
    @listen_for(User, 'before_insert')
    @staticmethod
    def generate_license(mapper, connect, self):
…

这将返回

NameError: name 'User' is not defined