feat(api): allow changing host and port via env vars

ILikeBananas · Dec 5, 2018 · 59bcfe6 · 59bcfe6
1 parent da0b64a
commit 59bcfe6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/youtube-dl-server.py b/youtube-dl-server.py
@@ -18,7 +18,9 @@
     'YDL_EXTRACT_AUDIO_QUALITY': '192',
     'YDL_RECODE_VIDEO_FORMAT': None,
     'YDL_OUTPUT_TEMPLATE': '/youtube-dl/%(title)s [%(id)s].%(ext)s',
-    'YDL_ARCHIVE_FILE': None
+    'YDL_ARCHIVE_FILE': None,
+    'YDL_SERVER_HOST': '0.0.0.0',
+    'YDL_SERVER_PORT': 8080,
 }
 
 
@@ -111,6 +113,8 @@ def download(url, request_options):
 
 print("Started download thread")
 
-app.run(host='0.0.0.0', port=8080, debug=True)
+app_vars = ChainMap(os.environ, app_defaults)
+
+app.run(host=app_vars['YDL_SERVER_HOST'], port=app_vars['YDL_SERVER_PORT'], debug=True)
 done = True
 dl_thread.join()