import uuid
from sqlalchemy import Column, SmallInteger, Boolean, TIMESTAMP, ForeignKey, UniqueConstraint, Index
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.sql import func
from app.db.engine import Base


class QuestionDeptMap(Base):
    __tablename__ = "question_dept_map"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    question_id = Column(UUID(as_uuid=True), ForeignKey("mstr_question.id", ondelete="CASCADE"), nullable=False)
    department_id = Column(UUID(as_uuid=True), ForeignKey("mstr_department.id", ondelete="CASCADE"), nullable=False)
    sequence_number = Column(SmallInteger, nullable=False)
    is_active = Column(Boolean, nullable=False, default=True)
    created_at = Column(TIMESTAMP(timezone=True), server_default=func.now(), nullable=False)

    __table_args__ = (
        UniqueConstraint("question_id", "department_id", name="uq_question_dept"),
        UniqueConstraint("department_id", "sequence_number", name="uq_dept_sequence"),
        Index("question_dept_map_dept_idx", "department_id", "is_active", "sequence_number"),
    )