100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
from datetime import date, time
|
|
|
|
from sqlalchemy import (
|
|
Boolean,
|
|
Column,
|
|
Date,
|
|
Enum,
|
|
ForeignKey,
|
|
Integer,
|
|
Numeric,
|
|
String,
|
|
Time,
|
|
func,
|
|
)
|
|
from sqlalchemy.dialects.postgresql import ARRAY
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class Business(Base):
|
|
__tablename__ = "businesses"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String, nullable=False)
|
|
type = Column(String, nullable=False)
|
|
timezone = Column(String, nullable=False, default="UTC")
|
|
status = Column(
|
|
Enum("trial", "active", "suspended", name="business_status"),
|
|
nullable=False,
|
|
default="trial",
|
|
)
|
|
plan = Column(
|
|
Enum("free", "basic", "pro", name="business_plan"),
|
|
nullable=False,
|
|
default="free",
|
|
)
|
|
meta_business_id = Column(String, nullable=True)
|
|
whatsapp_phone_number_id = Column(String, nullable=True, unique=True)
|
|
whatsapp_access_token = Column(String, nullable=True)
|
|
created_at = Column(Date, server_default=func.current_date())
|
|
|
|
users = relationship("User", back_populates="business", cascade="all, delete-orphan")
|
|
config = relationship("BusinessConfig", back_populates="business", uselist=False)
|
|
reservations = relationship("Reservation", back_populates="business")
|
|
table_types = relationship("TableType", back_populates="business", cascade="all, delete-orphan")
|
|
services = relationship("Service", back_populates="business", cascade="all, delete-orphan")
|
|
|
|
|
|
class BusinessConfig(Base):
|
|
__tablename__ = "business_configs"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
business_id = Column(
|
|
Integer, ForeignKey("businesses.id", ondelete="CASCADE"), nullable=False, unique=True
|
|
)
|
|
open_days = Column(ARRAY(Integer), nullable=False, default=list)
|
|
open_time = Column(Time, nullable=False, default=time(9, 0))
|
|
close_time = Column(Time, nullable=False, default=time(18, 0))
|
|
slot_duration = Column(Integer, nullable=False, default=60)
|
|
max_per_slot = Column(Integer, nullable=False, default=1)
|
|
blocked_dates = Column(ARRAY(Date), nullable=False, default=list)
|
|
assistant_name = Column(String, nullable=False, default="Hermes")
|
|
tone = Column(
|
|
Enum("formal", "casual", name="assistant_tone"), nullable=False, default="formal"
|
|
)
|
|
welcome_message = Column(String, nullable=True)
|
|
|
|
business = relationship("Business", back_populates="config")
|
|
|
|
|
|
class TableType(Base):
|
|
__tablename__ = "table_types"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
business_id = Column(
|
|
Integer, ForeignKey("businesses.id", ondelete="CASCADE"), nullable=False, index=True
|
|
)
|
|
capacity = Column(Integer, nullable=False)
|
|
quantity = Column(Integer, nullable=False)
|
|
label = Column(String, nullable=True)
|
|
|
|
business = relationship("Business", back_populates="table_types")
|
|
|
|
|
|
class Service(Base):
|
|
__tablename__ = "services"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
business_id = Column(
|
|
Integer, ForeignKey("businesses.id", ondelete="CASCADE"), nullable=False, index=True
|
|
)
|
|
name = Column(String, nullable=False)
|
|
description = Column(String, nullable=True)
|
|
price = Column(Numeric(10, 2), nullable=True)
|
|
duration_minutes = Column(Integer, nullable=True)
|
|
is_active = Column(Boolean, nullable=False, default=True)
|
|
|
|
business = relationship("Business", back_populates="services")
|