python - 如何在 sqlalchemy 中提交之前应用列默认值

Question

我有一个基于声明的模型:

class User(Base):
    id = Column(Integer, primary_key=True)
    money = Column(Integer, default=100)

然后我跑

>>> u = User()
>>> u.money
None

如何使用 sqlalchemy 填充默认值而不向数据库写入任何内容？

Answer 1

该列默认值仅适用于 INSERT 和 UPDATE 语句，因此在您 .flush() session 之前不会被应用。

要在刷新之前在新实例上看到相同的值，您需要在创建新实例时应用默认值；在 User 对象的 __init__ 方法中:

class User(Base):
    __tablename__ = 'users'

    def __init__(self, **kwargs):
        if 'money' not in kwargs:
             kwargs['money'] = self.__table__.c.money.default.arg
        super(User, self).__init__(**kwargs)

    id = Column(Integer, primary_key=True)
    money = Column(Integer, default=100)

如果没有设置 money 属性，我们直接根据为该列配置的默认值添加一个。

请注意，默认值是 SQL 表达式，而不是 Python 值，因此您可能必须先将它们映射到 Python 对象。例如， bool 字段将具有默认的 'false' 或 'true' 字符串值，而不是 False 或 True Python bool 对象。