feat: deal tags
This commit is contained in:
@ -10,6 +10,11 @@ from .built_in_module import ( # noqa: F401
|
||||
)
|
||||
from .deal import Deal as Deal
|
||||
from .deal_group import DealGroup as DealGroup
|
||||
from .deal_tag import (
|
||||
DealTag as DealTag,
|
||||
DealTagColor as DealTagColor,
|
||||
deals_deal_tags as deals_deal_tags,
|
||||
)
|
||||
from .project import Project as Project
|
||||
from .status import Status as Status, DealStatusHistory as DealStatusHistory
|
||||
|
||||
|
||||
@ -7,7 +7,7 @@ from models.base import BaseModel
|
||||
from models.mixins import SoftDeleteMixin, CreatedAtMixin, IdMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from models import Status, Board, DealStatusHistory, DealGroup
|
||||
from models import Status, Board, DealStatusHistory, DealGroup, DealTag
|
||||
from modules.clients.models import Client
|
||||
|
||||
|
||||
@ -41,6 +41,14 @@ class Deal(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
|
||||
lazy="noload", back_populates="deals"
|
||||
)
|
||||
|
||||
tags: Mapped[list["DealTag"]] = relationship(
|
||||
secondary="deals_deal_tags",
|
||||
back_populates="deals",
|
||||
lazy="selectin",
|
||||
primaryjoin="Deal.id == deals_deal_tags.c.deal_id",
|
||||
secondaryjoin="and_(DealTag.id == deals_deal_tags.c.deal_tag_id, DealTag.is_deleted == False)",
|
||||
)
|
||||
|
||||
# module client
|
||||
client_id: Mapped[Optional[int]] = mapped_column(
|
||||
ForeignKey("clients.id", ondelete="CASCADE"),
|
||||
|
||||
51
models/deal_tag.py
Normal file
51
models/deal_tag.py
Normal file
@ -0,0 +1,51 @@
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import ForeignKey, Column, Table, Index
|
||||
from sqlalchemy.orm import mapped_column, Mapped, relationship
|
||||
|
||||
from models import BaseModel
|
||||
from models.mixins import IdMixin, SoftDeleteMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from models import Project, Deal
|
||||
|
||||
deals_deal_tags = Table(
|
||||
"deals_deal_tags",
|
||||
BaseModel.metadata,
|
||||
Column("deal_id", ForeignKey("deals.id"), primary_key=True),
|
||||
Column("deal_tag_id", ForeignKey("deal_tags.id"), primary_key=True),
|
||||
)
|
||||
|
||||
|
||||
class DealTagColor(BaseModel, IdMixin):
|
||||
__tablename__ = "deal_tag_colors"
|
||||
|
||||
label: Mapped[str] = mapped_column(unique=True)
|
||||
color: Mapped[str] = mapped_column(unique=True)
|
||||
background_color: Mapped[str] = mapped_column(unique=True)
|
||||
is_deleted: Mapped[bool] = mapped_column(default=False)
|
||||
|
||||
|
||||
class DealTag(BaseModel, IdMixin, SoftDeleteMixin):
|
||||
__tablename__ = "deal_tags"
|
||||
|
||||
name: Mapped[str] = mapped_column()
|
||||
|
||||
project_id: Mapped[int] = mapped_column(ForeignKey("projects.id"))
|
||||
project: Mapped["Project"] = relationship(
|
||||
back_populates="tags",
|
||||
lazy="noload",
|
||||
)
|
||||
|
||||
deals: Mapped[list["Deal"]] = relationship(
|
||||
secondary="deals_deal_tags",
|
||||
lazy="noload",
|
||||
back_populates="tags",
|
||||
)
|
||||
|
||||
tag_color_id: Mapped[int] = mapped_column(ForeignKey("deal_tag_colors.id"))
|
||||
tag_color: Mapped[DealTagColor] = relationship(lazy="immediate")
|
||||
|
||||
__table_args__ = (
|
||||
Index("idx_deal_name_project_id", "name", "project_id", "is_deleted"),
|
||||
)
|
||||
@ -6,7 +6,7 @@ from models.base import BaseModel
|
||||
from models.mixins import SoftDeleteMixin, CreatedAtMixin, IdMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from models import Board, BuiltInModule
|
||||
from models import Board, BuiltInModule, DealTag
|
||||
|
||||
|
||||
class Project(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
|
||||
@ -25,3 +25,10 @@ class Project(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
|
||||
lazy="selectin",
|
||||
order_by="asc(BuiltInModule.id)",
|
||||
)
|
||||
|
||||
tags: Mapped[list["DealTag"]] = relationship(
|
||||
back_populates="project",
|
||||
primaryjoin="and_(Project.id == DealTag.project_id, DealTag.is_deleted == False)",
|
||||
order_by="asc(DealTag.id)",
|
||||
lazy="selectin",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user