from datetime import datetime, timezone
import uuid
from sqlalchemy import Column, String, JSON, DateTime, UniqueConstraint
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class ChannelDB(Base):
    """SQLAlchemy model for IPTV channels"""
    __tablename__ = "channels"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    tvg_id = Column(String, nullable=False)
    name = Column(String, nullable=False)
    group_title = Column(String, nullable=False)
    tvg_name = Column(String)

    __table_args__ = (
        UniqueConstraint('group_title', 'name', name='uix_group_title_name'),
    )
    tvg_logo = Column(String)
    urls = Column(JSON)  # Stores list of URLs as JSON
    created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
    updated_at = Column(DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))