From e82f2cdca9913fb990234b53e6459825d896a965 Mon Sep 17 00:00:00 2001 From: santiagosayshey Date: Sat, 27 Jan 2024 15:14:04 +1030 Subject: [PATCH] v0.2.3 - changed from json to yaml for config - added export_path setting to set where exported files go : fixes https://github.com/santiagosayshey/Profilarr/issues/5 - exports now save to {export_path}/{app_type}/{app_name} - can choose which instance to manually import to - sync automatically chooses correct custom formats / quality profiles for the app type - updated readme with usage examples - added install requirements Closes #4, #5 --- README.md | 236 +++++++++++++++++++++++++++++++++++++++++++---- config.json | 38 -------- config.yml | 20 ++++ exportarr.py | 71 +++++++++----- importarr.py | 198 ++++++++++++++++++++++++--------------- requirements.txt | Bin 0 -> 254 bytes syncarr.py | 42 +++++---- 7 files changed, 435 insertions(+), 170 deletions(-) delete mode 100644 config.json create mode 100644 config.yml create mode 100644 requirements.txt diff --git a/README.md b/README.md index 9299132..e0392d4 100644 --- a/README.md +++ b/README.md @@ -17,29 +17,20 @@ Profilarr is a Python-based tool designed to add import/export/sync functionalit ### 📦 Dependencies -- `requests` (Install using `pip install requests`) +- run `pip install -r requirements.txt` to install dependencies. ### Initial Setup 1. Download the latest Profilarr package from the release section. 2. Extract its contents into a folder. -3. Open the `config.json` file in a text editor. - - Add your Radarr / Sonarr API key and modify the base URL as necessary. - - If importing / exporting, only change the master installation's API key and base URL. - - If syncing, add the API keys and base URLs of all instances you want to sync. - - The master install will be the one that all other instances sync to. +3. Open the `config.yml` file in a text editor. + - Add the URL and API key to the master instances of Radarr / Sonarr. + - If syncing, add the URL, API key and a name to each extra instance of Radarr / Sonarr. + - If exporting, adjust the `export_path` to your desired export location. 4. Save the changes. ## 🚀 Usage -### Exporting - -1. Run `python exportarr.py` in your command line interface. -2. Follow the on-screen prompts to select the app (Radarr or Sonarr) and the data (Custom Formats or Quality Profiles) you want to export. -3. Exported data will be saved in respective directories within the tool's folder. -- Please be careful when running this tool, it will OVERWRITE the given custom formats file and you'll need to download it again if you want to import the quality profiles. -- In future, there will be a config option to set where the files export to. - ### Importing 1. Run `python importarr.py` in your command line interface. @@ -47,16 +38,225 @@ Profilarr is a Python-based tool designed to add import/export/sync functionalit 3. Choose the specific file for Custom Formats or select a profile for Quality Profiles. 4. The data will be imported to your selected Radarr or Sonarr installation. +#### Custom Format Import Example + +```bash +PS Z:\Profilarr> py importarr.py +Available instances to import to: +1. Sonarr [Master] +2. Radarr [Master] +3. Sonarr [4k-sonarr] +4. Radarr [4k-radarr] +Enter the number of the instance to import to: 4 + + +Choose what to import: +1. Custom Formats +2. Quality Profiles +Enter your choice (1/2): 1 + +Available files: +1. Custom Formats (Radarr).json +Select a file to import (or 'all' for all files): 1 + +Adding custom format 'D-Z0N3': SUCCESS +Adding custom format 'DON': SUCCESS +Adding custom format 'EbP': SUCCESS +Adding custom format 'Geek': SUCCESS +Adding custom format 'TayTo': SUCCESS +Adding custom format 'ZQ': SUCCESS +Adding custom format 'VietHD': SUCCESS +Adding custom format 'CtrlHD': SUCCESS +Adding custom format 'HiFi': SUCCESS +Adding custom format 'FoRM': SUCCESS +Adding custom format 'HiDt': SUCCESS +Adding custom format 'SA89': SUCCESS +... + +Successfully added 0 custom formats, updated 131 custom formats. +``` + +#### Quality Profile Import Example + +```bash +PS Z:\Profilarr> py importarr.py +Available instances to import to: +1. Sonarr [Master] +2. Radarr [Master] +3. Sonarr [4k-sonarr] +4. Radarr [4k-radarr] +Enter the number of the instance to import to: 4 + + +Choose what to import: +1. Custom Formats +2. Quality Profiles +Enter your choice (1/2): 2 + +Available files: +1. 1080p Balanced (Radarr).json +2. 1080p Balanced (Single Grab) (Radarr).json +3. 1080p h265 Balanced (Radarr).json +4. 1080p h265 Balanced (Single Grab) (Radarr).json +5. 1080p Optimal (Radarr).json +6. 1080p Optimal (Single Grab) (Radarr).json +7. 1080p Transparent (Double Grab) (Radarr).json +8. 1080p Transparent (Radarr).json +9. 1080p Transparent (Single Grab) (Radarr).json +10. 2160p Optimal (Radarr).json +11. 2160p Optimal (Single Grab) (Radarr).json +Select a file to import (or 'all' for all files): all + +Successfully added Quality Profile 1080p Balanced +Successfully added Quality Profile 1080p Balanced (Single Grab) +Successfully added Quality Profile 1080p h265 Balanced +Successfully added Quality Profile 1080p h265 Balanced (Single Grab) +Successfully added Quality Profile 1080p Optimal +Successfully added Quality Profile 1080p Optimal (Single Grab) +Successfully added Quality Profile 1080p Transparent (Double Grab) +Successfully added Quality Profile 1080p Transparent +Successfully added Quality Profile 1080p Transparent (Single Grab) +Successfully added Quality Profile 2160p Optimal +Successfully added Quality Profile 2160p Optimal (Single Grab) +PS Z:\Profilarr> +``` + +### Exporting + +1. Run `python exportarr.py` in your command line interface. +2. Choose the instance you want to export from. +3. Choose the data you want to export. +4. The data will be exported to `exports/{instance_type}/{instance_name}/{data_type}`. + +#### Example + +```bash +PS Z:\Profilarr> py exportarr.py +Available sources to export from: +1. Sonarr [Master] +2. Radarr [Master] +3. Sonarr [4k-sonarr] +4. Radarr [4k-radarr] +Enter the number of the app to export from: 2 + +Choose what to export: +1. Custom Formats +2. Quality Profiles +3. Both +Enter your choice (1/2/3): 3 + +Attempting to access Radarr at http://localhost:7878 +Found 131 custom formats. + - D-Z0N3 + - DON + - EbP + - Geek + - TayTo + - ZQ + - VietHD + - CtrlHD + - HiFi + - FoRM +... and 121 more. +Saved to './exports\radarr\master\custom_formats\Custom Formats (Radarr).json' + +Attempting to access Radarr at http://localhost:7878 +Found 13 quality profiles. + - 1080p Optimal + - 2160p Optimal + - 1080p Balanced + - 1080p Transparent + - 1080p Transparent (Double Grab) + - 1080p Transparent (Single Grab) + - 1080p Balanced (Single Grab) + - 1080p h265 Balanced + - 1080p h265 Balanced (Single Grab) + - 1080p x265 HDR Transparent +... and 3 more. +Saved to 'exports\radarr\master\profiles' + +PS Z:\Profilarr> +``` + ### Syncing 1. Run `python syncarr.py` in your command line interface. 2. The script will automatically export data from the master instance and import it to all other instances specified in `config.json`. 3. This feature is designed to manage multiple Radarr/Sonarr instances, syncing profiles and formats seamlessly. +#### Example + +```bash +PS Z:\Profilarr> py syncarr.py +Select the app you want to sync: +1. Radarr +2. Sonarr +Enter your choice (1 or 2): 2 +Attempting to access Sonarr at http://localhost:8989 +Found 135 custom formats. + - D-Z0N3 + - DON + - EbP + - Geek + - TayTo + - ZQ + - VietHD + - CtrlHD + - HiFi + - FoRM +... and 125 more. +Saved to './temp_directory/custom_formats\Custom Formats (Sonarr).json' + +Attempting to access Sonarr at http://localhost:8989 +Found 11 quality profiles. + - 1080p Transparent + - 2160p Optimal + - 1080p Transparent (Single Grab) + - 1080p Transparent (Double Grab) + - 1080p Balanced + - 1080p Balanced (Single Grab) + - 1080p h265 Balanced + - 1080p h265 Balanced (Single Grab) + - 1080p Optimal + - 1080p Optimal (Single Grab) +... and 1 more. +Saved to 'temp_directory\quality_profiles' + +Importing to instance: 4k-sonarr +Adding custom format 'D-Z0N3': SUCCESS +Adding custom format 'DON': SUCCESS +Adding custom format 'EbP': SUCCESS +Adding custom format 'Geek': SUCCESS +Adding custom format 'TayTo': SUCCESS +Adding custom format 'ZQ': SUCCESS +Adding custom format 'VietHD': SUCCESS +Adding custom format 'CtrlHD': SUCCESS +Adding custom format 'HiFi': SUCCESS +... and 125 more. + +Successfully added 135 custom formats, updated 0 custom formats. +Successfully added Quality Profile 1080p Balanced (Single Grab) +Successfully added Quality Profile 1080p Balanced +Successfully added Quality Profile 1080p h265 Balanced +Successfully added Quality Profile 1080p h265 Balanced (Single Grab) +Successfully added Quality Profile 1080p Optimal (Single Grab) +Successfully added Quality Profile 1080p Optimal +Successfully added Quality Profile 1080p Transparent (Double Grab) +Successfully added Quality Profile 1080p Transparent (Single Grab) +Successfully added Quality Profile 1080p Transparent +Successfully added Quality Profile 2160p Optimal (Single Grab) +Successfully added Quality Profile 2160p Optimal +Deleted temporary directory: ./temp_directory +PS Z:\Profilarr> +``` + ### Radarr and Sonarr Compatibility -- Custom formats _can_ be imported and exported between Radarr and Sonarr (but might not work as expected). -- Quality profiles are not directly interchangeable between Radarr and Sonarr due to differences in quality source names. If you want to use the same profile in both apps, you will need to manually edit the profile's quality source names before importing it. +- You are only able to import / sync files to the app that is included in the file name (e.g. `Radarr` or `Sonarr`). +- It is possible to manually rename the files to import them to the other app, but this is not recommended. + - Custom Formats will succesfully import, but will require manual editing to work with the other app, i.e. you must adjust the quality sources to match the other app's naming scheme. + - Quality Profiles will not import at all, as they are not compatible with the other app. It is possible to import them manually by editing the json directly, but this is not recommended. + - In future, I may add a feature to automatically convert profiles between the two apps, but this is not currently a priority. ## 🌟 Upcoming Features @@ -69,3 +269,7 @@ Profilarr is a Python-based tool designed to add import/export/sync functionalit Some custom formats found here have been interated on from the trash guides. Credit for these goes entirely to trash, and can be found on their site here. It is not my intention to steal their work, but rather to build on it and make it more accessible to the average user through my quality profiles. Please check out their site for more information on their work. https://trash-guides.info/ + +``` + +``` diff --git a/config.json b/config.json deleted file mode 100644 index f42dfea..0000000 --- a/config.json +++ /dev/null @@ -1,38 +0,0 @@ -{ - "master": { - "sonarr": { - "base_url": "http://localhost:8989", - "api_key": "API_KEY_HERE" - }, - "radarr": { - "base_url": "http://localhost:7878", - "api_key": "API_KEY_HERE" - } - }, - "extra_installations": { - "radarr": [ - { - "name": "extra_radarr1", - "base_url": "http://localhost:7788", - "api_key": "API_KEY_HERE" - }, - { - "name": "extra_radarr2", - "base_url": "http://localhost:7789", - "api_key": "API_KEY_HERE" - } - ], - "sonarr": [ - { - "name": "extra_sonarr1", - "base_url": "http://localhost:8988", - "api_key": "API_KEY_HERE" - }, - { - "name": "extra_sonarr2", - "base_url": "http://localhost:8987", - "api_key": "API_KEY_HERE" - } - ] - } -} diff --git a/config.yml b/config.yml new file mode 100644 index 0000000..81e9cbf --- /dev/null +++ b/config.yml @@ -0,0 +1,20 @@ +instances: + master: + sonarr: + base_url: "http://localhost:8989" + api_key: "API_KEY" + radarr: + base_url: "http://localhost:7878" + api_key: "API_KEY" + extras: + sonarr: + - name: "4k-sonarr" + base_url: "http://localhost:8998" + api_key: "API_KEY" + radarr: + - name: "4k-radarr" + base_url: "http://localhost:7887" + api_key: "API_KEY" + +settings: + export_path: "./exports" diff --git a/exportarr.py b/exportarr.py index a520859..debb63a 100644 --- a/exportarr.py +++ b/exportarr.py @@ -1,7 +1,8 @@ -import json import requests import os import re +import yaml +import json # ANSI escape sequences for colors class Colors: @@ -15,16 +16,39 @@ class Colors: UNDERLINE = '\033[4m' # Load configuration for main app -with open('config.json', 'r') as config_file: - config = json.load(config_file)['master'] +with open('config.yml', 'r') as config_file: + config = yaml.safe_load(config_file) + master_config = config['instances']['master'] + export_base_path = config['settings']['export_path'] def get_user_choice(): - choice = input("Enter an app to export from (radarr/sonarr): ").lower() - while choice not in ["radarr", "sonarr"]: - print(Colors.FAIL + "Invalid input. Please enter either 'radarr' or 'sonarr'." + Colors.ENDC) - choice = input("Enter the source (radarr/sonarr): ").lower() + sources = [] + print(Colors.HEADER + "Available sources to export from:" + Colors.ENDC) + + # Add master installations + for app in master_config: + sources.append((app, f"{app.capitalize()} [Master]", "master")) + + # Add extra installations + if "extras" in config['instances']: + for app, instances in config['instances']['extras'].items(): + for install in instances: + sources.append((app, f"{app.capitalize()} [{install['name']}]", install['name'])) + + # Display sources with numbers + for idx, (app, name, _) in enumerate(sources, start=1): + print(f"{idx}. {name}") + + # User selection + choice = input("Enter the number of the app to export from: ").strip() + while not choice.isdigit() or int(choice) < 1 or int(choice) > len(sources): + print(Colors.FAIL + "Invalid input. Please enter a valid number." + Colors.ENDC) + choice = input("Enter the number of the app to export from: ").strip() + + selected_app, instance_name = sources[int(choice) - 1][0], sources[int(choice) - 1][2] print() - return choice + return selected_app, instance_name + def get_export_choice(): print(Colors.HEADER + "Choose what to export:" + Colors.ENDC) @@ -39,7 +63,7 @@ def get_export_choice(): return choice def get_app_config(source): - app_config = config[source] + app_config = master_config[source] return app_config['base_url'], app_config['api_key'] def sanitize_filename(filename): @@ -70,8 +94,10 @@ def ensure_directory_exists(directory): os.makedirs(directory) print(Colors.OKBLUE + f"Created directory: {directory}" + Colors.ENDC) -def export_cf(source, save_path='./custom_formats'): - ensure_directory_exists(save_path) # Ensure the directory exists with the given save_path +def export_cf(source, instance_name, save_path=None): + if save_path is None: + save_path = os.path.join(export_base_path, source, instance_name, 'custom_formats') + ensure_directory_exists(save_path) base_url, api_key = get_app_config(source) headers = {"X-Api-Key": api_key} @@ -93,7 +119,7 @@ def export_cf(source, save_path='./custom_formats'): custom_format.pop('id', None) saved_formats.append(custom_format['name']) - file_path = f'{save_path}/Custom Formats ({source.capitalize()}).json' + file_path = os.path.join(save_path, f'Custom Formats ({source.capitalize()}).json') with open(file_path, 'w') as f: json.dump(data, f, indent=4) @@ -108,8 +134,10 @@ def export_cf(source, save_path='./custom_formats'): -def export_qf(source, save_path='./profiles'): - ensure_directory_exists(save_path) # Ensure the directory exists with the given save_path +def export_qf(source, instance_name, save_path=None): + if save_path is None: + save_path = os.path.join(export_base_path, source, instance_name, 'profiles') + ensure_directory_exists(save_path) base_url, api_key = get_app_config(source) headers = {"X-Api-Key": api_key} @@ -124,9 +152,6 @@ def export_qf(source, save_path='./profiles'): quality_profiles = response.json() print(Colors.OKGREEN + f"Found {len(quality_profiles)} quality profiles." + Colors.ENDC) - if not os.path.exists('./profiles'): - os.makedirs('./profiles') - saved_profiles = [] for profile in quality_profiles: profile.pop('id', None) @@ -134,12 +159,12 @@ def export_qf(source, save_path='./profiles'): profile_name = sanitize_filename(profile_name) profile_filename = f"{profile_name} ({source.capitalize()}).json" profile_filepath = os.path.join(save_path, profile_filename) - saved_profiles.append(profile_name) - with open(profile_filepath, 'w') as file: json.dump([profile], file, indent=4) + saved_profiles.append(profile_name) # Add the profile name to the list + print_saved_items(saved_profiles, "Quality Profiles") - print(Colors.OKGREEN + f"Saved to '{profile_filepath}'" + Colors.ENDC) + print(Colors.OKGREEN + f"Saved to '{os.path.normpath(save_path)}'" + Colors.ENDC) # Normalize the path print() else: handle_response_errors(response) @@ -149,10 +174,10 @@ def export_qf(source, save_path='./profiles'): if __name__ == "__main__": - user_choice = get_user_choice() + user_choice, instance_name = get_user_choice() export_choice = get_export_choice() if export_choice in ["1", "3"]: - export_cf(user_choice) + export_cf(user_choice, instance_name) if export_choice in ["2", "3"]: - export_qf(user_choice) + export_qf(user_choice, instance_name) \ No newline at end of file diff --git a/importarr.py b/importarr.py index 72250d3..eca09d4 100644 --- a/importarr.py +++ b/importarr.py @@ -1,6 +1,8 @@ -import json import requests import os +import re +import yaml +import json # ANSI escape sequences for colors class Colors: @@ -14,8 +16,9 @@ class Colors: UNDERLINE = '\033[4m' # Load configuration for main app -with open('config.json', 'r') as config_file: - config = json.load(config_file) +with open('config.yml', 'r') as config_file: + config = yaml.safe_load(config_file) + master_config = config['instances']['master'] def print_success(message): print(Colors.OKGREEN + message + Colors.ENDC) @@ -27,11 +30,32 @@ def print_connection_error(): print(Colors.FAIL + "Failed to connect to the service! Please check if it's running and accessible." + Colors.ENDC) def get_user_choice(): - choice = input("Enter the app you want to import to (radarr/sonarr): ").lower() - while choice not in ["radarr", "sonarr"]: - print_error("Invalid input. Please enter either 'radarr' or 'sonarr'.") - choice = input("Enter the source (radarr/sonarr): ").lower() - return choice + sources = [] + print(Colors.HEADER + "Available instances to import to:" + Colors.ENDC) + + # Add master installations + for app in master_config: + sources.append((app, f"{app.capitalize()} [Master]")) + + # Add extra installations + if "extras" in config['instances']: + for app, instances in config['instances']['extras'].items(): + for install in instances: + sources.append((app, f"{app.capitalize()} [{install['name']}]")) + + # Display sources with numbers + for idx, (app, name) in enumerate(sources, start=1): + print(f"{idx}. {name}") + + # User selection + choice = input("Enter the number of the instance to import to: ").strip() + while not choice.isdigit() or int(choice) < 1 or int(choice) > len(sources): + print_error("Invalid input. Please enter a valid number.") + choice = input("Enter the number of the instance to import to: ").strip() + + selected_app, selected_name = sources[int(choice) - 1] + print() + return selected_app, selected_name def get_import_choice(): print() @@ -44,67 +68,93 @@ def get_import_choice(): choice = input("Enter your choice (1/2): ").strip() return choice -def get_app_config(source): - return config['master'][source] +def get_app_config(app_name, instance_name): + if instance_name.endswith("[Master]"): + return master_config[app_name] + else: + instance_name = instance_name.replace(f"{app_name.capitalize()} [", "").replace("]", "") + extras = config['instances']['extras'].get(app_name, []) + for instance in extras: + if instance['name'] == instance_name: + return instance + raise ValueError(f"Configuration for {app_name} - {instance_name} not found.") + +def select_file(directory, app_name, sync_mode=False): + app_name = app_name.lower() + files = [f for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f)) and app_name in f.lower()] + if not files: + print_error(f"No files found for {app_name.capitalize()} in {directory}.") + return None + + if sync_mode: + # Automatically select all files in sync mode + return files -def select_file(directory): - files = [f for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))] print() print(Colors.OKBLUE + "Available files:" + Colors.ENDC) for i, file in enumerate(files, 1): print(f"{i}. {file}") - choice = int(input("Select a file to import: ")) - return files[choice - 1] -def import_custom_formats(source_config, import_path='./custom_formats', auto_select_file=False): + choice = input("Select a file to import (or 'all' for all files): ").strip() + print() + if choice.isdigit() and 1 <= int(choice) <= len(files): + return [files[int(choice) - 1]] + elif choice.lower() == 'all': + return files + else: + print_error("Invalid input. Please enter a valid number or 'all'.") + print() + return None + + + +def import_custom_formats(source_config, import_path='./custom_formats', selected_files=None, sync_mode=False): headers = {"X-Api-Key": source_config['api_key']} get_url = f"{source_config['base_url']}/api/v3/customformat" + try: response = requests.get(get_url, headers=headers) if response.status_code == 200: existing_formats = response.json() existing_names_to_id = {format['name']: format['id'] for format in existing_formats} - files = os.listdir(import_path) - if auto_select_file and len(files) == 1: - selected_file = files[0] - else: - selected_file = select_file(import_path) + if selected_files is None: + selected_files = select_file(import_path, source_config['app_name'], sync_mode=sync_mode) + if not selected_files: + return # Exit if no file is selected - added_count, updated_count = 0, 0 + for selected_file in selected_files: + added_count, updated_count = 0, 0 + with open(os.path.join(import_path, selected_file), 'r') as import_file: + import_formats = json.load(import_file) - with open(os.path.join(import_path, selected_file), 'r') as import_file: - import_formats = json.load(import_file) + for format in import_formats: + format_name = format['name'] + if format_name in existing_names_to_id: + format_id = existing_names_to_id[format_name] + put_url = f"{source_config['base_url']}/api/v3/customformat/{format_id}" + response = requests.put(put_url, json=format, headers=headers) + if response.status_code in [200, 201, 202]: + print(Colors.WARNING + f"Updating custom format '{format_name}': " + Colors.ENDC, end='') + print_success("SUCCESS") + updated_count += 1 + else: + print_error(f"Updating custom format '{format_name}': FAIL") + print(response.content.decode()) - print() - - for format in import_formats: - format_name = format['name'] - if format_name in existing_names_to_id: - format_id = existing_names_to_id[format_name] - put_url = f"{source_config['base_url']}/api/v3/customformat/{format_id}" - response = requests.put(put_url, json=format, headers=headers) - if response.status_code in [200, 201, 202]: - print(Colors.WARNING + f"Updating custom format '{format_name}': " + Colors.ENDC, end='') - print_success("SUCCESS") - updated_count += 1 else: - print_error(f"Updating custom format '{format_name}': FAIL") - print(response.content.decode()) + post_url = f"{source_config['base_url']}/api/v3/customformat" + response = requests.post(post_url, json=format, headers=headers) + if response.status_code in [200, 201]: + print(Colors.OKBLUE + f"Adding custom format '{format_name}': " + Colors.ENDC, end='') + print_success("SUCCESS") + added_count += 1 + else: + print_error(f"Adding custom format '{format_name}': FAIL") + print(response.content.decode()) - else: - post_url = f"{source_config['base_url']}/api/v3/customformat" - response = requests.post(post_url, json=format, headers=headers) - if response.status_code in [200, 201]: - print(Colors.OKBLUE + f"Adding custom format '{format_name}': " + Colors.ENDC, end='') - print_success("SUCCESS") - added_count += 1 - else: - print_error(f"Adding custom format '{format_name}': FAIL") - print(response.content.decode()) - - print() - print_success(f"Successfully added {added_count} custom formats, updated {updated_count} custom formats.") + print() + print_success(f"Successfully added {added_count} custom formats, updated {updated_count} custom formats.") else: print_error(f"Failed to retrieve existing custom formats from {get_url}! (HTTP {response.status_code})") @@ -113,35 +163,24 @@ def import_custom_formats(source_config, import_path='./custom_formats', auto_se except requests.exceptions.ConnectionError: print_connection_error() -def import_quality_profiles(source_config, import_path='./profiles'): + + +def import_quality_profiles(source_config, import_path='./profiles', selected_files=None, sync_mode=False): headers = {"X-Api-Key": source_config['api_key']} + try: cf_import_sync(source_config) - profile_dir = import_path - profiles = [f for f in os.listdir(profile_dir) if f.endswith('.json')] + if not selected_files: + if sync_mode: + # Automatically select all profile files + selected_files = [f for f in os.listdir(import_path) if os.path.isfile(os.path.join(import_path, f))] - print() - print(Colors.HEADER + "Available Profiles:" + Colors.ENDC) - for i, profile in enumerate(profiles, 1): - print(f"{i}. {profile}") - print(f"{len(profiles) + 1}. Import all profiles") - - print() - selection = input("Please enter the number of the profile you want to import (or enter " + str(len(profiles) + 1) + " to import all): ") - selected_files = [] - try: - selection = int(selection) - if selection == len(profiles) + 1: - selected_files = profiles - else: - selected_files = [profiles[selection - 1]] - except (ValueError, IndexError): - print_error("Invalid selection, please enter a valid number.") - return + if not selected_files: + return # Exit if no file is selected for selected_file in selected_files: - with open(os.path.join(profile_dir, selected_file), 'r') as file: + with open(os.path.join(import_path, selected_file), 'r') as file: try: quality_profiles = json.load(file) except json.JSONDecodeError as e: @@ -187,6 +226,8 @@ def import_quality_profiles(source_config, import_path='./profiles'): print_connection_error() + + def cf_import_sync(source_config): headers = {"X-Api-Key": source_config['api_key']} custom_format_url = f"{source_config['base_url']}/api/v3/customformat" @@ -209,11 +250,16 @@ def cf_import_sync(source_config): if __name__ == "__main__": - user_choice = get_user_choice() - source_config = get_app_config(user_choice) + selected_app, selected_instance = get_user_choice() + source_config = get_app_config(selected_app, selected_instance) + source_config['app_name'] = selected_app import_choice = get_import_choice() if import_choice == "1": - import_custom_formats(source_config) + selected_files = select_file('./custom_formats', selected_app) + if selected_files: + import_custom_formats(source_config, './custom_formats', selected_files) elif import_choice == "2": - import_quality_profiles(source_config) + selected_files = select_file('./profiles', selected_app) + if selected_files: + import_quality_profiles(source_config, './profiles', selected_files) \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..42a803dcaa2bee24bca5623dad4625d0d0431e97 GIT binary patch literal 254 zcmX|+y9&ZU5JgWd_$kU}BM}P=dj(s|7&R_LgLxGC@#>usWSD`ykJ&q)cg2~Gfy70< zRz-q3XHKTFxn#a1T~UZJRpuKy{r6zh?3Jc>MHHKR=HQx9`5I4m6#l!-{I