from enum import Enum as PyEnum

from sqlalchemy import (
    Boolean,
    Column,
    DateTime,
    Enum,
    Float,
    ForeignKey,
    Integer,
    String,
    func,
)
from sqlalchemy.dialects.postgresql import UUID
from pydantic import BaseModel
from ._base import Base


class RFPStatus(PyEnum):
    ACTIVE = "ACTIVE"
    IN_PROGRESS = "IN_PROGRESS"
    NOT_YET_STARTED = "NOT_YET_STARTED"
    CUSTOMER_HOLD = "CUSTOMER_HOLD"
    UNSOLVED_ISSUES = "UNSOLVED_ISSUES"
    AWARDED = "AWARDED"
    PROTESTED = "PROTESTED"


class RFP(Base):
    __tablename__ = "rfps"

    id = Column(UUID(as_uuid=True), primary_key=True, nullable=False)
    rfp_number = Column(String, nullable=False)
    name = Column(String, nullable=False)
    received_proposals = Column(Integer, nullable=False)
    evaluated_count = Column(Integer, nullable=False)
    awaiting_evaluation = Column(Integer, nullable=False)
    target_award_date = Column(DateTime, nullable=True)
    ko_name = Column(String, nullable=True)
    award_type = Column(String, nullable=True)
    google_drive_id = Column(String, nullable=True)
    status = Column(Enum(RFPStatus), nullable=False)
    comparative_weights_locked = Column(Boolean(), nullable=True, default=False)
    created_at = Column(DateTime, nullable=False, default=func.now())
    updated_at = Column(
        DateTime, nullable=False, default=func.now(), onupdate=func.now()
    )