rename module pyrohlik to rohlik

rohlik/models/product.py

@@ -0,0 +1,135 @@
+from decimal import Decimal
+from datetime import datetime
+from typing import List, Optional
+from pydantic import BaseModel, Field
+
+
+class Price(BaseModel):
+    amount: Decimal
+    currency: str    
+
+
+class Badge(BaseModel):
+    type: str
+    title: str
+    subtitle: Optional[str]
+    tooltip: Optional[str]
+
+
+class Country(BaseModel):
+    name: str
+    name_id: str = Field(..., alias='nameId')
+    code: str
+
+
+class SalePrice(BaseModel):
+    id: int
+    type: str
+    trigger_amount: int = Field(..., alias='triggerAmount')
+    price: Price
+    price_per_unit: Price = Field(..., alias='pricePerUnit')
+    badges: List[Badge]
+    valid_till: str = Field(..., alias='validTill')
+    active: bool    
+
+class ProductPrice(BaseModel):    
+    price: Price
+    price_per_unit: Price = Field(..., alias='pricePerUnit')
+    sales: List[SalePrice]    
+
+
+class SaleStock(BaseModel):
+    id: int
+    amount: int
+    unlimited_amount: bool = Field(..., alias='unlimitedAmount')
+    shelf_life: Optional[str] = Field(..., alias='shelfLife')    
+
+
+class ShelfLife(BaseModel):
+    average: int
+    minimal: int
+    type: str
+    bestBefore: Optional[str]
+
+
+class Stock(BaseModel):    
+    warehouse_id: int = Field(..., alias='warehouseId')
+    unavailability_reason: Optional[str] = Field(..., alias='unavailabilityReason')
+    preorder_enabled: bool = Field(..., alias='preorderEnabled')
+    max_basket_amount: int = Field(..., alias='maxBasketAmount')
+    max_basket_amount_reason: str = Field(..., alias='maxBasketAmountReason')
+    delivery_restriction: Optional[str] = Field(..., alias='deliveryRestriction')
+    expected_replenishment: Optional[datetime] = Field(..., alias='expectedReplenishment')
+    availability_dimension: int = Field(..., alias='availabilityDimension')
+    shelf_life: Optional[ShelfLife] = Field(..., alias='shelfLife')
+    billable_packaging: Optional[str] = Field(..., alias='billablePackaging')
+    sales: List[SaleStock]
+    in_stock: bool = Field(..., alias='inStock')
+
+
+class Ingredient(BaseModel):
+    type: str
+    ingredients: List['Ingredient']
+    title: str
+    code: Optional[str] = None
+    link: Optional[str] = None
+
+
+class Allergens(BaseModel):
+    contained: List[str]
+    possibly_contained: List[str] = Field(..., alias='possiblyContained')
+
+
+class AmountUnit(BaseModel):
+    amount: Optional[Decimal]
+    unit: str
+
+
+class NutritionaValues(BaseModel):
+    energy_kj: AmountUnit = Field(..., alias='energyKJ')
+    energy_k_cal: AmountUnit = Field(..., alias='energyKCal')
+    fats: AmountUnit
+    saturated_fats: AmountUnit = Field(..., alias='saturatedFats')
+    carbohydrates: AmountUnit
+    sugars: AmountUnit
+    protein: AmountUnit
+    salt: AmountUnit
+    fiber: AmountUnit
+
+
+class NutritionalValue(BaseModel):
+    portion: str
+    values: NutritionaValues
+
+
+class Composition(BaseModel):
+    nutritional_values: List[NutritionalValue] = Field(..., alias='nutritionalValues')
+    ingredients: List[Ingredient]
+    plain_ingredients: Optional[str] = Field(..., alias='plainIngredients')
+    allergens: Optional[Allergens]
+
+
+class Product(BaseModel):
+    id: int
+    name: str
+    slug: str
+    main_category_id: int = Field(..., alias='mainCategoryId')
+    unit: str
+    textual_amount: str = Field(..., alias='textualAmount')
+    badges: List[Badge]
+    archived: bool
+    premium_only: bool = Field(..., alias='premiumOnly')
+    brand: Optional[str]
+    images: List[str]
+    countries: List[Country]
+    can_be_favorite: bool = Field(..., alias='canBeFavorite')
+    information: List
+    weighted_item: bool = Field(..., alias='weightedItem')
+    package_ratio: int = Field(..., alias='packageRatio')
+    seller_id: int = Field(..., alias='sellerId')
+    flag: Optional[str]
+    attachments: List    
+    prices: ProductPrice
+    stock: Stock
+    description: str
+    composition: Composition