yandex-music-api/yandex_music/artist/brief_info.py

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

from yandex_music import YandexMusicModel
from yandex_music.utils import model

if TYPE_CHECKING:
    from yandex_music import (
        Album,
        Artist,
        Chart,
        ClientType,
        Cover,
        JSONType,
        Playlist,
        PlaylistId,
        Stats,
        Track,
        Video,
        Vinyl,
    )


@model
class BriefInfo(YandexMusicModel):
    """Класс, представляющий информацию об артисте.

    Attributes:
        artist (:obj:`yandex_music.Artist` | :obj:`None`): Артист.
        albums (:obj:`list` из :obj:`yandex_music.Album`): Альбомы.
        playlists (:obj:`list` из :obj:`yandex_music.Playlist`): Плейлисты.
        also_albums (:obj:`list` из :obj:`yandex_music.Album`): Сборники.
        last_release_ids (:obj:`list` из :obj:`int`): Уникальные идентификаторы последних выпущенных альбомов.
        last_releases (:obj:`list` из :obj:`yandex_music.Album`, optional): Последние выпущенные альбомы.
        popular_tracks (:obj:`list` из :obj:`yandex_music.Track`): Популярные треки.
        similar_artists (:obj:`list` из :obj:`yandex_music.Artist`): Похожие артисты.
        all_covers (:obj:`list` из :obj:`yandex_music.Cover`): Все обложки.
        concerts (:obj:`str`): Концерты (тест-кейс с ними потерялся, мало у кого есть).
        videos (:obj:`list` из :obj:`yandex_music.Video`): Видео.
        vinyls (:obj:`list` из :obj:`yandex_music.Vinyl`): Пластинки.
        has_promotions (:obj:`bool`): Рекламируется ли TODO.
        playlist_ids (:obj:`list` из :obj:`yandex_music.PlaylistId`): Уникальные идентификаторы плейлистов.
        stats (:obj:`yandex_music.Stats`, optional): Статистика прослушиваний за месяц.
        tracks_in_chart (:obj:`list` из :obj:`yandex_music.Chart`, optional): Треки в чарте.
        client (:obj:`yandex_music.Client`, optional): Клиент Yandex Music.
    """

    artist: Optional['Artist']
    albums: List['Album']
    playlists: List['Playlist']
    also_albums: List['Album']
    last_release_ids: List[int]
    last_releases: List['Album']
    popular_tracks: List['Track']
    similar_artists: List['Artist']
    all_covers: List['Cover']
    concerts: Any
    videos: List['Video']
    vinyls: List['Vinyl']
    has_promotions: bool
    playlist_ids: List['PlaylistId']
    stats: Optional['Stats'] = None
    tracks_in_chart: List['Chart'] = field(default_factory=list)
    client: Optional['ClientType'] = None

    def __post_init__(self) -> None:
        self._id_attrs = (
            self.artist,
            self.albums,
            self.playlists,
            self.also_albums,
            self.last_release_ids,
            self.popular_tracks,
            self.similar_artists,
            self.all_covers,
            self.concerts,
            self.videos,
            self.vinyls,
            self.has_promotions,
            self.playlist_ids,
        )

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

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

        Returns:
            :obj:`yandex_music.BriefInfo`: Информация об артисте.
        """
        if not cls.is_dict_model_data(data):
            return None

        cls_data = cls.cleanup_data(data, client)
        from yandex_music import Album, Artist, Chart, Cover, Playlist, PlaylistId, Stats, Track, Video, Vinyl

        cls_data['playlists'] = Playlist.de_list(data.get('playlists'), client)
        cls_data['artist'] = Artist.de_json(data.get('artist'), client)
        cls_data['similar_artists'] = Artist.de_list(data.get('similar_artists'), client)
        cls_data['popular_tracks'] = Track.de_list(data.get('popular_tracks'), client)
        cls_data['albums'] = Album.de_list(data.get('albums'), client)
        cls_data['also_albums'] = Album.de_list(data.get('also_albums'), client)
        cls_data['last_releases'] = Album.de_list(data.get('last_releases'), client)
        cls_data['all_covers'] = Cover.de_list(data.get('all_covers'), client)
        cls_data['playlist_ids'] = PlaylistId.de_list(data.get('playlist_ids'), client)
        cls_data['videos'] = Video.de_list(data.get('videos'), client)
        cls_data['tracks_in_chart'] = Chart.de_list(data.get('tracks_in_chart'), client)
        cls_data['vinyls'] = Vinyl.de_list(data.get('vinyls'), client)
        cls_data['stats'] = Stats.de_json(data.get('stats'), client)

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