# ================================================================
# services/data/app/schemas/sales.py
# ================================================================
"""Sales data schemas"""

from pydantic import BaseModel, validator
from datetime import datetime
from typing import Optional, List
import uuid

class SalesDataCreate(BaseModel):
    tenant_id: uuid.UUID
    date: datetime
    product_name: str
    quantity_sold: int
    revenue: Optional[float] = None
    location_id: Optional[str] = None
    source: str = "manual"
    raw_data: Optional[str] = None

class SalesDataResponse(BaseModel):
    id: uuid.UUID
    tenant_id: uuid.UUID
    date: datetime
    product_name: str
    quantity_sold: int
    revenue: Optional[float]
    location_id: Optional[str]
    source: str
    created_at: datetime
    updated_at: datetime
    
    class Config:
        from_attributes = True

class SalesDataImport(BaseModel):
    tenant_id: uuid.UUID
    data_format: str  # csv, excel, pos
    data: str  # Base64 encoded or JSON string
    
class SalesDataQuery(BaseModel):
    tenant_id: uuid.UUID
    start_date: datetime
    end_date: datetime
    product_name: Optional[str] = None
    location_id: Optional[str] = None