63 行
2.0 KiB
Python
63 行
2.0 KiB
Python
from typing import TYPE_CHECKING, Optional, List, Union
|
|
|
|
if TYPE_CHECKING:
|
|
from yandex_music import Client, BlockEntity, PersonalPlaylistsData, PlayContextsData
|
|
|
|
from yandex_music import YandexMusicObject
|
|
|
|
|
|
class Block(YandexMusicObject):
|
|
def __init__(self,
|
|
id_: str,
|
|
type_: str,
|
|
type_for_from: str,
|
|
title: str,
|
|
entities: List['BlockEntity'],
|
|
description: Optional[str] = None,
|
|
data: Optional[Union['PersonalPlaylistsData', 'PlayContextsData']] = None,
|
|
client: Optional['Client'] = None,
|
|
**kwargs) -> None:
|
|
|
|
self.id = id_
|
|
self.type = type_
|
|
self.type_for_from = type_for_from
|
|
self.title = title
|
|
self.entities = entities
|
|
|
|
self.description = description
|
|
self.data = data
|
|
|
|
self.client = client
|
|
self._id_attrs = (self.id, self.type, self.type_for_from, self.title, self.entities)
|
|
|
|
def __getitem__(self, item: int) -> 'BlockEntity':
|
|
return self.entities[item]
|
|
|
|
@classmethod
|
|
def de_json(cls, data: dict, client: 'Client') -> Optional['Block']:
|
|
if not data:
|
|
return None
|
|
|
|
data = super(Block, cls).de_json(data, client)
|
|
from yandex_music import BlockEntity, PlayContextsData, PersonalPlaylistsData
|
|
data['entities'] = BlockEntity.de_list(data.get('entities'), client)
|
|
|
|
block_type = data.get('type_')
|
|
if block_type == 'personal-playlists':
|
|
data['data'] = PersonalPlaylistsData.de_json(data.get('data'), client)
|
|
elif block_type == 'play-contexts':
|
|
data['data'] = PlayContextsData.de_json(data.get('data'), client)
|
|
|
|
return cls(client=client, **data)
|
|
|
|
@classmethod
|
|
def de_list(cls, data: dict, client: 'Client') -> List['Block']:
|
|
if not data:
|
|
return []
|
|
|
|
blocks = list()
|
|
for block in data:
|
|
blocks.append(cls.de_json(block, client))
|
|
|
|
return blocks
|