116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
from typing import List, Optional, Dict, Type
|
|
from enum import Enum
|
|
|
|
from ...utils.config import logging_settings
|
|
from ...objects import Source, DatabaseObject
|
|
from ..abstract import Page
|
|
from ...objects import (
|
|
Artist,
|
|
Source,
|
|
Song,
|
|
Album,
|
|
Label,
|
|
Target
|
|
)
|
|
from ._music_object_render import parse_run_list, parse_run_element
|
|
|
|
LOGGER = logging_settings["youtube_music_logger"]
|
|
|
|
|
|
def music_card_shelf_renderer(renderer: dict) -> List[DatabaseObject]:
|
|
results = parse_run_list(renderer.get("title", {}).get("runs", []))
|
|
|
|
for sub_renderer in renderer.get("contents", []):
|
|
results.extend(parse_renderer(sub_renderer))
|
|
return results
|
|
|
|
def music_responsive_list_item_flex_column_renderer(renderer: dict) -> List[DatabaseObject]:
|
|
return parse_run_list(renderer.get("text", {}).get("runs", []))
|
|
|
|
|
|
def music_responsive_list_item_renderer(renderer: dict) -> List[DatabaseObject]:
|
|
results = []
|
|
|
|
for i, column in enumerate(renderer.get("flexColumns", [])):
|
|
_r = parse_renderer(column)
|
|
if i == 0 and len(_r) == 0:
|
|
renderer["text"] = \
|
|
column.get("musicResponsiveListItemFlexColumnRenderer", {}).get("text", {}).get("runs", [{}])[0].get(
|
|
"text")
|
|
|
|
results.extend(_r)
|
|
|
|
_r = parse_run_element(renderer)
|
|
if _r is not None:
|
|
results.append(_r)
|
|
|
|
song_list: List[Song] = []
|
|
album_list: List[Album] = []
|
|
artist_list: List[Artist] = []
|
|
_map: Dict[Type[DatabaseObject], List[DatabaseObject]] = {Song: song_list, Album: album_list, Artist: artist_list}
|
|
|
|
for result in results:
|
|
_map[type(result)].append(result)
|
|
|
|
if len(song_list) == 1:
|
|
song = song_list[0]
|
|
song.feature_artist_collection.extend(artist_list)
|
|
song.album_collection.extend(album_list)
|
|
return [song]
|
|
|
|
if len(album_list) == 1:
|
|
album = album_list[0]
|
|
album.artist_collection.extend(artist_list)
|
|
album.song_collection.extend(song_list)
|
|
return [album]
|
|
|
|
"""
|
|
if len(artist_list) == 1:
|
|
artist = artist_list[0]
|
|
artist.main_album_collection.extend(album_list)
|
|
return [artist]
|
|
"""
|
|
|
|
return results
|
|
|
|
|
|
def music_shelf_renderer(renderer: dict) -> List[DatabaseObject]:
|
|
result = []
|
|
for subrenderer in renderer.get("contents"):
|
|
result.extend(parse_renderer(subrenderer))
|
|
|
|
return result
|
|
|
|
|
|
def music_carousel_shelf_renderer(renderer: dict):
|
|
return music_shelf_renderer(renderer=renderer)
|
|
|
|
|
|
def music_two_row_item_renderer(renderer: dict):
|
|
return parse_run_list(renderer.get("title", {}).get("runs", []))
|
|
|
|
|
|
RENDERER_PARSERS = {
|
|
"musicCardShelfRenderer": music_card_shelf_renderer,
|
|
"musicResponsiveListItemRenderer": music_responsive_list_item_renderer,
|
|
"musicResponsiveListItemFlexColumnRenderer": music_responsive_list_item_flex_column_renderer,
|
|
"musicShelfRenderer": music_card_shelf_renderer,
|
|
"musicCarouselShelfRenderer": music_carousel_shelf_renderer,
|
|
"musicTwoRowItemRenderer": music_two_row_item_renderer,
|
|
|
|
"itemSectionRenderer": lambda _: [],
|
|
}
|
|
|
|
|
|
def parse_renderer(renderer: dict) -> List[DatabaseObject]:
|
|
result: List[DatabaseObject] = []
|
|
|
|
for renderer_name, renderer in renderer.items():
|
|
if renderer_name not in RENDERER_PARSERS:
|
|
LOGGER.warning(f"Can't parse the renderer {renderer_name}.")
|
|
continue
|
|
|
|
result.extend(RENDERER_PARSERS[renderer_name](renderer))
|
|
|
|
return result
|