diff --git a/models/drug.py b/models/drug.py new file mode 100644 index 0000000..dcddf05 --- /dev/null +++ b/models/drug.py @@ -0,0 +1,16 @@ +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()) \ No newline at end of file