SQLAlchemy一对多关系

Question

我目前有一个SQLAlchemy模型User，它有两种相关的测量类型，重量和身高。这些都是作为外键添加的。

class User(UserMixin, db.Model):
__tablename__ = "user"
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(100), unique=True)
password = db.Column(db.String(100))
name = db.Column(db.String(1000))
weight = db.relationship("Weight", cascade="all,delete")
height = db.relationship("Height", cascade="all,delete")
class Weight(db.Model):
__tablename__ = "weight"
id = db.Column(db.Integer, primary_key=True)
date = db.Column(db.DateTime)
weight = db.Column(db.Float)
user_id = db.Column(db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"))
class Height(db.Model):
__tablename__ = "height"
id = db.Column(db.Integer, primary_key=True)
date = db.Column(db.DateTime)
height = db.Column(db.Float)
user_id = db.Column(db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"))

我希望能够添加更多类型的测量，而不必每次都重建数据库结构。因此，我考虑使用一个通用的Measurement模型，它有一列用于测量类型(而不是Enum，所以我可以很容易地添加新类型(：

class User(UserMixin, db.Model):
__tablename__ = "user"
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(100), unique=True)
password = db.Column(db.String(100))
name = db.Column(db.String(1000))
measurements = db.relationship("Measurement", cascade="all,delete")
@property
def weight(self):
pass    # <-- No idea what to do here
@property
def height(self):
pass    # <-- No idea what to do here
class Measurement(db.Model):
__tablename__ = "measurement"
id = db.Column(db.Integer, primary_key=True)
date = db.Column(db.DateTime)
value = db.Column(db.Float)
user_id = db.Column(db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"))
measurement_type_id = db.Column(db.Integer, db.ForeignKey('measurement_type.id'))
measurement_type = db.relationship("MeasurementType")
class MeasurementType(db.Model):
__tablename__ = "measurement_type"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100))
unit = db.Column(db.String(100), nullable=True)

有了这个新模型，我如何只得到例如权重？对于当前的模型，这就这么简单，其中current_user是当前登录的用户：

current_user.weight

但我想它一定是这样的：

current_user.measurements.filter(Measurement.measurement_type == "Weight")

(这不起作用，因为current_user.measurements返回一个列表。(

同样，我该如何在测量中添加一个新值？目前我这样做：

current_user.weight.append(Weight(date=date, weight=weight))
db.session.commit()

我是否基本上需要复制relationship的底层实现，以便通过用户ID和测量类型这两件事进行过滤？

或者我可以使用关联代理以某种方式实现这一点吗？或者(正确地(使用relationship的primaryjoin参数？

Answer 1

我发现实现这一点的最简单方法是对关系使用动态延迟加载，这也是对类似问题的回答所建议的。这样，返回类型就不再是一个简单的列表，而是一个SQL对象。这也使用一个简单的字符串作为测量类型，而不是查找表。

class User(UserMixin, db.Model):
__tablename__ = "user"
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(100), unique=True)
password = db.Column(db.String(100))
name = db.Column(db.String(1000))
measurements = db.relationship("Measurement", cascade="all,delete", lazy="dynamic")

class Measurement(db.Model):
__tablename__ = "measurement"
id = db.Column(db.Integer, primary_key=True)
m_type = db.Column(db.String(20))
date = db.Column(db.DateTime)
value = db.Column(db.Float)
user_id = db.Column(db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"))

之后，您可以对测量使用正常的过滤方法：

list(current_user.measurements.filter_by(m_type="weight"))

添加新行也很简单：

current_user.measurements.append(Measurement(m_type="weight", date=date, value=weight))