mirror of
https://github.com/ManiMatter/decluttarr.git
synced 2026-04-18 09:54:04 +02:00
Initial commit (fork from ManiMatter/mirrarr)
Carved out the queue cleaning bits from mirrarr, without taking over the instance synchronization items. Not tested yet.
This commit is contained in:
114
config/config.py
Normal file
114
config/config.py
Normal file
@@ -0,0 +1,114 @@
|
||||
#!/usr/bin/env python
|
||||
import sys
|
||||
import os
|
||||
import configparser
|
||||
|
||||
########################################################################################################################
|
||||
# Check if in Docker
|
||||
IS_IN_DOCKER = os.environ.get('IS_IN_DOCKER')
|
||||
|
||||
########################################################################################################################
|
||||
def ConfigSectionMap(section):
|
||||
'Load the config file into a dictionary'
|
||||
dict1 = {}
|
||||
options = config.options(section)
|
||||
for option in options:
|
||||
try:
|
||||
dict1[option] = config.get(section, option)
|
||||
except:
|
||||
print("exception on %s!" % option)
|
||||
dict1[option] = None
|
||||
return dict1
|
||||
|
||||
def cast(value, type_):
|
||||
return type_(value)
|
||||
|
||||
def get_config_value(key, config_section, is_mandatory, datatype, default_value = None):
|
||||
'Return for each key the corresponding value from the Docker Environment or the Config File'
|
||||
if IS_IN_DOCKER:
|
||||
config_value = os.environ.get(key)
|
||||
if config_value is not None:
|
||||
# print(f'The value retrieved for [{config_section}]: {key} is "{config_value}"')
|
||||
config_value = config_value
|
||||
# return config_value
|
||||
elif is_mandatory:
|
||||
print(f'[ ERROR ]: Variable not specified in Docker environment: {key}' )
|
||||
sys.exit(0)
|
||||
else:
|
||||
# return default_value
|
||||
config_value = default_value
|
||||
|
||||
else:
|
||||
try:
|
||||
config_value = ConfigSectionMap(config_section).get(key)
|
||||
except configparser.NoSectionError:
|
||||
config_value = None
|
||||
if config_value is not None:
|
||||
# print(f'The value retrieved for [{config_section}]: {key} is "{config_value}"')
|
||||
config_value = config_value
|
||||
# return config_value
|
||||
elif is_mandatory:
|
||||
print(f'[ ERROR ]: Mandatory variable not specified in config file, section [{config_section}]: {key} (data type: {datatype.__name__})')
|
||||
sys.exit(0)
|
||||
else:
|
||||
# return default_value
|
||||
config_value = default_value
|
||||
|
||||
# Apply data type
|
||||
try:
|
||||
if datatype == bool:
|
||||
config_value = eval(str(config_value).capitalize())
|
||||
if config_value is not None: config_value = cast(config_value, datatype)
|
||||
except:
|
||||
print(f'[ ERROR ]: The value retrieved for [{config_section}]: {key} is "{config_value}" and cannot be converted to data type {datatype}')
|
||||
sys.exit(0)
|
||||
return config_value
|
||||
|
||||
########################################################################################################################
|
||||
# Load Config File
|
||||
Config_FileName = 'config.conf'
|
||||
Config_FileFullPath = os.path.join(os.path.abspath(os.path.dirname(__file__)), Config_FileName)
|
||||
sys.tracebacklimit = 0 # dont show stack traces in prod mode
|
||||
config = configparser.ConfigParser()
|
||||
config.optionxform = str # maintain capitalization of config keys
|
||||
config.read(Config_FileFullPath)
|
||||
|
||||
########################################################################################################################
|
||||
# Load Config
|
||||
# General
|
||||
LOG_LEVEL = get_config_value('LOG_LEVEL', 'general', False, str, 'INFO')
|
||||
TEST_RUN = get_config_value('TEST_RUN', 'general', False, bool, False)
|
||||
|
||||
# Features
|
||||
REMOVE_TIMER = get_config_value('REMOVE_TIMER', 'features', False, int, 10)
|
||||
REMOVE_FAILED = get_config_value('REMOVE_FAILED', 'features', False, bool, False)
|
||||
REMOVE_STALLED = get_config_value('REMOVE_STALLED', 'features', False, bool, False)
|
||||
REMOVE_METADATA_MISSING = get_config_value('REMOVE_METADATA_MISSING', 'features', False, bool, False)
|
||||
REMOVE_ORPHANS = get_config_value('REMOVE_ORPHANS' , 'features', False, bool, False)
|
||||
REMOVE_UNMONITORED = get_config_value('REMOVE_UNMONITORED' , 'features', False, bool, False)
|
||||
PERMITTED_ATTEMPTS = get_config_value('PERMITTED_ATTEMPTS', 'features', False, int, 3)
|
||||
NO_STALLED_REMOVAL_QBIT_TAG = get_config_value('NO_STALLED_REMOVAL_QBIT_TAG', 'features', False, str, 'Don\'t Kill If Stalled')
|
||||
|
||||
# Radarr
|
||||
RADARR_URL = get_config_value('RADARR_URL', 'radarr', False, str)
|
||||
RADARR_KEY = None if RADARR_URL == None else \
|
||||
get_config_value('RADARR_KEY', 'radarr', True, str)
|
||||
|
||||
# Sonarr
|
||||
SONARR_URL = get_config_value('SONARR_URL', 'sonarr', False, str)
|
||||
SONARR_KEY = None if SONARR_URL == None else \
|
||||
get_config_value('SONARR_KEY', 'sonarr', True, str)
|
||||
|
||||
# qBittorrent
|
||||
QBITTORRENT_URL = get_config_value('QBITTORRENT_URL', 'qbittorrent', False, str, '')
|
||||
|
||||
########################################################################################################################
|
||||
if not (RADARR_URL or SONARR_URL):
|
||||
print(f'[ ERROR ]: No Radarr/Sonarr URLs specified (nothing to monitor)')
|
||||
sys.exit(0)
|
||||
|
||||
########### Add API to URLs
|
||||
if RADARR_URL: RADARR_URL += '/api/v3'
|
||||
if SONARR_URL: SONARR_URL += '/api/v3'
|
||||
if QBITTORRENT_URL: QBITTORRENT_URL += '/api/v2'
|
||||
|
||||
Reference in New Issue
Block a user