musicdlp/backend/ydl_wrap.py

import asyncio
from typing import Iterable

from yt_dlp import YoutubeDL


async def get_playlist_items(ydl: YoutubeDL, url: str) -> list[str]:

    return await asyncio.get_event_loop().run_in_executor(
        None,
        _target_get_playlist_items,
        ydl,
        url,
    )


def _target_get_playlist_items(ydl: YoutubeDL, url: str) -> list[str]:

    info = ydl.extract_info(url, download=False, process=True)
    if info is None:
        raise RuntimeError('ydl.extract_info returned None')
    return [
        entry['track'] if 'track' in entry else entry['title']
        for entry in info['entries']
    ]


async def download(
        ydl: YoutubeDL,
        url: str,
        playlist_items: Iterable[int] | None = None) -> int:

    return await asyncio.get_event_loop().run_in_executor(
        None,
        _target_download,
        ydl,
        url,
        playlist_items,
    )


def _target_download(
        ydl: YoutubeDL,
        url: str,
        playlist_items: Iterable[int] | None = None) -> int:

    if playlist_items:
        ydl.params['playlist_items'] = ','.join(str(i) for i in playlist_items)

    ret = ydl.download(url)
    del ydl.params['playlist_items']

    return ret