mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-10-27 08:14:30 -04:00
a98e863bca
* fixed missing migration name * added unique constraints to all m2m tables * fixed bug trying to create duplicate tags * added more unique constraints * fixed duplicate seeder data * updated tests * fixed seed rollback error
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import sqlalchemy as sa
|
|
import sqlalchemy.orm as orm
|
|
from slugify import slugify
|
|
from sqlalchemy.orm import Mapped, mapped_column, validates
|
|
|
|
from mealie.core import root_logger
|
|
|
|
from .._model_base import BaseMixins, SqlAlchemyBase
|
|
from .._model_utils.guid import GUID
|
|
|
|
if TYPE_CHECKING:
|
|
from ..group import Group
|
|
from . import RecipeModel
|
|
logger = root_logger.get_logger()
|
|
|
|
|
|
group_to_categories = sa.Table(
|
|
"group_to_categories",
|
|
SqlAlchemyBase.metadata,
|
|
sa.Column("group_id", GUID, sa.ForeignKey("groups.id"), index=True),
|
|
sa.Column("category_id", GUID, sa.ForeignKey("categories.id"), index=True),
|
|
sa.UniqueConstraint("group_id", "category_id", name="group_id_category_id_key"),
|
|
)
|
|
|
|
plan_rules_to_categories = sa.Table(
|
|
"plan_rules_to_categories",
|
|
SqlAlchemyBase.metadata,
|
|
sa.Column("group_plan_rule_id", GUID, sa.ForeignKey("group_meal_plan_rules.id"), index=True),
|
|
sa.Column("category_id", GUID, sa.ForeignKey("categories.id"), index=True),
|
|
sa.UniqueConstraint("group_plan_rule_id", "category_id", name="group_plan_rule_id_category_id_key"),
|
|
)
|
|
|
|
recipes_to_categories = sa.Table(
|
|
"recipes_to_categories",
|
|
SqlAlchemyBase.metadata,
|
|
sa.Column("recipe_id", GUID, sa.ForeignKey("recipes.id"), index=True),
|
|
sa.Column("category_id", GUID, sa.ForeignKey("categories.id"), index=True),
|
|
sa.UniqueConstraint("recipe_id", "category_id", name="recipe_id_category_id_key"),
|
|
)
|
|
|
|
cookbooks_to_categories = sa.Table(
|
|
"cookbooks_to_categories",
|
|
SqlAlchemyBase.metadata,
|
|
sa.Column("cookbook_id", GUID, sa.ForeignKey("cookbooks.id"), index=True),
|
|
sa.Column("category_id", GUID, sa.ForeignKey("categories.id"), index=True),
|
|
sa.UniqueConstraint("cookbook_id", "category_id", name="cookbook_id_category_id_key"),
|
|
)
|
|
|
|
|
|
class Category(SqlAlchemyBase, BaseMixins):
|
|
__tablename__ = "categories"
|
|
__table_args__ = (sa.UniqueConstraint("slug", "group_id", name="category_slug_group_id_key"),)
|
|
|
|
# ID Relationships
|
|
group_id: Mapped[GUID] = mapped_column(GUID, sa.ForeignKey("groups.id"), nullable=False, index=True)
|
|
group: Mapped["Group"] = orm.relationship("Group", back_populates="categories", foreign_keys=[group_id])
|
|
|
|
id: Mapped[GUID] = mapped_column(GUID, primary_key=True, default=GUID.generate)
|
|
name: Mapped[str] = mapped_column(sa.String, index=True, nullable=False)
|
|
slug: Mapped[str] = mapped_column(sa.String, index=True, nullable=False)
|
|
recipes: Mapped[list["RecipeModel"]] = orm.relationship(
|
|
"RecipeModel", secondary=recipes_to_categories, back_populates="recipe_category"
|
|
)
|
|
|
|
@validates("name")
|
|
def validate_name(self, key, name):
|
|
assert name != ""
|
|
return name
|
|
|
|
def __init__(self, name, group_id, **_) -> None:
|
|
self.group_id = group_id
|
|
self.name = name.strip()
|
|
self.slug = slugify(name)
|