social-app/backend/src/models/schedule_subscriptions.py

from __future__ import annotations

import uuid

from sqlalchemy import Integer, String
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import Mapped, mapped_column

from core.db.base import Base, TimestampMixin
from schemas.enums import NotifyLevel, SubscriptionPermission, SubscriptionStatus

__all__ = [
    "ScheduleSubscription",
    "SubscriptionStatus",
    "NotifyLevel",
    "SubscriptionPermission",
]


class ScheduleSubscription(TimestampMixin, Base):
    __tablename__: str = "schedule_subscriptions"
    __table_args__ = {"extend_existing": True}

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    item_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True),
        nullable=False,
    )
    subscriber_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True),
        nullable=False,
    )
    permission: Mapped[int] = mapped_column(
        Integer,
        nullable=False,
        default=1,
    )
    notify_level: Mapped[NotifyLevel] = mapped_column(
        String(20),
        nullable=False,
        default=NotifyLevel.ALL,
    )
    status: Mapped[SubscriptionStatus] = mapped_column(
        String(20),
        nullable=False,
        default=SubscriptionStatus.ACTIVE,
    )
    created_by: Mapped[uuid.UUID | None] = mapped_column(
        UUID(as_uuid=True),
        nullable=True,
    )