mirror of
https://github.com/chibicitiberiu/ytsm.git
synced 2024-02-24 05:43:31 +00:00
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
import logging
|
|
import os
|
|
import os.path
|
|
from shutil import copyfile
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
|
|
from .models import UserSettings
|
|
from .utils.customconfigparser import ConfigParserWithEnv
|
|
|
|
|
|
class AppConfig(object):
|
|
__SETTINGS_FILE = 'config.ini'
|
|
__LOG_FILE = 'log.log'
|
|
|
|
DEFAULT_SETTINGS = {
|
|
'global': {
|
|
'YouTubeApiKey': 'AIzaSyBabzE4Bup77WexdLMa9rN9z-wJidEfNX8',
|
|
'SynchronizationSchedule': '0 * * * * *',
|
|
'SchedulerConcurrency': '2',
|
|
},
|
|
'user': {
|
|
'MarkDeletedAsWatched': 'True',
|
|
'DeleteWatched': 'True',
|
|
'AutoDownload': 'True',
|
|
'DownloadMaxAttempts': '3',
|
|
'DownloadGlobalLimit': '',
|
|
'DownloadSubscriptionLimit': '5',
|
|
'DownloadOrder': 'playlist_index',
|
|
'DownloadPath': '${env:USERPROFILE}${env:HOME}/Downloads',
|
|
'DownloadFilePattern': '${channel}/${playlist}/S01E${playlist_index} - ${title} [${id}]',
|
|
'DownloadFormat': 'bestvideo+bestaudio',
|
|
'DownloadSubtitles': 'True',
|
|
'DownloadAutogeneratedSubtitles': 'False',
|
|
'DownloadSubtitlesAll': 'False',
|
|
'DownloadSubtitlesLangs': 'en,ro',
|
|
'DownloadSubtitlesFormat': '',
|
|
}
|
|
}
|
|
|
|
def __init__(self):
|
|
self.log_path = os.path.join(settings.BASE_DIR, 'config', AppConfig.__LOG_FILE)
|
|
self.settings_path = os.path.join(settings.BASE_DIR, 'config', AppConfig.__SETTINGS_FILE)
|
|
|
|
self.settings = ConfigParserWithEnv(defaults=AppConfig.DEFAULT_SETTINGS, allow_no_value=True)
|
|
self.load_settings()
|
|
|
|
def load_settings(self):
|
|
if os.path.exists(self.settings_path):
|
|
with open(self.settings_path, 'r') as f:
|
|
self.settings.read_file(f)
|
|
|
|
def save_settings(self):
|
|
if os.path.exists(self.settings_path):
|
|
# Create a backup
|
|
copyfile(self.settings_path, self.settings_path + ".backup")
|
|
else:
|
|
# Ensure directory exists
|
|
settings_dir = os.path.dirname(self.settings_path)
|
|
os.makedirs(settings_dir, exist_ok=True)
|
|
|
|
with open(self.settings_path, 'w') as f:
|
|
self.settings.write(f)
|
|
|
|
def get_user_config(self, user: User) -> ConfigParserWithEnv:
|
|
user_settings = UserSettings.find_by_user(user)
|
|
if user_settings is not None:
|
|
user_config = ConfigParserWithEnv(defaults=self.settings, allow_no_value=True)
|
|
user_config.read_dict({
|
|
'user': user_settings.to_dict()
|
|
})
|
|
return user_config
|
|
|
|
return settings
|
|
|
|
|
|
instance: AppConfig = None
|
|
|
|
|
|
def __initialize_logger():
|
|
# Parse log level
|
|
log_level_str = instance.settings.get('global', 'LogLevel', fallback='INFO')
|
|
levels = {
|
|
'NOTSET': logging.NOTSET,
|
|
'DEBUG': logging.DEBUG,
|
|
'INFO': logging.INFO,
|
|
'WARNING': logging.WARNING,
|
|
'ERROR': logging.ERROR,
|
|
'CRITICAL': logging.CRITICAL
|
|
}
|
|
if log_level_str.upper() not in levels:
|
|
log_level_str = 'INFO'
|
|
|
|
# Init
|
|
logging.basicConfig(filename=instance.log_path, level=levels[log_level_str])
|
|
|
|
|
|
def initialize_config():
|
|
global instance
|
|
instance = AppConfig()
|
|
|
|
# Load settings
|
|
instance.load_settings()
|
|
|
|
# Initialize logger
|
|
__initialize_logger()
|
|
logging.info('Application started!')
|