from datetime import datetime

from fansifter_common.utils import timezone
from fansifter_common.utils.uuid import uuid_string
from sqlalchemy.orm import Mapped, mapped_column

from email_campaigns.adapters.db import Model


class CustomList(Model, kw_only=True):
    __tablename__ = "custom_list"

    id: Mapped[str] = mapped_column(primary_key=True)
    name: Mapped[str]
    vendor_id: Mapped[int]
    subaccount_id: Mapped[int]


class ArtistRosterMainRep(Model, kw_only=True):
    __tablename__ = "artist_roster_main_rep"

    id: Mapped[str] = mapped_column(primary_key=True, default_factory=uuid_string)
    global_participant_id: Mapped[str]
    vendor_id: Mapped[int]
    subaccount_id: Mapped[int]
    created_at: Mapped[datetime] = mapped_column(default_factory=timezone.now)


class ArtistRosterLocalRep(Model, kw_only=True):
    __tablename__ = "artist_roster_local_rep"

    id: Mapped[str] = mapped_column(primary_key=True, default_factory=uuid_string)
    global_participant_id: Mapped[str]
    vendor_id: Mapped[int]
    subaccount_id: Mapped[int]
    country_code: Mapped[str]
    created_at: Mapped[datetime] = mapped_column(default_factory=timezone.now)