From 952650a344c9b690bb4c9322837dde553bd94f77 Mon Sep 17 00:00:00 2001 From: Aevann Date: Tue, 5 Mar 2024 05:02:39 +0200 Subject: [PATCH] dont use subprocess --- files/routes/admin.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/files/routes/admin.py b/files/routes/admin.py index 1cda6cd71..569fb0044 100644 --- a/files/routes/admin.py +++ b/files/routes/admin.py @@ -4,6 +4,7 @@ import os import ffmpeg import random import isodate +import yt_dlp from sqlalchemy.orm import load_only @@ -1956,9 +1957,18 @@ def schedule_orgy(v): duration = isodate.parse_duration(duration).total_seconds() end_utc = int(start_utc + duration) orgy_type = 'file' - params = ["yt-dlp", "--get-url", f"https://www.youtube.com/watch?v={data}", "-f", "b", "--proxy", PROXY_URL] - data = subprocess.check_output(params, timeout=30) - data = data.decode("utf-8") + + ydl_opts = { + "quiet": True, + "simulate": True, + "forceurl": True, + 'format': 'b', + 'proxy': PROXY_URL + } + + with yt_dlp.YoutubeDL(ydl_opts) as ydl: + info = ydl.extract_info(f"https://www.youtube.com/watch?v={data}") + data = info["url"] elif rumble_regex.match(normalized_link): orgy_type = 'rumble' data = normalized_link