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

separated YTDL_OPTIONS_FILE

Browse source
This commit is contained in:
Alex 2023-08-19 21:01:45 +00:00
parent 207f8fe916
commit 27def1f670
2 changed files with 21 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

35
app/main.py
View file

@ -12,8 +12,6 @@ import pathlib
from ytdl import DownloadQueueNotifier, DownloadQueue
log = logging.getLogger('main')
if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG)
class Config:
_DEFAULTS = {
@ -29,6 +27,7 @@ class Config:
'OUTPUT_TEMPLATE': '%(title)s.%(ext)s',
'OUTPUT_TEMPLATE_CHAPTER': '%(title)s - %(section_number)s %(section_title)s.%(ext)s',
'YTDL_OPTIONS': '{}',
'YTDL_OPTIONS_FILE': '',
'HOST': '0.0.0.0',
'PORT': '8081',
'BASE_DIR': '',
@ -39,8 +38,6 @@ class Config:
def __init__(self):
for k, v in self._DEFAULTS.items():
if k in os.environ:
log.debug(f"ENV override for {k} = {os.environ[k]}")
setattr(self, k, os.environ[k] if k in os.environ else v)
for k, v in self.__dict__.items():
@ -56,21 +53,26 @@ class Config:
self.URL_PREFIX += '/'
try:
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)
self.YTDL_OPTIONS = json.loads(self.YTDL_OPTIONS)
assert isinstance(self.YTDL_OPTIONS, dict)
if len(self.YTDL_OPTIONS) != 0:
log.debug(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)}")
except (json.decoder.JSONDecodeError, AssertionError):
log.error('YTDL_OPTIONS is invalid')
sys.exit(1)
if self.YTDL_OPTIONS_FILE:
log.info(f'Loading yt-dlp custom options from "{self.YTDL_OPTIONS_FILE}"')
if not os.path.exists(self.YTDL_OPTIONS_FILE):
log.error(f'File "{self.YTDL_OPTIONS_FILE}" not found')
sys.exit(1)
try:
with open(self.YTDL_OPTIONS_FILE) as json_data:
opts = json.load(json_data)
assert isinstance(opts, dict)
except (json.decoder.JSONDecodeError, AssertionError):
log.error('YTDL_OPTIONS_FILE contents is invalid')
sys.exit(1)
self.YTDL_OPTIONS.update(opts)
config = Config()
class ObjectSerializer(json.JSONEncoder):
@ -211,5 +213,6 @@ 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)