diff --git a/models/food.py b/models/food.py index 1d6ec96..4d215b0 100644 --- a/models/food.py +++ b/models/food.py @@ -1,24 +1,21 @@ -from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, Float, Text, Enum +from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, Float, Text, JSON, Date, Time, Enum from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.orm import relationship from sqlalchemy.sql import func from core.database import Base import uuid -import enum - -class FoodCategory(enum.Enum): - VEGAN = "vegan" - VEGETARIAN = "vegetarian" - NON_VEGETARIAN = "non-vegetarian" +from datetime import datetime class Food(Base): __tablename__ = "foods" id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) - name = Column(String, nullable=False, unique=True) + name = Column(String, nullable=False) description = Column(Text, nullable=True) price = Column(Float, nullable=False) - category = Column(Enum(FoodCategory), nullable=False) is_available = Column(Boolean, default=True) + category = Column(String, nullable=True) + cuisine = Column(String, nullable=True) + allergens = Column(JSON, nullable=True) created_at = Column(DateTime, default=func.now()) updated_at = Column(DateTime, default=func.now(), onupdate=func.now()) \ No newline at end of file