from __future__ import annotations

from typing import ClassVar, List, Optional

from common import ApiModel


class Product(ApiModel):
    KEYS: ClassVar[List[str]] = [
        "id",
        "product_id",
        "shop_code",
        "link",
        "name",
        "quantity",
        "unit",
        "img_small",
        "img_large",
    ]
    NON_INSERT_KEYS: ClassVar[List[str]] = ["id"]

    id: int = -1
    product_id: str
    shop_code: str
    link: str
    name: str
    quantity: int
    unit: str
    img_small: str
    img_large: str
    # Non-persisted field used in tests and insert helper
    raw_data: Optional[dict] = None