feat: mixins

This commit is contained in:
2025-08-05 15:48:03 +04:00
parent be1ea4009d
commit 3e2839e725
5 changed files with 33 additions and 41 deletions

View File

@ -1,37 +1,24 @@
from datetime import datetime
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from sqlalchemy import ForeignKey from sqlalchemy import ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
from models.base import BaseModel from models.base import BaseModel
from models.mixins import SoftDeleteMixin, CreatedAtMixin, IdMixin
if TYPE_CHECKING: if TYPE_CHECKING:
from models import Project, Status, Deal from models import Project, Status, Deal
class Board(BaseModel): class Board(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
__tablename__ = "boards" __tablename__ = "boards"
id: Mapped[int] = mapped_column(primary_key=True)
name: Mapped[str] = mapped_column(nullable=False) name: Mapped[str] = mapped_column(nullable=False)
lexorank: Mapped[str] = mapped_column(nullable=False) lexorank: Mapped[str] = mapped_column(nullable=False)
created_at: Mapped[datetime] = mapped_column(nullable=False)
is_deleted: Mapped[bool] = mapped_column(default=False)
project_id: Mapped[int] = mapped_column(ForeignKey("projects.id"), nullable=False) project_id: Mapped[int] = mapped_column(ForeignKey("projects.id"), nullable=False)
project: Mapped["Project"] = relationship( project: Mapped["Project"] = relationship(back_populates="boards")
"Project",
back_populates="boards",
)
statuses: Mapped[list["Status"]] = relationship( statuses: Mapped[list["Status"]] = relationship(back_populates="board")
"Status",
back_populates="board",
)
deals: Mapped[list["Deal"]] = relationship( deals: Mapped[list["Deal"]] = relationship(back_populates="board")
"Deal",
back_populates="board",
)

View File

@ -4,15 +4,15 @@ from sqlalchemy import ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
from models.base import BaseModel from models.base import BaseModel
from models.mixins import SoftDeleteMixin, CreatedAtMixin, IdMixin
if TYPE_CHECKING: if TYPE_CHECKING:
from models import Status, Board from models import Status, Board
class Deal(BaseModel): class Deal(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
__tablename__ = "deals" __tablename__ = "deals"
id: Mapped[int] = mapped_column(primary_key=True)
name: Mapped[str] = mapped_column(nullable=False) name: Mapped[str] = mapped_column(nullable=False)
lexorank: Mapped[str] = mapped_column(nullable=False) lexorank: Mapped[str] = mapped_column(nullable=False)
@ -21,14 +21,9 @@ class Deal(BaseModel):
nullable=False, nullable=False,
comment="Текущий статус", comment="Текущий статус",
) )
status: Mapped["Status"] = relationship(lazy="selectin") status: Mapped["Status"] = relationship(lazy="noload")
is_deleted: Mapped[bool] = mapped_column(default=False)
board_id: Mapped[int] = mapped_column( board_id: Mapped[int] = mapped_column(
ForeignKey("boards.id"), nullable=True, server_default="1" ForeignKey("boards.id"), nullable=True, server_default="1"
) )
board: Mapped["Board"] = relationship( board: Mapped["Board"] = relationship(back_populates="deals")
"Board",
back_populates="deals",
)

18
models/mixins.py Normal file
View File

@ -0,0 +1,18 @@
from datetime import datetime
from sqlalchemy.orm import Mapped, mapped_column
class IdMixin:
id: Mapped[int] = mapped_column(primary_key=True)
class SoftDeleteMixin:
is_deleted: Mapped[bool] = mapped_column(
default=False,
nullable=False,
)
class CreatedAtMixin:
created_at: Mapped[datetime] = mapped_column(nullable=False)

View File

@ -1,24 +1,20 @@
from datetime import datetime
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
from models.base import BaseModel from models.base import BaseModel
from models.mixins import SoftDeleteMixin, CreatedAtMixin, IdMixin
if TYPE_CHECKING: if TYPE_CHECKING:
from models import Board from models import Board
class Project(BaseModel): class Project(BaseModel, IdMixin, SoftDeleteMixin, CreatedAtMixin):
__tablename__ = "projects" __tablename__ = "projects"
id: Mapped[int] = mapped_column(primary_key=True)
name: Mapped[str] = mapped_column(nullable=False) name: Mapped[str] = mapped_column(nullable=False)
created_at: Mapped[datetime] = mapped_column(nullable=False)
is_deleted: Mapped[bool] = mapped_column(default=False)
boards: Mapped[list["Board"]] = relationship( boards: Mapped[list["Board"]] = relationship(
"Board",
back_populates="project", back_populates="project",
lazy="noload", lazy="noload",
) )

View File

@ -3,22 +3,18 @@ from typing import TYPE_CHECKING
from sqlalchemy import ForeignKey from sqlalchemy import ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
from models import BaseModel from models.base import BaseModel
from models.mixins import SoftDeleteMixin, IdMixin
if TYPE_CHECKING: if TYPE_CHECKING:
from models import Board from models import Board
class Status(BaseModel): class Status(BaseModel, IdMixin, SoftDeleteMixin):
__tablename__ = "statuses" __tablename__ = "statuses"
id: Mapped[int] = mapped_column(primary_key=True)
name: Mapped[str] = mapped_column(nullable=False) name: Mapped[str] = mapped_column(nullable=False)
lexorank: Mapped[str] = mapped_column(nullable=False) lexorank: Mapped[str] = mapped_column(nullable=False)
is_deleted: Mapped[bool] = mapped_column(default=False, nullable=False)
board_id: Mapped[int] = mapped_column(ForeignKey("boards.id"), nullable=False) board_id: Mapped[int] = mapped_column(ForeignKey("boards.id"), nullable=False)
board: Mapped["Board"] = relationship( board: Mapped["Board"] = relationship(back_populates="statuses")
"Board",
back_populates="statuses",
)