f213/education-backend

from dataclasses import dataclass
from typing import Annotated, NotRequired, TypedDict

from apps.amocrm.client import http
from apps.products.models import Course


class ProductFieldValue(TypedDict):
    value: str  # any product's value stores as string in amo


class ProductField(TypedDict):
    field_id: int
    values: Annotated[list[ProductFieldValue], 1]  # we don't need to use more than 1 value per field


class Product(TypedDict):
    id: NotRequired[int]
    name: str
    custom_fields_values: list[ProductField]


@dataclass
class AmoCRMProductDTO:
    course: Course

    def create(self) -> int:
        """Create product (the element of catalog in amocrm's terms) in amocrm and returns its amocrm_id"""

        response = http.post(
            url=f"/api/v4/catalogs/{self._get_catalog_id()}/elements",
            data=[self._get_course_as_product()],
        )
        return response["_embedded"]["elements"][0]["id"]

    def update(self) -> None:
        """Update product in amocrm for given course"""
        data = self._get_course_as_product()
        data.update({"id": self.course.amocrm_course.amocrm_id})

        http.patch(
            url=f"/api/v4/catalogs/{self._get_catalog_id()}/elements",
            data=[data],
        )

    def _get_course_as_product(self) -> Product:
        from apps.amocrm.ids import product_field_id

        price = ProductField(
            field_id=product_field_id(field_code="PRICE"),
            values=[ProductFieldValue(value=str(self.course.price))],
        )
        slug = ProductField(
            field_id=product_field_id(field_code="SKU"),
            values=[ProductFieldValue(value=self.course.slug)],
        )
        unit = ProductField(
            field_id=product_field_id(field_code="UNIT"),
            values=[ProductFieldValue(value="шт")],  # 'piece' as unit type suits best for courses
        )
        fields_values = [price, slug, unit]

        if self.course.group is not None:
            fields_values.append(
                ProductField(
                    field_id=product_field_id(field_code="GROUP"),
                    values=[ProductFieldValue(value=self.course.group.slug)],
                )
            )
        return Product(
            name=self.course.name,
            custom_fields_values=fields_values,
        )

    @staticmethod
    def _get_catalog_id() -> int:
        from apps.amocrm.ids import products_catalog_id

        return products_catalog_id()