yandex-music-api/yandex_music/account/product.py

from dataclasses import field
from typing import TYPE_CHECKING, List, Optional

from yandex_music import YandexMusicModel
from yandex_music.utils import model

if TYPE_CHECKING:
    from yandex_music import ClientType, JSONType, Price


@model
class Product(YandexMusicModel):
    """Класс, представляющий продаваемый продукт.

    Attributes:
        product_id (:obj:`str`): Уникальный идентификатор.
        type (:obj:`str`): Тип продаваемого.
        duration (:obj:`int`): Длительность.
        trial_duration (:obj:`int`): Длительность испытательного срока.
        feature (:obj:`str`): Предоставляемая возможность.
        debug (:obj:`bool`): Отладочный продукт.
        plus (:obj:`bool`): Даёт ли подписку "Плюс".
        price (:obj:`yandex_music.Price`, optional): Цена.
        common_period_duration (:obj:`str`, optional): Длительность общего периода.
        cheapest (:obj:`bool`, optional): Самый дешёвый (лучшее предложение).
        title (:obj:`str`, optional): Заголовок продукта.
        family_sub (:obj:`bool`, optional): Семейная ли подписка.
        fb_image (:obj:`str`, optional): Картинка для превью на facebook.
        fb_name (:obj:`str`, optional): Заголовок превью на facebook.
        family (:obj:`bool`, optional): Доступно ли для семьи.
        features (:obj:`list` из :obj:`str`, optional): Список предоставляемых возможностей.
        description (:obj:`str`, optional): Описание.
        available (:obj:`bool`, optional): Доступна ли покупка.
        trial_available (:obj:`bool`, optional): Доступен ли пробный период.
        trial_period_duration (:obj:`str`, optional): Длительность пробного периода.
        intro_period_duration (:obj:`str`, optional): Длительность вступительного периода TODO.
        intro_price (:obj:`yandex_music.Price`, optional): Цена вступительного периода.
        start_period_duration (:obj:`str`, optional): Длительность первого срока (за меньшую цену).
        start_price (:obj:`yandex_music.Price`, optional): Цена за первый срок.
        licence_text_parts (:obj:`list` из :obj:`yandex_music.LicenceTextPart`, optional):
            Длительность пробного периода.
        vendor_trial_available (:obj:`bool`, optional): Доступен испытательный срок продавца TODO.
        button_text (:obj:`str`, optional): Текст кнопки.
        button_additional_text (:obj:`str`, optional): Дополнительный текст кнопки.
        payment_method_types (:obj:`list` из :obj:`str`, optional): Способы оплаты.
        client (:obj:`yandex_music.Client`, optional): Клиент Yandex Music.
    """

    product_id: str
    type: str
    duration: int
    trial_duration: int
    feature: str
    debug: bool
    plus: bool
    price: Optional['Price'] = None
    common_period_duration: Optional[str] = None
    cheapest: Optional[bool] = None
    title: Optional[str] = None
    family_sub: Optional[bool] = None
    fb_image: Optional[str] = None
    fb_name: Optional[str] = None
    family: Optional[bool] = None
    features: List[str] = None
    description: Optional[str] = None
    available: Optional[bool] = None
    trial_available: Optional[bool] = None
    trial_period_duration: Optional[str] = None
    intro_period_duration: Optional[str] = None
    intro_price: Optional['Price'] = None
    start_period_duration: Optional[str] = None
    start_price: Optional['Price'] = None
    licence_text_parts: List['Price'] = field(default_factory=list)
    vendor_trial_available: Optional[bool] = None
    button_text: Optional[str] = None
    button_additional_text: Optional[str] = None
    payment_method_types: List[str] = field(default_factory=list)
    client: Optional['ClientType'] = None

    def __post_init__(self) -> None:
        self._id_attrs = (
            self.product_id,
            self.type,
            self.duration,
            self.trial_duration,
            self.product_id,
            self.feature,
            self.debug,
        )

    @classmethod
    def de_json(cls, data: 'JSONType', client: 'ClientType') -> Optional['Product']:
        """Десериализация объекта.

        Args:
            data (:obj:`dict`): Поля и значения десериализуемого объекта.
            client (:obj:`yandex_music.Client`, optional): Клиент Yandex Music.

        Returns:
            :obj:`yandex_music.Product`: Продаваемый продукт.
        """
        if not cls.is_dict_model_data(data):
            return None

        cls_data = cls.cleanup_data(data, client)
        from yandex_music import LicenceTextPart, Price

        cls_data['price'] = Price.de_json(data.get('price'), client)
        cls_data['intro_price'] = Price.de_json(data.get('intro_price'), client)
        cls_data['start_price'] = Price.de_json(data.get('start_price'), client)
        cls_data['licence_text_parts'] = LicenceTextPart.de_list(data.get('licence_text_parts'), client)

        return cls(client=client, **cls_data)  # type: ignore