Agent Zero
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
"""Account model: companies/organizations that the CRM tracks."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from sqlalchemy import JSON, ForeignKey, String
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.models.base import Base, OrgScopedMixin, SoftDeleteMixin, TimestampMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.activity import Activity
|
|
from app.models.contact import Contact
|
|
from app.models.deal import Deal
|
|
from app.models.note import Note
|
|
from app.models.tag_link import TagLink
|
|
from app.models.user import User
|
|
|
|
|
|
class Industry(str, Enum):
|
|
"""Industry classification for accounts."""
|
|
|
|
startup = "startup"
|
|
sme = "sme"
|
|
midmarket = "midmarket"
|
|
enterprise = "enterprise"
|
|
other = "other"
|
|
|
|
|
|
class AccountSize(str, Enum):
|
|
"""Company size category."""
|
|
|
|
startup = "startup"
|
|
sme = "sme"
|
|
midmarket = "midmarket"
|
|
enterprise = "enterprise"
|
|
|
|
|
|
class Account(Base, TimestampMixin, SoftDeleteMixin, OrgScopedMixin):
|
|
__tablename__ = "accounts"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
name: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
|
|
website: Mapped[str | None] = mapped_column(String(512), nullable=True)
|
|
industry: Mapped[Industry | None] = mapped_column(String(32), nullable=True, index=True)
|
|
size: Mapped[AccountSize | None] = mapped_column(String(32), nullable=True)
|
|
address: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
|
|
owner_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False, index=True)
|
|
|
|
# Relationships
|
|
owner: Mapped[User] = relationship("User", foreign_keys=[owner_id], lazy="joined")
|
|
contacts: Mapped[list[Contact]] = relationship(
|
|
"Contact", back_populates="account", lazy="selectin"
|
|
)
|
|
deals: Mapped[list[Deal]] = relationship("Deal", back_populates="account", lazy="selectin")
|
|
activities: Mapped[list[Activity]] = relationship(
|
|
"Activity", back_populates="account", lazy="selectin"
|
|
)
|
|
notes: Mapped[list[Note]] = relationship(
|
|
"Note",
|
|
primaryjoin="and_(Account.id==foreign(Note.parent_id), Note.parent_type=='account')",
|
|
viewonly=True,
|
|
lazy="selectin",
|
|
)
|
|
tags: Mapped[list[TagLink]] = relationship(
|
|
"TagLink",
|
|
primaryjoin="and_(Account.id==foreign(TagLink.parent_id), TagLink.parent_type=='account')",
|
|
viewonly=True,
|
|
lazy="selectin",
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<Account id={self.id} name={self.name!r}>"
|
|
|
|
|
|
__all__ = ["Account", "AccountSize", "Industry"]
|