diff options
author | Parker <contact@pkrm.dev> | 2023-09-18 01:13:15 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-18 01:13:15 +0000 |
commit | 86c6806d9365d7b4872b174c1c4b3db7cfc007e7 (patch) | |
tree | c55386cc5965981edffcf9d9b23b43d7c0102933 /app/initialize_variables.py | |
parent | 8b94436453fa618e5fc787611862804cd62961a9 (diff) | |
parent | eddeca8c67caf2e916af60880b327bbb2b4b8f20 (diff) |
Merge branch 'main' into dependabot/pip/flask-2.2.5
Diffstat (limited to 'app/initialize_variables.py')
-rw-r--r-- | app/initialize_variables.py | 86 |
1 files changed, 51 insertions, 35 deletions
diff --git a/app/initialize_variables.py b/app/initialize_variables.py index 6da1c34..33e2642 100644 --- a/app/initialize_variables.py +++ b/app/initialize_variables.py @@ -4,55 +4,71 @@ import requests supported_sms_services = ['telnyx'] - -radarr_host = os.environ['RADARR_HOST'] -radarr_api_key = os.environ['RADARR_API_KEY'] -try: - home_domain = os.environ['HOME_DOMAIN'] -except: - home_domain = None -api_number = os.environ['API_NUMBER'] -val_nums = os.environ['VALID_SENDERS'] -root_folder_path = os.environ['ROOT_FOLDER_PATH'] -sms_service = os.environ['SMS_SERVICE'] -if sms_service not in supported_sms_services: - print(f'{sms_service} is not a supported SMS service. Please choose from the supported list: {supported_sms_services}') - exit() -sms_api_key = os.environ['SMS_API_KEY'] +sms_service = str(os.environ['SMS_SERVICE']).lower() +radarr_host_url = str(os.environ['RADARR_HOST_URL']) +radarr_api_key = str(os.environ['RADARR_API_KEY']) headers = { 'Content-Type': 'application/json', 'X-Api-Key': radarr_api_key } -numbers_responses = { - '1': 1, 'one': 1, '1.': 1, - '2': 2, 'two': 2, '2.': 2, - '3': 3, 'three': 3, '3.': 3 -} - -# Open the quality_profile_id.yaml file and see if the quality_profile_id is set +# Open the config.yaml file and see if the config is set try: - with open('/data/quality_profile_id.yaml', 'r') as f: + with open('/data/config.yaml', 'r') as f: file = yaml.load(f, Loader=yaml.FullLoader) try: quality_profile_id = int(file['quality_profile_id']) + + if str(file['home_domain']) != 'null': + home_domain = str(file['home_domain']) + + api_number = str(file['api_number']) + val_nums = str(file['valid_senders']) + root_folder_path = str(file['root_folder_path']) + if sms_service not in supported_sms_services: + print(f'{sms_service} is not a supported SMS service. Please choose from the supported list: {supported_sms_services}') + exit() + + if sms_service == 'telnyx': + telnyx_api_key = str(file['telnyx_api_key']) + + value_not_set = False except: - print('quality_profile_id is not set or is invalid. Please edit the quality_profile_id.yaml file and add the quality_profile_id from one of the integer values listed within the file') + print('One or more values are not set or not set correctly within the config.yaml file. Please edit the file or refer to the docs for more information.') exit() + except FileNotFoundError: - # Create the quality_profile_id.yaml file - with open('/data/quality_profile_id.yaml', 'w') as f: - quality_profile_id = None - -if not quality_profile_id: - print('No quality_profile_id found. Please edit the quality_profile_id.yaml file and add the quality_profile_id from one of the integer values listed within the file') - data = requests.get(f'{radarr_host}/api/v3/qualityprofile', headers=headers).json() - # Open quality_profile_id.yaml and write each profile as a comment to the file - with open('/data/quality_profile_id.yaml', 'w') as f: + # Create the config.yaml file + with open('/data/config.yaml', 'w') as f: + value_not_set = True + +if value_not_set: + print('One or more values are not set or not set correctly within the config.yaml file. Please edit the file or refer to the docs for more information.') + data = requests.get(f'{radarr_host_url}/api/v3/qualityprofile', headers=headers).json() + # Open config.yaml and write each profile as a comment to the file + with open('/data/config.yaml', 'w') as f: f.write('# Quality Profile ID\'s\n') for entry in data: f.write(f'# {entry["id"]} - {entry["name"]}\n') - f.write("quality_profile_id: ") - exit()
\ No newline at end of file + f.write("quality_profile_id:\n") + f.write("home_domain: null\n") + f.write("api_number:\n") + f.write("valid_senders:\n") + f.write("root_folder_path:\n") + if sms_service not in supported_sms_services: + print(f'{sms_service} is not a supported SMS service. Please choose from the supported list: {supported_sms_services}') + exit() + + if sms_service == 'telnyx': + f.write("telnyx_api_key:\n") + + f.write("\n\n# INFORMATION: There should be NO trailing spaced after you enter a value,\n# this will cause errors.\n# There should be one space after the colon though (e.g. quality_profile_id: 1)\n# Check docs for information on each value.") + exit() + +numbers_responses = { + '1': 1, 'one': 1, '1.': 1, + '2': 2, 'two': 2, '2.': 2, + '3': 3, 'three': 3, '3.': 3 +}
\ No newline at end of file |