90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
import enum
|
|
from typing import TYPE_CHECKING, Optional
|
|
|
|
from sqlalchemy import Table, Column, ForeignKey
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from models import AttributeValue, Attribute
|
|
from models.base import BaseModel
|
|
|
|
if TYPE_CHECKING:
|
|
from models import Project
|
|
|
|
project_module = Table(
|
|
"project_module",
|
|
BaseModel.metadata,
|
|
Column("project_id", ForeignKey("projects.id"), primary_key=True),
|
|
Column("module_id", ForeignKey("modules.id"), primary_key=True),
|
|
)
|
|
|
|
|
|
module_dependencies = Table(
|
|
"module_dependencies",
|
|
BaseModel.metadata,
|
|
Column("module_id", ForeignKey("modules.id"), primary_key=True),
|
|
Column("depends_on_id", ForeignKey("modules.id"), primary_key=True),
|
|
)
|
|
|
|
|
|
class Module(BaseModel):
|
|
__tablename__ = "modules"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
key: Mapped[str] = mapped_column(unique=True)
|
|
label: Mapped[str] = mapped_column()
|
|
description: Mapped[Optional[str]] = mapped_column()
|
|
is_deleted: Mapped[bool] = mapped_column(default=False)
|
|
is_built_in: Mapped[bool] = mapped_column(default=False, server_default="0")
|
|
|
|
depends_on: Mapped[list["Module"]] = relationship(
|
|
secondary=module_dependencies,
|
|
primaryjoin="Module.id == module_dependencies.c.module_id",
|
|
secondaryjoin="Module.id == module_dependencies.c.depends_on_id",
|
|
back_populates="depended_on_by",
|
|
lazy="immediate",
|
|
)
|
|
|
|
depended_on_by: Mapped[list["Module"]] = relationship(
|
|
secondary="module_dependencies",
|
|
primaryjoin="Module.id == module_dependencies.c.depends_on_id",
|
|
secondaryjoin="Module.id == module_dependencies.c.module_id",
|
|
back_populates="depends_on",
|
|
lazy="noload",
|
|
)
|
|
|
|
projects: Mapped[list["Project"]] = relationship(
|
|
uselist=True,
|
|
secondary="project_module",
|
|
back_populates="modules",
|
|
lazy="noload",
|
|
)
|
|
|
|
tabs: Mapped[list["ModuleTab"]] = relationship(
|
|
lazy="immediate", backref="module", cascade="all, delete-orphan"
|
|
)
|
|
|
|
attributes: Mapped[list["Attribute"]] = relationship(
|
|
secondary="module_attribute", back_populates="modules", lazy="noload"
|
|
)
|
|
|
|
attribute_values: Mapped[list["AttributeValue"]] = relationship(
|
|
lazy="noload", back_populates="module"
|
|
)
|
|
|
|
|
|
class DeviceType(enum.StrEnum):
|
|
MOBILE = "mobile"
|
|
DESKTOP = "desktop"
|
|
BOTH = "both"
|
|
|
|
|
|
class ModuleTab(BaseModel):
|
|
__tablename__ = "module_tab"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
key: Mapped[str] = mapped_column(unique=True)
|
|
label: Mapped[str] = mapped_column()
|
|
icon_name: Mapped[Optional[str]] = mapped_column()
|
|
module_id: Mapped[int] = mapped_column(ForeignKey("modules.id"))
|
|
device: Mapped[DeviceType] = mapped_column(default=DeviceType.BOTH)
|