forked from kenzuya/unshackle
36 lines
887 B
Python
36 lines
887 B
Python
"""Utility functions for parsing M3U8 playlists."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional, Union
|
|
|
|
import httpx
|
|
import m3u8
|
|
from curl_cffi.requests import Session as CurlSession
|
|
from requests import Session
|
|
|
|
from unshackle.core.manifests.hls import HLS
|
|
from unshackle.core.tracks import Tracks
|
|
|
|
|
|
def parse(
|
|
master: m3u8.M3U8,
|
|
language: str,
|
|
*,
|
|
session: Optional[Union[Session, CurlSession, httpx.Client]] = None,
|
|
) -> Tracks:
|
|
"""Parse a variant playlist to ``Tracks`` with basic information, defer DRM loading."""
|
|
tracks = HLS(master, session=session).to_tracks(language)
|
|
|
|
bool(master.session_keys or HLS.parse_session_data_keys(master, session or Session()))
|
|
|
|
if True:
|
|
for t in tracks.videos + tracks.audio:
|
|
t.needs_drm_loading = True
|
|
t.session = session
|
|
|
|
return tracks
|
|
|
|
|
|
__all__ = ["parse"]
|