feat: 为多个模型的外键字段添加级联删除和其他约束

This commit is contained in:
2025-12-23 11:00:09 +08:00
parent 1a78c76d02
commit 96bf447426
15 changed files with 206 additions and 47 deletions

View File

@@ -15,10 +15,18 @@ if TYPE_CHECKING:
class GroupPolicyLink(SQLModelBase, table=True):
"""用户组与存储策略的多对多关联表"""
group_id: UUID = Field(foreign_key="group.id", primary_key=True)
group_id: UUID = Field(
foreign_key="group.id",
primary_key=True,
ondelete="CASCADE"
)
"""用户组UUID"""
policy_id: UUID = Field(foreign_key="policy.id", primary_key=True)
policy_id: UUID = Field(
foreign_key="policy.id",
primary_key=True,
ondelete="CASCADE"
)
"""存储策略UUID"""
@@ -52,7 +60,11 @@ class PolicyOptionsBase(SQLModelBase):
class PolicyOptions(PolicyOptionsBase, UUIDTableBaseMixin):
"""存储策略选项模型与Policy一对一关联"""
policy_id: UUID = Field(foreign_key="policy.id", unique=True)
policy_id: UUID = Field(
foreign_key="policy.id",
unique=True,
ondelete="CASCADE"
)
"""关联的策略UUID"""
# 反向关系
@@ -105,7 +117,7 @@ class Policy(SQLModelBase, UUIDTableBaseMixin):
# 一对一关系:策略选项
options: PolicyOptions | None = Relationship(
back_populates="policy",
sa_relationship_kwargs={"uselist": False},
sa_relationship_kwargs={"uselist": False, "cascade": "all, delete-orphan"},
)
"""策略的扩展选项"""