59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from sqlmodel import SQLModel, Field, Column, Relationship
|
|
from datetime import datetime
|
|
from typing import Optional, List, TYPE_CHECKING
|
|
from uuid import UUID, uuid4
|
|
from sqlalchemy import DateTime, func
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.coupon import Coupon
|
|
|
|
class Business(SQLModel, table=True):
|
|
"""Business record for local partners."""
|
|
|
|
__tablename__ = "businesses"
|
|
|
|
id: Optional[UUID] = Field(default_factory=uuid4, primary_key=True)
|
|
name: str
|
|
address: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
image_url: Optional[str] = None
|
|
social_media: Optional[str] = None
|
|
category: Optional[str] = None
|
|
latitude: Optional[float] = None
|
|
longitude: Optional[float] = None
|
|
area: Optional[str] = Field(default="Boquete")
|
|
|
|
# Relationship to coupons
|
|
coupons: List["Coupon"] = Relationship(back_populates="business")
|
|
|
|
created_at: Optional[datetime] = Field(
|
|
default=None,
|
|
sa_column=Column(DateTime(timezone=True), server_default=func.now())
|
|
)
|
|
updated_at: Optional[datetime] = Field(
|
|
default=None,
|
|
sa_column=Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
|
)
|
|
|
|
class BusinessCreate(SQLModel):
|
|
name: str
|
|
address: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
image_url: Optional[str] = None
|
|
social_media: Optional[str] = None
|
|
category: Optional[str] = None
|
|
latitude: Optional[float] = None
|
|
longitude: Optional[float] = None
|
|
area: Optional[str] = "Boquete"
|
|
|
|
class BusinessUpdate(SQLModel):
|
|
name: Optional[str] = None
|
|
address: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
image_url: Optional[str] = None
|
|
social_media: Optional[str] = None
|
|
category: Optional[str] = None
|
|
latitude: Optional[float] = None
|
|
longitude: Optional[float] = None
|
|
area: Optional[str] = None
|