API and database models refactoring

models/api/auth.py

@@ -0,0 +1,62 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+
+
+# User Authentication Models
+class UserRegister(BaseModel):
+    email: EmailStr = Field(..., description="Valid email address")
+    username: str = Field(
+        ..., min_length=3, max_length=50, description="Username (3-50 characters)"
+    )
+    password: str = Field(
+        ..., min_length=6, description="Password (minimum 6 characters)"
+    )
+
+    @field_validator("username")
+    @classmethod
+    def validate_username(cls, v):
+        if not re.match(r"^[a-zA-Z0-9_]+$", v):
+            raise ValueError(
+                "Username must contain only letters, numbers, or underscores"
+            )
+        return v.lower().strip()
+
+    @field_validator("password")
+    @classmethod
+    def validate_password(cls, v):
+        if len(v) < 6:
+            raise ValueError("Password must be at least 6 characters long")
+        return v
+
+
+class UserLogin(BaseModel):
+    username: str = Field(..., description="Username")
+    password: str = Field(..., description="Password")
+
+    @field_validator("username")
+    @classmethod
+    def validate_username(cls, v):
+        return v.strip()
+
+
+class UserResponse(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    email: str
+    username: str
+    role: str
+    is_active: bool
+    last_login: Optional[datetime] = None
+    created_at: datetime
+    updated_at: datetime
+
+
+class LoginResponse(BaseModel):
+    access_token: str
+    token_type: str = "bearer"
+    expires_in: int
+    user: UserResponse

models/api/base.py

@@ -0,0 +1,16 @@
+from typing import List, TypeVar, Generic
+from pydantic import BaseModel
+
+T = TypeVar('T')
+
+class ListResponse(BaseModel, Generic[T]):
+    """Generic list response model"""
+    items: List[T]
+    total: int
+    skip: int
+    limit: int
+
+class StatusResponse(BaseModel):
+    """Generic status response"""
+    success: bool
+    message: str

models/api/marketplace.py

@@ -0,0 +1,65 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+
+
+# Marketplace Import Models
+class MarketplaceImportRequest(BaseModel):
+    url: str = Field(..., description="URL to CSV file from marketplace")
+    marketplace: str = Field(
+        default="Letzshop",
+        description="Name of the marketplace (e.g., Letzshop, Amazon, eBay)",
+    )
+    shop_code: str = Field(..., description="Shop code to associate products with")
+    batch_size: Optional[int] = Field(
+        1000, gt=0, le=10000, description="Batch size for processing"
+    )
+
+    @field_validator("url")
+    @classmethod
+    def validate_url(cls, v):
+        if not v.startswith(("http://", "https://")):
+            raise ValueError("URL must start with http:// or https://")
+        return v
+
+    @field_validator("marketplace")
+    @classmethod
+    def validate_marketplace(cls, v):
+        # You can add validation for supported marketplaces here
+        supported_marketplaces = [
+            "Letzshop",
+            "Amazon",
+            "eBay",
+            "Etsy",
+            "Shopify",
+            "Other",
+        ]
+        if v not in supported_marketplaces:
+            # For now, allow any marketplace but log it
+            pass
+        return v.strip()
+
+    @field_validator("shop_code")
+    @classmethod
+    def validate_shop_code(cls, v):
+        return v.upper().strip()
+
+
+class MarketplaceImportJobResponse(BaseModel):
+    job_id: int
+    status: str
+    marketplace: str
+    shop_id: int
+    shop_code: Optional[str] = None  # Will be populated from shop relationship
+    shop_name: str
+    message: Optional[str] = None
+    imported: Optional[int] = 0
+    updated: Optional[int] = 0
+    total_processed: Optional[int] = 0
+    error_count: Optional[int] = 0
+    error_message: Optional[str] = None
+    created_at: Optional[datetime] = None
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None

models/api/product.py

@@ -0,0 +1,83 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+from models.api.stock import StockSummaryResponse
+
+class ProductBase(BaseModel):
+    product_id: Optional[str] = None
+    title: Optional[str] = None
+    description: Optional[str] = None
+    link: Optional[str] = None
+    image_link: Optional[str] = None
+    availability: Optional[str] = None
+    price: Optional[str] = None
+    brand: Optional[str] = None
+    gtin: Optional[str] = None
+    mpn: Optional[str] = None
+    condition: Optional[str] = None
+    adult: Optional[str] = None
+    multipack: Optional[int] = None
+    is_bundle: Optional[str] = None
+    age_group: Optional[str] = None
+    color: Optional[str] = None
+    gender: Optional[str] = None
+    material: Optional[str] = None
+    pattern: Optional[str] = None
+    size: Optional[str] = None
+    size_type: Optional[str] = None
+    size_system: Optional[str] = None
+    item_group_id: Optional[str] = None
+    google_product_category: Optional[str] = None
+    product_type: Optional[str] = None
+    custom_label_0: Optional[str] = None
+    custom_label_1: Optional[str] = None
+    custom_label_2: Optional[str] = None
+    custom_label_3: Optional[str] = None
+    custom_label_4: Optional[str] = None
+    additional_image_link: Optional[str] = None
+    sale_price: Optional[str] = None
+    unit_pricing_measure: Optional[str] = None
+    unit_pricing_base_measure: Optional[str] = None
+    identifier_exists: Optional[str] = None
+    shipping: Optional[str] = None
+    currency: Optional[str] = None
+    # New marketplace fields
+    marketplace: Optional[str] = None
+    shop_name: Optional[str] = None
+
+
+class ProductCreate(ProductBase):
+    product_id: str = Field(..., min_length=1, description="Product ID is required")
+    title: str = Field(..., min_length=1, description="Title is required")
+
+    @field_validator("product_id", "title")
+    @classmethod
+    def validate_required_fields(cls, v):
+        if not v or not v.strip():
+            raise ValueError("Field cannot be empty")
+        return v.strip()
+
+
+class ProductUpdate(ProductBase):
+    pass
+
+
+class ProductResponse(ProductBase):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    created_at: datetime
+    updated_at: datetime
+
+class ProductListResponse(BaseModel):
+    products: List[ProductResponse]
+    total: int
+    skip: int
+    limit: int
+
+
+class ProductDetailResponse(BaseModel):
+    product: ProductResponse
+    stock_info: Optional[StockSummaryResponse] = None

models/api/shop.py

@@ -0,0 +1,125 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+from models.api.product import ProductResponse
+
+class ShopCreate(BaseModel):
+    shop_code: str = Field(
+        ...,
+        min_length=3,
+        max_length=50,
+        description="Unique shop code (e.g., TECHSTORE)",
+    )
+    shop_name: str = Field(
+        ..., min_length=1, max_length=200, description="Display name of the shop"
+    )
+    description: Optional[str] = Field(
+        None, max_length=2000, description="Shop description"
+    )
+    contact_email: Optional[str] = None
+    contact_phone: Optional[str] = None
+    website: Optional[str] = None
+    business_address: Optional[str] = None
+    tax_number: Optional[str] = None
+
+    @field_validator("shop_code")
+    def validate_shop_code(cls, v):
+        # Convert to uppercase and check format
+        v = v.upper().strip()
+        if not v.replace("_", "").replace("-", "").isalnum():
+            raise ValueError(
+                "Shop code must be alphanumeric (underscores and hyphens allowed)"
+            )
+        return v
+
+    @field_validator("contact_email")
+    def validate_contact_email(cls, v):
+        if v and ("@" not in v or "." not in v):
+            raise ValueError("Invalid email format")
+        return v.lower() if v else v
+
+
+class ShopUpdate(BaseModel):
+    shop_name: Optional[str] = Field(None, min_length=1, max_length=200)
+    description: Optional[str] = Field(None, max_length=2000)
+    contact_email: Optional[str] = None
+    contact_phone: Optional[str] = None
+    website: Optional[str] = None
+    business_address: Optional[str] = None
+    tax_number: Optional[str] = None
+
+    @field_validator("contact_email")
+    def validate_contact_email(cls, v):
+        if v and ("@" not in v or "." not in v):
+            raise ValueError("Invalid email format")
+        return v.lower() if v else v
+
+
+class ShopResponse(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    shop_code: str
+    shop_name: str
+    description: Optional[str]
+    owner_id: int
+    contact_email: Optional[str]
+    contact_phone: Optional[str]
+    website: Optional[str]
+    business_address: Optional[str]
+    tax_number: Optional[str]
+    is_active: bool
+    is_verified: bool
+    created_at: datetime
+    updated_at: datetime
+
+
+class ShopListResponse(BaseModel):
+    shops: List[ShopResponse]
+    total: int
+    skip: int
+    limit: int
+
+class ShopProductCreate(BaseModel):
+    product_id: str = Field(..., description="Product ID to add to shop")
+    shop_product_id: Optional[str] = Field(
+        None, description="Shop's internal product ID"
+    )
+    shop_price: Optional[float] = Field(
+        None, ge=0, description="Shop-specific price override"
+    )
+    shop_sale_price: Optional[float] = Field(
+        None, ge=0, description="Shop-specific sale price"
+    )
+    shop_currency: Optional[str] = Field(None, description="Shop-specific currency")
+    shop_availability: Optional[str] = Field(
+        None, description="Shop-specific availability"
+    )
+    shop_condition: Optional[str] = Field(None, description="Shop-specific condition")
+    is_featured: bool = Field(False, description="Featured product flag")
+    min_quantity: int = Field(1, ge=1, description="Minimum order quantity")
+    max_quantity: Optional[int] = Field(
+        None, ge=1, description="Maximum order quantity"
+    )
+
+
+class ShopProductResponse(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    shop_id: int
+    product: ProductResponse
+    shop_product_id: Optional[str]
+    shop_price: Optional[float]
+    shop_sale_price: Optional[float]
+    shop_currency: Optional[str]
+    shop_availability: Optional[str]
+    shop_condition: Optional[str]
+    is_featured: bool
+    is_active: bool
+    min_quantity: int
+    max_quantity: Optional[int]
+    created_at: datetime
+    updated_at: datetime

models/api/stats.py

@@ -0,0 +1,22 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+
+
+class StatsResponse(BaseModel):
+    total_products: int
+    unique_brands: int
+    unique_categories: int
+    unique_marketplaces: int = 0
+    unique_shops: int = 0
+    total_stock_entries: int = 0
+    total_inventory_quantity: int = 0
+
+
+class MarketplaceStatsResponse(BaseModel):
+    marketplace: str
+    total_products: int
+    unique_shops: int
+    unique_brands: int

models/api/stock.py

@@ -0,0 +1,46 @@
+import re
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, EmailStr, Field, field_validator
+
+
+# Stock Models
+class StockBase(BaseModel):
+    gtin: str = Field(..., min_length=1, description="GTIN is required")
+    location: str = Field(..., min_length=1, description="Location is required")
+
+
+class StockCreate(StockBase):
+    quantity: int = Field(ge=0, description="Quantity must be non-negative")
+
+
+class StockAdd(StockBase):
+    quantity: int = Field(gt=0, description="Quantity to add must be positive")
+
+
+class StockUpdate(BaseModel):
+    quantity: int = Field(ge=0, description="Quantity must be non-negative")
+
+
+class StockResponse(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    gtin: str
+    location: str
+    quantity: int
+    created_at: datetime
+    updated_at: datetime
+
+
+class StockLocationResponse(BaseModel):
+    location: str
+    quantity: int
+
+
+class StockSummaryResponse(BaseModel):
+    gtin: str
+    total_quantity: int
+    locations: List[StockLocationResponse]
+    product_title: Optional[str] = None