from sqlalchemy import Column, String, Text
from sqlalchemy.dialects.postgresql import ARRAY
from app.api.db.base_class import Base
from sqlalchemy.sql import func
import uuid

class Drug(Base):
    __tablename__ = "drugs"

    id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
    name = Column(String, nullable=False, unique=True, index=True)
    description = Column(Text, nullable=True)
    drug_types = Column(ARRAY(String), nullable=False)

    created_at = Column(TIMESTAMP(timezone=True), server_default=func.now())
    updated_at = Column(TIMESTAMP(timezone=True), onupdate=func.now())