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 SOW(Base):
    __tablename__ = "sows"
    id = Column(UUID(as_uuid=True), primary_key=True, nullable=False)
    # rfp_id = Column(
    #     UUID(as_uuid=True),
    #     ForeignKey("rfps.id", ondelete="CASCADE"),
    #     nullable=False,
    #     unique=True,
    # )
    requirement = Column(String, nullable=True)
    additional_info = Column(String, nullable=True)
    sow_generated_text = Column(String, nullable=True)
    created_at = Column(DateTime, nullable=False, default=func.now())
    updated_at = Column(
        DateTime, nullable=False, default=func.now(), onupdate=func.now()
    )