95 lines
3.0 KiB
Python
95 lines
3.0 KiB
Python
|
"""Deal model: sales opportunities tied to an account."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import date
|
||
|
|
from decimal import Decimal
|
||
|
|
from enum import Enum
|
||
|
|
from typing import TYPE_CHECKING, Optional
|
||
|
|
|
||
|
|
from sqlalchemy import Date, ForeignKey, Numeric, String
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.models.base import Base, OrgScopedMixin, SoftDeleteMixin, TimestampMixin
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from app.models.user import User
|
||
|
|
from app.models.account import Account
|
||
|
|
from app.models.activity import Activity
|
||
|
|
from app.models.note import Note
|
||
|
|
from app.models.tag_link import TagLink
|
||
|
|
from app.models.deal_stage_history import DealStageHistory
|
||
|
|
|
||
|
|
|
||
|
|
class DealStage(str, Enum):
|
||
|
|
"""Sales pipeline stage for a deal."""
|
||
|
|
|
||
|
|
lead = "lead"
|
||
|
|
qualified = "qualified"
|
||
|
|
proposal = "proposal"
|
||
|
|
negotiation = "negotiation"
|
||
|
|
won = "won"
|
||
|
|
lost = "lost"
|
||
|
|
|
||
|
|
|
||
|
|
class Deal(Base, TimestampMixin, SoftDeleteMixin, OrgScopedMixin):
|
||
|
|
__tablename__ = "deals"
|
||
|
|
|
||
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||
|
|
title: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
|
|
value: Mapped[Decimal] = mapped_column(Numeric(12, 2), nullable=False, default=Decimal("0"))
|
||
|
|
currency: Mapped[str] = mapped_column(
|
||
|
|
String(3), nullable=False, default="EUR", server_default="EUR"
|
||
|
|
)
|
||
|
|
stage: Mapped[DealStage] = mapped_column(
|
||
|
|
String(32),
|
||
|
|
nullable=False,
|
||
|
|
default=DealStage.lead,
|
||
|
|
server_default=DealStage.lead.value,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
close_date: Mapped[Optional[date]] = mapped_column(Date, nullable=True)
|
||
|
|
account_id: Mapped[int] = mapped_column(
|
||
|
|
ForeignKey("accounts.id"), nullable=False, index=True
|
||
|
|
)
|
||
|
|
owner_id: Mapped[int] = mapped_column(
|
||
|
|
ForeignKey("users.id"), nullable=False, index=True
|
||
|
|
)
|
||
|
|
won_lost_reason: Mapped[Optional[str]] = mapped_column(String(512), nullable=True)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
account: Mapped["Account"] = relationship(
|
||
|
|
"Account", back_populates="deals", lazy="selectin"
|
||
|
|
)
|
||
|
|
owner: Mapped["User"] = relationship(
|
||
|
|
"User", foreign_keys=[owner_id], lazy="joined"
|
||
|
|
)
|
||
|
|
stage_history: Mapped[list["DealStageHistory"]] = relationship(
|
||
|
|
"DealStageHistory",
|
||
|
|
back_populates="deal",
|
||
|
|
cascade="all, delete-orphan",
|
||
|
|
lazy="selectin",
|
||
|
|
order_by="DealStageHistory.created_at",
|
||
|
|
)
|
||
|
|
activities: Mapped[list["Activity"]] = relationship(
|
||
|
|
"Activity", back_populates="deal", lazy="selectin"
|
||
|
|
)
|
||
|
|
notes: Mapped[list["Note"]] = relationship(
|
||
|
|
"Note",
|
||
|
|
primaryjoin="and_(Deal.id==foreign(Note.parent_id), Note.parent_type=='deal')",
|
||
|
|
viewonly=True,
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
tags: Mapped[list["TagLink"]] = relationship(
|
||
|
|
"TagLink",
|
||
|
|
primaryjoin="and_(Deal.id==foreign(TagLink.parent_id), TagLink.parent_type=='deal')",
|
||
|
|
viewonly=True,
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
|
||
|
|
def __repr__(self) -> str:
|
||
|
|
return f"<Deal id={self.id} title={self.title!r} stage={self.stage}>"
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ["Deal", "DealStage"]
|