mirrors/metube
1
0
Fork 0
mirror of https://github.com/alexta69/metube.git synced 2025-04-04 04:37:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

Added support for loading YTDL_OPTIONS from file.

Browse source
This commit is contained in:
Abdulmohsen 2023-08-13 19:14:58 +03:00 • committed by GitHub
parent 8c994bc45f
commit a1e143a5a6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

28
app/main.py
View file

@ -3,6 +3,7 @@
import os
import sys
import traceback
from aiohttp import web
import socketio
import logging
@ -12,7 +13,9 @@ import pathlib
from ytdl import DownloadQueueNotifier, DownloadQueue
log = logging.getLogger('main')
if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG)
class Config:
_DEFAULTS = {
'DOWNLOAD_DIR': '.',
@ -38,6 +41,7 @@ class Config:
def __init__(self):
for k, v in self._DEFAULTS.items():
setattr(self, k, os.environ[k] if k in os.environ else v)
for k, v in self.__dict__.items():
if v.startswith('%%'):
setattr(self, k, getattr(self, v[2:]))
@ -46,13 +50,24 @@ class Config:
log.error(f'Environment variable "{k}" is set to a non-boolean value "{v}"')
sys.exit(1)
setattr(self, k, v in ('true', 'True', 'on', '1'))
if not self.URL_PREFIX.endswith('/'):
self.URL_PREFIX += '/'
try:
self.YTDL_OPTIONS = json.loads(self.YTDL_OPTIONS)
if isinstance(self.YTDL_OPTIONS, str) and os.path.exists(self.YTDL_OPTIONS):
log.info(f"Loading yt-dlp custom options from {self.YTDL_OPTIONS}")
with open(self.YTDL_OPTIONS) as json_data:
self.YTDL_OPTIONS = json.load(json_data)
else:
log.info(f"Loading yt-dlp custom options from Environment variable.")
self.YTDL_OPTIONS = json.loads(self.YTDL_OPTIONS)
assert isinstance(self.YTDL_OPTIONS, dict)
except (json.decoder.JSONDecodeError, AssertionError):
log.error('YTDL_OPTIONS is invalid')
if len(self.YTDL_OPTIONS) != 0:
log.info(f"Using custom yt-dlp options:\n{json.dumps(self.YTDL_OPTIONS, indent=2, ensure_ascii=False)}")
except (json.decoder.JSONDecodeError, AssertionError) as e:
log.error(f"Unable to parse YTDL_OPTIONS value. {str(e)}")
sys.exit(1)
config = Config()
@ -140,13 +155,13 @@ def get_custom_dirs():
dirs = list(filter(None, map(convert, path.glob('**'))))
return dirs
download_dir = recursive_dirs(config.DOWNLOAD_DIR)
audio_download_dir = download_dir
if config.DOWNLOAD_DIR != config.AUDIO_DOWNLOAD_DIR:
audio_download_dir = recursive_dirs(config.AUDIO_DOWNLOAD_DIR)
return {
"download_dir": download_dir,
"audio_download_dir": audio_download_dir
@ -195,6 +210,5 @@ app.on_response_prepare.append(on_prepare)
if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG)
log.info(f"Listening on {config.HOST}:{config.PORT}")
web.run_app(app, host=config.HOST, port=config.PORT, reuse_port=True)