如何建立允许不同类型用户在 SqlAlchemy 中注册的数据库关系

Question

我有不同的用户，他们将访问平台中的不同功能，并拥有与他们相关的不同信息。我根据下图实现了这些表，其中我有一个包含身份验证数据的用户表，我有两个表，用于两种用户类型及其相应的信息。

我与“zpeedr <-> user”有一对一的关系，对于“merchant <-> user”也是如此。我将 flask_sqlalchemy 与 postgresql 数据库一起使用。如果不按预期将其与现有 user.id 关联，则无法实例化 zpeedr 或商家类型用户。我的问题是，使用此设置，我可以为 zpeedr 和商家分配相同的 user_id。我如何确保一个 user.id 只能由 zpeedr 用户或商家用户关联？

我创建表的简化代码是：

类用户（UserMixin，db.Model）：

    '''用户属性'''

    id = db.Column(db.Integer, primary_key=True)

    角色 = db.Column(db.String(10), nullable=False)

    名称 = db.Column(db.String(50), nullable=False)

    电子邮件 = db.Column（db.String(50)，唯一 = True，可为空 = False）

    password_hash = db.Column(db.String(250), nullable=False)

    状态 = db.Column(db.String(15), server_default="Pending")

    registration_date = db.Column(db.DateTime, server_default=db.func.current_timestamp())

    zpeedr = db.relationship('Zpeedr', backref='user', uselist=False)

    商家 = db.relationship('Merchant', backref='user', uselist=False)

def __init__(self, role, name, email, password_hash, status, registration_date, zpeedr, merchant):

    self.role = role

    self.name = name

    self.email = email

    self.password_hash = password_hash

    self.status = status

    self.registration_date = registration_date

    self.zpeedr = zpeedr

    self.merchant = merchant

Zpeedr 类（db.Model）：

    ''' Zpeedr 属性 '''

    id = db.Column(db.Integer, primary_key=True)

    # 用户表的外键

    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False, unique=True)

    地址 = db.Column(db.String(200), nullable=False)

    位置 = db.Column(db.String(50), nullable=False)

    邮政编码 = db.Column(db.String(10), nullable=False)

    联系人 = db.Column(db.String(15), nullable=False)

    增值税 = db.Column(db.String(20), nullable=False, unique=True)

    nin = db.Column(db.String(20), nullable=False, unique=True)

def __init__(self, user_id, address, location, zipcode, contact, vat, nin):

    self.user_id = user_id

    self.address = address

    self.location = location

    self.zipcode = zipcode

    self.contact = contact

    self.vat = vat

    self.nin = nin

Answer 1

如果表用户的目的只是存储公共数据，您可以使用 postgres继承。在一个非常简化的模型中，您可以拥有：包含所有常见数据的父表create table user_(&nbsp; type varchar(1)&nbsp; );和你的孩子：&nbsp;create table zeepdr(&nbsp; &nbsp;id integer primary key,&nbsp; &nbsp;nin varchar&nbsp; &nbsp;) inherits(user_);&nbsp;create table merchant(&nbsp; &nbsp;id integer primary key,&nbsp; &nbsp;ticker char(4)&nbsp; ) inherits(user_);SQLAlchemy 中对应的代码应该是这样的：Table("zeepdr", metadata, ..., postgresql_inherits="user_")基本用法插入根据我对山羊的看法，您不需要既不是商人也不是 zeepdr 的用户，因此您将插入数据，就好像父表不存在一样：insert into zeepdr(id, type, nin) values(1,'Z','a');insert into merchant(id, type, ticker) values(1,'M','aaaa');这也意味着用户将是商人或 zeepdr。选择select * from zeepdr;行为如你所料。此外，如果您只想从所有用户那里检索公共数据，您可以使用关键字ONLY：select * from ONLY user_;或在 SQLAlchemyresult = table.select().with_hint(table, 'ONLY', 'postgresql')更多信息：约束除非您使用NO INHERIT. 其他类型没有继承。最后一点有关继承和 SQLAlchemy 的更多信息here