Agent Zero
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
"""Base model and reusable mixins for the CRM system."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, ForeignKey, func
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.core.db import Base
|
|
|
|
|
|
class TimestampMixin:
|
|
"""Adds created_at and updated_at columns to a model."""
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True),
|
|
server_default=func.now(),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True),
|
|
server_default=func.now(),
|
|
onupdate=func.now(),
|
|
nullable=False,
|
|
)
|
|
|
|
|
|
class SoftDeleteMixin:
|
|
"""Adds deleted_at column for soft-delete pattern."""
|
|
|
|
deleted_at: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True),
|
|
nullable=True,
|
|
default=None,
|
|
index=True,
|
|
)
|
|
|
|
|
|
class OrgScopedMixin:
|
|
"""Adds org_id foreign key. All queries must filter by org_id (OrgScopedQuery)."""
|
|
|
|
org_id: Mapped[int] = mapped_column(
|
|
ForeignKey("orgs.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
|
|
|
|
__all__ = ["Base", "OrgScopedMixin", "SoftDeleteMixin", "TimestampMixin"]
|