crm-system/app/models/activity.py

"""Activity model: calls, meetings, emails, tasks, notes tied to any entity."""

from __future__ import annotations

from datetime import datetime
from enum import Enum
from typing import TYPE_CHECKING

from sqlalchemy import DateTime, ForeignKey, String, Text
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.models.base import Base, OrgScopedMixin, SoftDeleteMixin, TimestampMixin

if TYPE_CHECKING:
    from app.models.account import Account
    from app.models.contact import Contact
    from app.models.deal import Deal
    from app.models.user import User


class ActivityType(str, Enum):
    """Type of activity."""

    call = "call"
    meeting = "meeting"
    email = "email"
    task = "task"
    note = "note"


class Activity(Base, TimestampMixin, SoftDeleteMixin, OrgScopedMixin):
    __tablename__ = "activities"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    type: Mapped[ActivityType] = mapped_column(String(32), nullable=False, index=True)
    subject: Mapped[str] = mapped_column(String(255), nullable=False)
    body: Mapped[str | None] = mapped_column(Text, nullable=True)
    due_date: Mapped[datetime | None] = mapped_column(
        DateTime(timezone=True), nullable=True, index=True
    )
    completed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    account_id: Mapped[int | None] = mapped_column(
        ForeignKey("accounts.id"), nullable=True, index=True
    )
    contact_id: Mapped[int | None] = mapped_column(
        ForeignKey("contacts.id"), nullable=True, index=True
    )
    deal_id: Mapped[int | None] = mapped_column(ForeignKey("deals.id"), nullable=True, index=True)
    owner_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False, index=True)

    # Relationships
    account: Mapped[Account | None] = relationship(
        "Account", back_populates="activities", lazy="selectin"
    )
    contact: Mapped[Contact | None] = relationship(
        "Contact", back_populates="activities", lazy="selectin"
    )
    deal: Mapped[Deal | None] = relationship("Deal", back_populates="activities", lazy="selectin")
    owner: Mapped[User] = relationship("User", foreign_keys=[owner_id], lazy="joined")

    def __repr__(self) -> str:
        return f"<Activity id={self.id} type={self.type} subject={self.subject!r}>"


__all__ = ["Activity", "ActivityType"]