diff --git a/database_setup/database_test.py b/database_setup/database_test.py index c91c78f..05e7090 100644 --- a/database_setup/database_test.py +++ b/database_setup/database_test.py @@ -60,6 +60,7 @@ class CategoryGroup(SQLModel, table=True): __table_args__ = (UniqueConstraint("guild_id", "name", name="UniqueGuildCategoryGroup"),) guild: Guild | None = Relationship(back_populates="category_groups") + categories: list["Category"] | None = Relationship(back_populates="category_group", cascade_delete=True) class Category(SQLModel, table=True): @@ -81,6 +82,9 @@ class Category(SQLModel, table=True): __table_args__ = (UniqueConstraint("group_id", "name", name="UniqueCategoryInGroup"),) + category_group: CategoryGroup | None = Relationship(back_populates="categories") + scores: list["Score"] | None = Relationship(back_populates="category", cascade_delete=True) + class Score(SQLModel, table=True): __tablename__: str = "Scores" @@ -96,6 +100,8 @@ class Score(SQLModel, table=True): ), ) + category: Category | None = Relationship(back_populates="scores") + class Submission(SQLModel, table=True): __tablename__: str = "Submissions"