Add option to include cookiefile

This allows users to download age-restricted videos from the web interface.
manbearwiz · Jan 10, 2023 · bc44e54 · bc44e54
1 parent cdeaf3b
commit bc44e54
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -21,12 +21,15 @@ docker run -d --net="host" --name youtube-dl -v /home/core/youtube-dl:/youtube-d
 
 ### Docker Compose
 
-This is an example service definition that could be put in `docker-compose.yml`. This service uses a VPN client container for its networking.
+This is an example service definition that could be put in `docker-compose.yml`.
+This service uses a VPN client container for its networking and has a cookie file for downloading age-restricted videos.
 
 ```yml
   youtube-dl:
     image: "kmb32123/youtube-dl-server"
     network_mode: "service:vpn"
+    environment:
+      - YDL_COOKIE_FILE=youtube.com_cookies.txt
     volumes:
       - /home/core/youtube-dl:/youtube-dl
     restart: always

diff --git a/youtube-dl-server.py b/youtube-dl-server.py
@@ -111,14 +111,18 @@ def get_ydl_options(request_options):
             }
         )
 
-    return {
+    options = {
         "format": ydl_vars["YDL_FORMAT"],
         "postprocessors": postprocessors,
         "outtmpl": ydl_vars["YDL_OUTPUT_TEMPLATE"],
         "download_archive": ydl_vars["YDL_ARCHIVE_FILE"],
         "updatetime": ydl_vars["YDL_UPDATE_TIME"] == "True",
     }
 
+    if ydl_vars["YDL_COOKIE_FILE"]:
+        options["cookiefile"] = f"/youtube-dl/{ydl_vars['YDL_COOKIE_FILE']}"
+
+    return options
 
 def download(url, request_options):
     with YoutubeDL(get_ydl_options(request_options)) as ydl: