139 lines
4.2 KiB
Python
139 lines
4.2 KiB
Python
from enum import Enum
|
|
from typing import Annotated, Any, Dict, List, Optional
|
|
from uuid import uuid4
|
|
from pydantic import BaseModel, Field, validator
|
|
from pydantic.deprecated.class_validators import root_validator
|
|
|
|
|
|
def to_camel_case(snake_str: str) -> str:
|
|
components = snake_str.split("_")
|
|
return components[0] + "".join(x.title() for x in components[1:])
|
|
|
|
|
|
class ColumnItemType(str, Enum):
|
|
PRODUCT = "PRODUCT"
|
|
COMPONENT = "COMPONENT"
|
|
|
|
|
|
QuantityInt = Annotated[int, Field(ge=0, le=2147483647)]
|
|
StrictSmallInt = Annotated[int, Field(ge=0, le=32767)]
|
|
|
|
|
|
class CorrelationId(BaseModel):
|
|
correlation_id: str = Field(default_factory=lambda: uuid4().hex)
|
|
|
|
|
|
class ColumnsInput(BaseModel):
|
|
column_number: StrictSmallInt = Field(
|
|
alias="columnNumber", description="View index in Televend"
|
|
)
|
|
external_product_id: Optional[str] = Field(
|
|
default=None,
|
|
alias="externalProductId",
|
|
description="Product or Component external ID used for product/component identification in external partner's ERP system",
|
|
min_length=1,
|
|
max_length=32,
|
|
)
|
|
old_qty: Optional[QuantityInt] = Field(
|
|
default_factory=lambda: None,
|
|
alias="oldQty",
|
|
)
|
|
new_qty: Optional[QuantityInt] = Field(
|
|
default_factory=lambda: None,
|
|
alias="newQty",
|
|
)
|
|
|
|
old_price: Optional[float] = Field(
|
|
default_factory=lambda: None,
|
|
alias="oldPrc",
|
|
ge=0,
|
|
le=99999999.99,
|
|
)
|
|
new_price: Optional[float] = Field(
|
|
default_factory=lambda: None,
|
|
alias="newPrc",
|
|
ge=0,
|
|
le=99999999.99,
|
|
)
|
|
|
|
select_map: Optional[List[StrictSmallInt]] = Field(
|
|
default_factory=lambda: None, alias="selectMap"
|
|
)
|
|
|
|
item_type: Optional[ColumnItemType] = Field(
|
|
default_factory=lambda: ColumnItemType.PRODUCT,
|
|
alias="itemType",
|
|
description="MUST be set if item is COMPONENT",
|
|
)
|
|
|
|
class Config:
|
|
populate_by_name = True
|
|
alias_generator = to_camel_case
|
|
str_strip_whitespace = True
|
|
|
|
# @root_validator
|
|
# def check_required_fields(cls, values):
|
|
# if values.get("external_product_id"):
|
|
# if values.get("old_qty") is None:
|
|
# raise ValueError(
|
|
# f"provide oldQty for product {values['external_product_id']}"
|
|
# )
|
|
# if values.get("new_qty") is None:
|
|
# raise ValueError(
|
|
# f"provide newQty for product {values['external_product_id']}"
|
|
# )
|
|
# if values.get("old_price") is None:
|
|
# raise ValueError(
|
|
# f"provide oldPrc for product {values['external_product_id']}"
|
|
# )
|
|
# if values.get("new_price") is None:
|
|
# raise ValueError(
|
|
# f"provide newPrc for product {values['external_product_id']}"
|
|
# )
|
|
# return values
|
|
|
|
@validator("item_type")
|
|
def set_item_type(cls, value):
|
|
return value or ColumnItemType.PRODUCT
|
|
|
|
|
|
class PlanogramInput(CorrelationId, BaseModel):
|
|
machine_external_id: str = Field(
|
|
alias="machineExternalId",
|
|
description="Machine external ID",
|
|
min_length=1,
|
|
max_length=32,
|
|
)
|
|
columns: List[ColumnsInput] = Field(
|
|
title="Columns", description="A list of columns this planogram specifies"
|
|
)
|
|
|
|
class Config:
|
|
title = "Planogram"
|
|
alias_generator = to_camel_case
|
|
populate_by_name = True
|
|
str_strip_whitespace = True
|
|
|
|
|
|
class PlanogramsBulkInputPayload(BaseModel):
|
|
planograms: List[PlanogramInput] = Field(
|
|
title="Planograms",
|
|
description="A list of Planograms this bulk carries",
|
|
)
|
|
|
|
class Config:
|
|
populate_by_name = True
|
|
alias_generator = to_camel_case
|
|
|
|
# @root_validator
|
|
# def check_machine_unique(cls, values):
|
|
# planograms = values.get("planograms", [])
|
|
# external_ids = set()
|
|
# for planogram in planograms:
|
|
# if planogram.machine_external_id in external_ids:
|
|
# raise ValueError(
|
|
# f"Machine externalId must be unique! Duplicate {planogram.machine_external_id}"
|
|
# )
|
|
# external_ids.add(planogram.machine_external_id)
|
|
# return values
|