mirror of
https://github.com/Dictionarry-Hub/profilarr.git
synced 2026-01-22 10:51:02 +01:00
d7d6b13e46
- added option to set radarr/sonarr specific scores that profilarr's compiler will handle on import - revise design for arr settings container - now styled as a table - completely rewrote import module. Now uses connection pooling to reuse connections. - fixed import progress bug where 1 failed format causes all other formats to be labelled as failed (even if they succeeded) - fixed bug where on pull sync wasn't working - improve styling for link / unlink database modals - fixed issue where 0 score formats were removed in selective mode
260 lines
9.7 KiB
Python
260 lines
9.7 KiB
Python
# app/importarr/__init__.py
|
|
from flask import Blueprint, request, jsonify
|
|
from flask_cors import cross_origin
|
|
import logging
|
|
import asyncio
|
|
from pathlib import Path
|
|
from ..arr.manager import get_arr_config
|
|
from ..data.utils import get_category_directory, load_yaml_file
|
|
from .format import import_formats_to_arr, async_import_formats_to_arr
|
|
from .profile import import_profiles_to_arr, async_import_profiles_to_arr
|
|
from ..db import get_unique_arrs
|
|
|
|
logger = logging.getLogger('importarr')
|
|
|
|
bp = Blueprint('import', __name__)
|
|
|
|
|
|
@bp.route('/format', methods=['POST', 'OPTIONS'])
|
|
@cross_origin()
|
|
def import_formats():
|
|
if request.method == 'OPTIONS':
|
|
return jsonify({}), 200
|
|
|
|
try:
|
|
data = request.get_json()
|
|
arr_id = data.get('arrId')
|
|
all_formats = data.get('all', False)
|
|
format_names = data.get('formatNames', [])
|
|
|
|
if not arr_id:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Arr ID is required'
|
|
}), 400
|
|
|
|
if not all_formats and not format_names:
|
|
return jsonify({
|
|
'success':
|
|
False,
|
|
'error':
|
|
'Either formatNames or all=true is required'
|
|
}), 400
|
|
|
|
# Get import_as_unique setting using the new function
|
|
import_settings = get_unique_arrs([arr_id])
|
|
arr_settings = import_settings.get(arr_id, {
|
|
'import_as_unique': False,
|
|
'name': 'Unknown'
|
|
})
|
|
import_as_unique = arr_settings['import_as_unique']
|
|
|
|
if import_as_unique:
|
|
logger.info(
|
|
f"Unique imports for {arr_settings['name']} are on, adjusting names for custom formats"
|
|
)
|
|
else:
|
|
logger.info(
|
|
f"Unique imports for {arr_settings['name']} is off, using original names"
|
|
)
|
|
|
|
# Get arr configuration
|
|
arr_config = get_arr_config(arr_id)
|
|
if not arr_config['success']:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Arr configuration not found'
|
|
}), 404
|
|
|
|
arr_data = arr_config['data']
|
|
|
|
# If all=true, get all format names from the custom_format directory
|
|
if all_formats:
|
|
try:
|
|
format_dir = Path(get_category_directory('custom_format'))
|
|
format_names = [f.stem for f in format_dir.glob('*.yml')]
|
|
if not format_names:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'No custom formats found'
|
|
}), 404
|
|
except Exception as e:
|
|
logger.error(
|
|
f"Error reading custom formats directory: {str(e)}")
|
|
return jsonify({
|
|
'success':
|
|
False,
|
|
'error':
|
|
'Failed to read custom formats directory'
|
|
}), 500
|
|
|
|
# Store original names for file lookups
|
|
original_names = format_names.copy()
|
|
|
|
# Modify format names if import_as_unique is true
|
|
if import_as_unique:
|
|
format_names = [f"{name} [Dictionarry]" for name in format_names]
|
|
logger.info(
|
|
f"Modified format names for unique import: {format_names}")
|
|
|
|
# Import formats with arr type from config, but use original names for file lookups
|
|
result = import_formats_to_arr(format_names=format_names,
|
|
original_names=original_names,
|
|
base_url=arr_data['arrServer'],
|
|
api_key=arr_data['apiKey'],
|
|
arr_type=arr_data['type'])
|
|
|
|
return jsonify(result), 200 if result['success'] else 400
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error importing custom formats: {str(e)}")
|
|
return jsonify({'success': False, 'error': str(e)}), 400
|
|
|
|
|
|
@bp.route('/profile', methods=['POST', 'OPTIONS'])
|
|
@cross_origin()
|
|
def import_profiles():
|
|
if request.method == 'OPTIONS':
|
|
return jsonify({}), 200
|
|
|
|
try:
|
|
data = request.get_json()
|
|
arr_id = data.get('arrId')
|
|
all_profiles = data.get('all', False)
|
|
profile_names = data.get('profileNames', [])
|
|
|
|
if not arr_id:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Arr ID is required'
|
|
}), 400
|
|
|
|
if not all_profiles and not profile_names:
|
|
return jsonify({
|
|
'success':
|
|
False,
|
|
'error':
|
|
'Either profileNames or all=true is required'
|
|
}), 400
|
|
|
|
# Get import_as_unique setting
|
|
import_settings = get_unique_arrs([arr_id])
|
|
arr_settings = import_settings.get(arr_id, {
|
|
'import_as_unique': False,
|
|
'name': 'Unknown'
|
|
})
|
|
import_as_unique = arr_settings['import_as_unique']
|
|
|
|
if import_as_unique:
|
|
logger.info(
|
|
f"Unique imports for {arr_settings['name']} are on, adjusting names for quality profiles"
|
|
)
|
|
else:
|
|
logger.info(
|
|
f"Unique imports for {arr_settings['name']} is off, using original names"
|
|
)
|
|
|
|
# Get arr configuration
|
|
arr_config = get_arr_config(arr_id)
|
|
if not arr_config['success']:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Arr configuration not found'
|
|
}), 404
|
|
|
|
arr_data = arr_config['data']
|
|
|
|
# If all=true, get all profile names
|
|
if all_profiles:
|
|
try:
|
|
profile_dir = Path(get_category_directory('profile'))
|
|
profile_names = [f.stem for f in profile_dir.glob('*.yml')]
|
|
if not profile_names:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'No quality profiles found'
|
|
}), 404
|
|
except Exception as e:
|
|
logger.error(f"Error reading profiles directory: {str(e)}")
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Failed to read profiles directory'
|
|
}), 500
|
|
|
|
# Store original names for file lookups
|
|
original_names = profile_names.copy()
|
|
|
|
# Modify profile names if import_as_unique is true
|
|
if import_as_unique:
|
|
profile_names = [f"{name} [Dictionarry]" for name in profile_names]
|
|
logger.info(
|
|
f"Modified profile names for unique import: {profile_names}")
|
|
|
|
logger.debug(
|
|
f"Attempting to import profiles: {profile_names} for {arr_data['type']}: {arr_data['name']}"
|
|
)
|
|
|
|
# Get any custom formats referenced in these profiles
|
|
format_names = set()
|
|
for profile_name in original_names: # Use original names for file lookup
|
|
try:
|
|
profile_file = f"{get_category_directory('profile')}/{profile_name}.yml"
|
|
format_data = load_yaml_file(profile_file)
|
|
|
|
# Extract from main custom_formats
|
|
for cf in format_data.get('custom_formats', []):
|
|
format_names.add(cf['name'])
|
|
|
|
# Extract from app-specific custom_formats
|
|
for cf in format_data.get('custom_formats_radarr', []):
|
|
format_names.add(cf['name'])
|
|
for cf in format_data.get('custom_formats_sonarr', []):
|
|
format_names.add(cf['name'])
|
|
except Exception as e:
|
|
logger.error(f"Error loading profile {profile_name}: {str(e)}")
|
|
continue
|
|
|
|
# Import/Update formats first - use async version for larger batch sizes
|
|
if format_names:
|
|
format_names_list = list(format_names)
|
|
# When we have more than a few formats, use the async import path
|
|
# which will parallelize the requests
|
|
if import_as_unique:
|
|
modified_format_names = [
|
|
f"{name} [Dictionarry]" for name in format_names_list
|
|
]
|
|
# Use the regular import function which will detect large batches
|
|
# and automatically use async when appropriate
|
|
import_formats_to_arr(
|
|
format_names=modified_format_names,
|
|
original_names=format_names_list,
|
|
base_url=arr_data['arrServer'],
|
|
api_key=arr_data['apiKey'],
|
|
arr_type=arr_data['type']
|
|
)
|
|
else:
|
|
# Use the regular import function which will detect large batches
|
|
# and automatically use async when appropriate
|
|
import_formats_to_arr(
|
|
format_names=format_names_list,
|
|
original_names=format_names_list,
|
|
base_url=arr_data['arrServer'],
|
|
api_key=arr_data['apiKey'],
|
|
arr_type=arr_data['type']
|
|
)
|
|
|
|
# Import profiles
|
|
result = import_profiles_to_arr(profile_names=profile_names,
|
|
original_names=original_names,
|
|
base_url=arr_data['arrServer'],
|
|
api_key=arr_data['apiKey'],
|
|
arr_type=arr_data['type'],
|
|
arr_id=arr_id,
|
|
import_as_unique=import_as_unique)
|
|
|
|
return jsonify(result), 200 if result['success'] else 400
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error importing quality profiles: {str(e)}")
|
|
return jsonify({'success': False, 'error': str(e)}), 400
|