From c353ed5eb7e2e98bd4d68aedc99ab9f86be62ced Mon Sep 17 00:00:00 2001 From: "Alexander \"PapaTutuWawa" Date: Mon, 13 Jan 2025 22:02:19 +0000 Subject: [PATCH] Format and lint youtube.py --- scripts/youtube.py | 29 ++++++++++++++++++----------- 1 file changed, 18 insertions(+), 11 deletions(-) diff --git a/scripts/youtube.py b/scripts/youtube.py index 43cf93b..6784ea0 100644 --- a/scripts/youtube.py +++ b/scripts/youtube.py @@ -7,10 +7,8 @@ from microkodi.helpers import recursive_dict_merge import yt_dlp -DEFAULT_CONFIG = { - "format": "bestvideo[width<=1920]+bestaudio", - "ytdlp_options": {} -} +DEFAULT_CONFIG = {"format": "bestvideo[width<=1920]+bestaudio", "ytdlp_options": {}} + def youtube_url_transformer(url: ParseResult) -> tuple[list[str], str]: logger = logging.getLogger("Youtube") @@ -24,21 +22,26 @@ def youtube_url_transformer(url: ParseResult) -> tuple[list[str], str]: with yt_dlp.YoutubeDL(opts) as ytdl: info = ytdl.extract_info(urlunparse(url), download=False) - user_agent = None + #user_agent = None audio_url = None video_url = None for format in info["requested_formats"]: if format["width"] is None: audio_url = format["url"] else: - user_agent = format["http_headers"]["User-Agent"] + #user_agent = format["http_headers"]["User-Agent"] video_url = format["url"] - args = [ - f'--input-slave={audio_url}', - ] if audio_url else None + args = ( + [ + f"--input-slave={audio_url}", + ] + if audio_url + else None + ) return args, urlparse(video_url) + def init(): # Create the config config: Config = I.get("Config") @@ -53,5 +56,9 @@ def init(): # Register the transformers I.get("VlcConfig").register_domain_transformer("youtu.be", youtube_url_transformer) - I.get("VlcConfig").register_domain_transformer("youtube.com", youtube_url_transformer) - I.get("VlcConfig").register_domain_transformer("www.youtube.com", youtube_url_transformer) \ No newline at end of file + I.get("VlcConfig").register_domain_transformer( + "youtube.com", youtube_url_transformer + ) + I.get("VlcConfig").register_domain_transformer( + "www.youtube.com", youtube_url_transformer + )