mirror of
https://github.com/Dictionarry-Hub/profilarr.git
synced 2026-01-22 10:51:02 +01:00
22d4029e20
- implemented new data page for media management - renaming options to set movie / episode / folder formats - misc options for propers/repacks/video analysis - quality definitions - syncing with instances - improved mobile view for tabs / navbar
258 lines
12 KiB
Python
258 lines
12 KiB
Python
import logging
|
|
import requests
|
|
from typing import Dict, Any, Tuple
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def sync_naming_config(base_url: str, api_key: str, arr_type: str, naming_data: Dict[str, Any]) -> Tuple[bool, str]:
|
|
"""
|
|
Sync naming configuration to arr instance.
|
|
First GET current config, update with our data, then PUT back.
|
|
|
|
Args:
|
|
base_url: The arr instance base URL
|
|
api_key: The arr instance API key
|
|
arr_type: Either 'radarr' or 'sonarr'
|
|
naming_data: The naming configuration from our YML file
|
|
|
|
Returns:
|
|
Tuple of (success, message)
|
|
"""
|
|
try:
|
|
# Construct the endpoint URL
|
|
endpoint = f"{base_url}/api/v3/config/naming"
|
|
headers = {
|
|
"X-Api-Key": api_key,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
# GET current naming config
|
|
logger.info(f"Fetching current naming config from {arr_type} at {base_url}")
|
|
response = requests.get(endpoint, headers=headers, timeout=10)
|
|
response.raise_for_status()
|
|
|
|
current_config = response.json()
|
|
logger.info(f"Current naming config for {arr_type}:")
|
|
logger.info(current_config)
|
|
|
|
# Update current_config with fields from naming_data
|
|
if arr_type == 'radarr':
|
|
# Map our YML fields to Radarr API fields
|
|
if 'rename' in naming_data:
|
|
current_config['renameMovies'] = naming_data['rename']
|
|
if 'replaceIllegalCharacters' in naming_data:
|
|
current_config['replaceIllegalCharacters'] = naming_data['replaceIllegalCharacters']
|
|
if 'colonReplacementFormat' in naming_data:
|
|
current_config['colonReplacementFormat'] = naming_data['colonReplacementFormat']
|
|
if 'movieFormat' in naming_data:
|
|
current_config['standardMovieFormat'] = naming_data['movieFormat']
|
|
if 'movieFolderFormat' in naming_data:
|
|
current_config['movieFolderFormat'] = naming_data['movieFolderFormat']
|
|
else: # sonarr
|
|
# Map our YML fields to Sonarr API fields
|
|
if 'rename' in naming_data:
|
|
current_config['renameEpisodes'] = naming_data['rename']
|
|
if 'replaceIllegalCharacters' in naming_data:
|
|
current_config['replaceIllegalCharacters'] = naming_data['replaceIllegalCharacters']
|
|
if 'colonReplacementFormat' in naming_data:
|
|
current_config['colonReplacementFormat'] = naming_data['colonReplacementFormat']
|
|
if 'customColonReplacementFormat' in naming_data:
|
|
current_config['customColonReplacementFormat'] = naming_data['customColonReplacementFormat']
|
|
if 'multiEpisodeStyle' in naming_data:
|
|
current_config['multiEpisodeStyle'] = naming_data['multiEpisodeStyle']
|
|
if 'standardEpisodeFormat' in naming_data:
|
|
current_config['standardEpisodeFormat'] = naming_data['standardEpisodeFormat']
|
|
if 'dailyEpisodeFormat' in naming_data:
|
|
current_config['dailyEpisodeFormat'] = naming_data['dailyEpisodeFormat']
|
|
if 'animeEpisodeFormat' in naming_data:
|
|
current_config['animeEpisodeFormat'] = naming_data['animeEpisodeFormat']
|
|
if 'seriesFolderFormat' in naming_data:
|
|
current_config['seriesFolderFormat'] = naming_data['seriesFolderFormat']
|
|
if 'seasonFolderFormat' in naming_data:
|
|
current_config['seasonFolderFormat'] = naming_data['seasonFolderFormat']
|
|
if 'specialsFolderFormat' in naming_data:
|
|
current_config['specialsFolderFormat'] = naming_data['specialsFolderFormat']
|
|
|
|
# PUT the updated config back
|
|
logger.info(f"Updating naming config for {arr_type}")
|
|
logger.info(f"Request body for naming sync:")
|
|
logger.info(current_config)
|
|
put_response = requests.put(endpoint, json=current_config, headers=headers, timeout=10)
|
|
put_response.raise_for_status()
|
|
|
|
logger.info(f"Successfully synced naming config for {arr_type}")
|
|
return True, "Naming config sync successful"
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
error_msg = f"Failed to sync naming config: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg
|
|
except Exception as e:
|
|
error_msg = f"Unexpected error syncing naming config: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg
|
|
|
|
|
|
def sync_media_management_config(base_url: str, api_key: str, arr_type: str, misc_data: Dict[str, Any]) -> Tuple[bool, str]:
|
|
"""
|
|
Sync media management (misc) configuration to arr instance.
|
|
First GET current config, update with our data, then PUT back.
|
|
|
|
Args:
|
|
base_url: The arr instance base URL
|
|
api_key: The arr instance API key
|
|
arr_type: Either 'radarr' or 'sonarr'
|
|
misc_data: The misc configuration from our YML file
|
|
|
|
Returns:
|
|
Tuple of (success, message)
|
|
"""
|
|
try:
|
|
# Construct the endpoint URL
|
|
endpoint = f"{base_url}/api/v3/config/mediamanagement"
|
|
headers = {
|
|
"X-Api-Key": api_key,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
# GET current media management config
|
|
logger.info(f"Fetching current media management config from {arr_type} at {base_url}")
|
|
response = requests.get(endpoint, headers=headers, timeout=10)
|
|
response.raise_for_status()
|
|
|
|
current_config = response.json()
|
|
logger.info(f"Current media management config for {arr_type}:")
|
|
logger.info(current_config)
|
|
|
|
# Update current_config with fields from misc_data
|
|
# We only manage two fields: propersRepacks and enableMediaInfo
|
|
if 'propersRepacks' in misc_data:
|
|
current_config['downloadPropersAndRepacks'] = misc_data['propersRepacks']
|
|
if 'enableMediaInfo' in misc_data:
|
|
current_config['enableMediaInfo'] = misc_data['enableMediaInfo']
|
|
|
|
# PUT the updated config back
|
|
logger.info(f"Updating media management config for {arr_type}")
|
|
logger.info(f"Request body for media management sync:")
|
|
logger.info(current_config)
|
|
put_response = requests.put(endpoint, json=current_config, headers=headers, timeout=10)
|
|
put_response.raise_for_status()
|
|
|
|
logger.info(f"Successfully synced media management config for {arr_type}")
|
|
return True, "Media management config sync successful"
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
error_msg = f"Failed to sync media management config: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg
|
|
except Exception as e:
|
|
error_msg = f"Unexpected error syncing media management config: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg
|
|
|
|
|
|
def sync_quality_definitions(base_url: str, api_key: str, arr_type: str, quality_data: Dict[str, Any]) -> Tuple[bool, str]:
|
|
"""
|
|
Sync quality definitions to arr instance.
|
|
Quality definitions contain all required data, so we can directly PUT.
|
|
|
|
Args:
|
|
base_url: The arr instance base URL
|
|
api_key: The arr instance API key
|
|
arr_type: Either 'radarr' or 'sonarr'
|
|
quality_data: The quality definitions from our YML file
|
|
|
|
Returns:
|
|
Tuple of (success, message)
|
|
"""
|
|
try:
|
|
# Construct the endpoint URL
|
|
endpoint = f"{base_url}/api/v3/qualitydefinition"
|
|
headers = {
|
|
"X-Api-Key": api_key,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
# GET current quality definitions (for logging/comparison)
|
|
logger.info(f"Fetching current quality definitions from {arr_type} at {base_url}")
|
|
response = requests.get(endpoint, headers=headers, timeout=10)
|
|
response.raise_for_status()
|
|
|
|
current_definitions = response.json()
|
|
logger.info(f"Current quality definitions for {arr_type}:")
|
|
logger.info(current_definitions)
|
|
|
|
if arr_type == 'sonarr':
|
|
# Log the quality data we received from YML
|
|
logger.info(f"Quality data from YML:")
|
|
logger.info(quality_data)
|
|
|
|
# Create a mapping of quality names to current definitions for easier lookup
|
|
quality_map = {def_['quality']['name']: def_ for def_ in current_definitions}
|
|
|
|
# Update each quality definition with our values
|
|
for quality_name, settings in quality_data.items():
|
|
if quality_name in quality_map:
|
|
definition = quality_map[quality_name]
|
|
# Update size limits from our YML data
|
|
if 'min' in settings:
|
|
definition['minSize'] = settings['min']
|
|
if 'preferred' in settings:
|
|
definition['preferredSize'] = settings['preferred']
|
|
if 'max' in settings:
|
|
definition['maxSize'] = settings['max']
|
|
|
|
# PUT the updated definitions back
|
|
logger.info(f"Updating quality definitions for {arr_type}")
|
|
logger.info(f"Request body for quality definitions sync:")
|
|
logger.info(current_definitions)
|
|
|
|
# Sonarr expects the full array of definitions at the update endpoint
|
|
update_endpoint = f"{base_url}/api/v3/qualitydefinition/update"
|
|
put_response = requests.put(update_endpoint, json=current_definitions, headers=headers, timeout=10)
|
|
put_response.raise_for_status()
|
|
|
|
logger.info(f"Successfully synced quality definitions for {arr_type}")
|
|
return True, "Quality definitions sync successful"
|
|
|
|
else: # radarr
|
|
# Log the quality data we received from YML
|
|
logger.info(f"Quality data from YML:")
|
|
logger.info(quality_data)
|
|
|
|
# Create a mapping of quality names to current definitions for easier lookup
|
|
quality_map = {def_['quality']['name']: def_ for def_ in current_definitions}
|
|
|
|
# Update each quality definition with our values
|
|
for quality_name, settings in quality_data.items():
|
|
if quality_name in quality_map:
|
|
definition = quality_map[quality_name]
|
|
# Update size limits from our YML data
|
|
if 'min' in settings:
|
|
definition['minSize'] = settings['min']
|
|
if 'preferred' in settings:
|
|
definition['preferredSize'] = settings['preferred']
|
|
if 'max' in settings:
|
|
definition['maxSize'] = settings['max']
|
|
|
|
# PUT the updated definitions back
|
|
logger.info(f"Updating quality definitions for {arr_type}")
|
|
logger.info(f"Request body for quality definitions sync:")
|
|
logger.info(current_definitions)
|
|
|
|
# Radarr expects the full array of definitions at the update endpoint
|
|
update_endpoint = f"{base_url}/api/v3/qualitydefinition/update"
|
|
put_response = requests.put(update_endpoint, json=current_definitions, headers=headers, timeout=10)
|
|
put_response.raise_for_status()
|
|
|
|
logger.info(f"Successfully synced quality definitions for {arr_type}")
|
|
return True, "Quality definitions sync successful"
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
error_msg = f"Failed to sync quality definitions: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg
|
|
except Exception as e:
|
|
error_msg = f"Unexpected error syncing quality definitions: {str(e)}"
|
|
logger.error(error_msg)
|
|
return False, error_msg |