如何将sqlalchemy中列的默认值设置为关系中列的值？

Posted 2023-02-04

我正在为sqlite数据库设置Sqlalchemy映射器。我的User类与我的Team类之间存在不可为空的关系。我已经拥有的代码如下：

class Team(Base):
    __tablename__ = 'teams'

    team_id = Column(Integer, primary_key=True)
    # Using Integer as holder for boolean
    is_local = Column(Integer, default=0)

class User(Base):
    __tablename__ = 'users'

    user_id = Column(Integer, primary_key=True)
    team_id = Column(Integer, ForeignKey(Team.team_id), default=1, nullable=False)
    team = relationship('Team')
    is_local = Column(Integer, default=0)

我想确定User.is_local的值默认为用户链接团队的Team.is_local值。但是，在创建用户之后，我仍然希望能够在不更改团队或团队中任何其他用户的值的情况下修改用户的is_local值。

所以，如果我要执行

faraway = Team(is_local=1)
session.add(faraway)
session.commit()

u = User(team=faraway)
session.add(u)
session.commit()

print(bool(u.is_local))

结果应该是True

到目前为止，我已经尝试过https://stackoverflow.com/a/36579924建议的上下文敏感的默认函数，但我找不到允许我引用Team.is_local的语法

有一个简单的方法吗？

答案

SuperShoot的第一个建议，使用sql expression as the default似乎工作。特别，

    is_local = Column(Integer, default=select([Team.is_local]).where(Team.team_id==team_id))

给了我我需要的逻辑。