Fix getting info with cookies

Merge pull request #328 from Bartkk0/fix-age-restriction
azlux · Apr 3, 2022 · 0650e72 · 0650e72
2 parents 16f18a5 + aa6632b
commit 0650e72
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/media/url.py b/media/url.py
@@ -129,6 +129,15 @@ def _get_info_from_url(self):
         ydl_opts = {
             'noplaylist': True
         }
+
+        cookie = var.config.get('youtube_dl', 'cookie_file')
+        if cookie:
+            ydl_opts['cookiefile'] = var.config.get('youtube_dl', 'cookie_file')
+
+        user_agent = var.config.get('youtube_dl', 'user_agent')
+        if user_agent:
+            youtube_dl.utils.std_headers['User-Agent'] = var.config.get('youtube_dl', 'user_agent')\
+
         succeed = False
         with youtube_dl.YoutubeDL(ydl_opts) as ydl:
             attempts = var.config.getint('bot', 'download_attempts')