from sqlalchemy import Column, Integer, String, Boolean, Date, Time, DateTime, Enum, ForeignKey, Text
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
from app.database import Base
import enum

class UnavailabilityType(str, enum.Enum):
    recurring = "recurring"
    date_range = "date_range"

class ApprovalStatus(str, enum.Enum):
    pending = "pending"
    approved = "approved"
    rejected = "rejected"

class Unavailability(Base):
    __tablename__ = "unavailabilities"
    id = Column(Integer, primary_key=True, index=True)
    asim_id = Column(Integer, ForeignKey("asim.id"), nullable=False)
    unavail_type = Column(Enum(UnavailabilityType), nullable=False)
    day_of_week = Column(Integer, nullable=True)
    mass_time = Column(Time, nullable=True)
    date_from = Column(Date, nullable=True)
    date_to = Column(Date, nullable=True)
    reason = Column(Text, nullable=True)
    status = Column(Enum(ApprovalStatus), default=ApprovalStatus.pending)
    is_active = Column(Boolean, default=True)
    approved_by = Column(Integer, ForeignKey("users.id"), nullable=True)
    approved_at = Column(DateTime(timezone=True), nullable=True)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    asim = relationship("Asim", back_populates="unavailabilities")