"""Track Snowflake Model."""

from typing import Optional

from sqlalchemy.orm import Mapped, mapped_column

from collaborator import config
from collaborator.connectors import snowflake


class Track(snowflake.BaseModel):
    """Track Model."""

    __tablename__ = "DIM_TRACK"

    if config.ENVIRONMENT != config.TEST_ENVIRONMENT:
        __table_args__ = {"schema": config.SNOWFLAKE_SCHEMAS["facts"]}

    id: Mapped[int] = mapped_column("trackid", primary_key=True)
    tuid: Mapped[int] = mapped_column("track_unique_id")
    upc: Mapped[Optional[int]]
    track_name: Mapped[Optional[str]] = mapped_column("TRACKNAME")
    isrc: Mapped[Optional[str]] = mapped_column("ISRC")