Initial curated Unraid-AppOps-Gateway source from 3bd6f6fee310125bfbb4d4650389a817f498724d
This commit is contained in:
Public Source Publisher committed
commit
1dfbca08a1
528 files changed
+131054
No files matched your search
@@ -0,0 +1,300 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from typing import Any
|
||||
|
||||
import httpx
|
||||
|
||||
from ..core.config import settings
|
||||
from ..core.redaction import redact_message
|
||||
from .base import BaseConnector
|
||||
|
||||
|
||||
_SONARR_DEFAULT_TIMEOUT_SECONDS = 8.0
|
||||
|
||||
|
||||
@dataclass
|
||||
class _SonarrConfig:
|
||||
url: str
|
||||
api_key: str
|
||||
verify_tls: bool
|
||||
default_root_folder: str | None
|
||||
default_quality_profile: int | None
|
||||
search_on_add: bool
|
||||
|
||||
|
||||
def _as_int(value: Any, default: int = 0) -> int:
|
||||
if value is None or value == '':
|
||||
return default
|
||||
if isinstance(value, bool):
|
||||
return int(value)
|
||||
if isinstance(value, int | float):
|
||||
return int(value)
|
||||
try:
|
||||
return int(str(value).strip())
|
||||
except ValueError:
|
||||
return default
|
||||
|
||||
|
||||
def _as_bool(value: Any, default: bool = False) -> bool:
|
||||
if isinstance(value, bool):
|
||||
return value
|
||||
if value is None:
|
||||
return default
|
||||
return str(value).strip().lower() in {'1', 'true', 'yes', 'on'}
|
||||
|
||||
|
||||
def _safe_url() -> str:
|
||||
value = settings.apis.get('SONARR_URL')
|
||||
if not value:
|
||||
return ''
|
||||
return str(value).rstrip('/')
|
||||
|
||||
|
||||
class SonarrConnector(BaseConnector):
|
||||
name = 'Sonarr'
|
||||
|
||||
def is_configured(self) -> bool:
|
||||
return bool(settings.apis.get('SONARR_URL') and settings.apis.get('SONARR_API_KEY'))
|
||||
|
||||
def _config(self) -> _SonarrConfig | None:
|
||||
if not self.is_configured():
|
||||
return None
|
||||
|
||||
return _SonarrConfig(
|
||||
url=str(settings.apis['SONARR_URL']).rstrip('/'),
|
||||
api_key=str(settings.apis['SONARR_API_KEY']),
|
||||
verify_tls=bool(settings.verify_tls),
|
||||
default_root_folder=(
|
||||
str(settings.apis.get('SONARR_DEFAULT_ROOT_FOLDER'))
|
||||
if settings.apis.get('SONARR_DEFAULT_ROOT_FOLDER')
|
||||
else None
|
||||
),
|
||||
default_quality_profile=(
|
||||
_as_int(settings.apis.get('SONARR_DEFAULT_QUALITY_PROFILE'))
|
||||
if settings.apis.get('SONARR_DEFAULT_QUALITY_PROFILE')
|
||||
else None
|
||||
),
|
||||
search_on_add=_as_bool(settings.apis.get('SONARR_SEARCH_ON_ADD'), True),
|
||||
)
|
||||
|
||||
def _request(self, method: str, path: str, **kwargs: Any) -> Any:
|
||||
config = self._config()
|
||||
if not config:
|
||||
raise RuntimeError('missing_config')
|
||||
|
||||
headers = dict(kwargs.pop('headers', {}) or {})
|
||||
headers['X-Api-Key'] = config.api_key
|
||||
url = f'{config.url}{path}'
|
||||
|
||||
try:
|
||||
self.clear_error()
|
||||
with httpx.Client(timeout=_SONARR_DEFAULT_TIMEOUT_SECONDS, verify=config.verify_tls) as client:
|
||||
response = client.request(method=method, url=url, headers=headers, **kwargs)
|
||||
except (httpx.RequestError, ValueError) as exc:
|
||||
message = redact_message(str(exc))
|
||||
self.set_error(message)
|
||||
raise RuntimeError(message)
|
||||
|
||||
if response.status_code in {401, 403}:
|
||||
self.set_error('unauthorized')
|
||||
raise RuntimeError('unauthorized')
|
||||
if response.status_code >= 500:
|
||||
message = f'Sonarr server error: HTTP {response.status_code}'
|
||||
self.set_error(message)
|
||||
raise RuntimeError(message)
|
||||
if response.status_code >= 400:
|
||||
message = redact_message(f'Sonarr API error: HTTP {response.status_code} {(response.text or "")[:256]}')
|
||||
self.set_error(message)
|
||||
raise RuntimeError(message)
|
||||
|
||||
if response.status_code == 204:
|
||||
return None
|
||||
|
||||
try:
|
||||
return response.json()
|
||||
except ValueError as exc:
|
||||
message = redact_message(str(exc))
|
||||
self.set_error(message)
|
||||
raise RuntimeError(message)
|
||||
|
||||
def test_connection(self) -> dict[str, Any]:
|
||||
if not self.is_configured():
|
||||
return {'status': 'missing_config', 'reason': 'missing_config'}
|
||||
|
||||
try:
|
||||
payload = self._request('GET', '/api/v3/system/status')
|
||||
except RuntimeError as exc:
|
||||
reason = str(exc)
|
||||
if reason == 'missing_config':
|
||||
return {'status': 'missing_config', 'reason': reason}
|
||||
return {'status': 'error', 'reason': redact_message(reason)}
|
||||
|
||||
version = payload.get('version') if isinstance(payload, dict) else None
|
||||
return {'status': 'connected', 'version': str(version or 'unknown')}
|
||||
|
||||
def health(self) -> dict[str, Any]:
|
||||
if not self.is_configured():
|
||||
return {
|
||||
'success': False,
|
||||
'status': 'missing_config',
|
||||
'message': 'missing_config',
|
||||
'version': 'unknown',
|
||||
'url': '',
|
||||
'required_env_vars': ['SONARR_URL', 'SONARR_API_KEY'],
|
||||
}
|
||||
|
||||
result = self.test_connection()
|
||||
status = str(result.get('status', 'error'))
|
||||
return {
|
||||
'success': status == 'connected',
|
||||
'status': status,
|
||||
'message': 'Sonarr connector ready.' if status == 'connected' else redact_message(str(result.get('reason', 'connection_failed'))),
|
||||
'version': str(result.get('version', 'unknown')),
|
||||
'url': _safe_url(),
|
||||
}
|
||||
|
||||
def search_series(self, query: str) -> dict[str, Any]:
|
||||
if not self.is_configured():
|
||||
return {'success': False, 'status': 'missing_config', 'message': 'missing_config', 'items': []}
|
||||
|
||||
try:
|
||||
payload = self._request('GET', '/api/v3/series/lookup', params={'term': query})
|
||||
except RuntimeError as exc:
|
||||
return {'success': False, 'status': 'error', 'message': redact_message(str(exc)), 'items': []}
|
||||
|
||||
if not isinstance(payload, list):
|
||||
return {'success': False, 'status': 'error', 'message': 'unexpected_payload', 'items': []}
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'status': 'connected',
|
||||
'items': [self._map_series(item) for item in payload if isinstance(item, dict)],
|
||||
}
|
||||
|
||||
def queue_status(self) -> dict[str, Any]:
|
||||
if not self.is_configured():
|
||||
return self._empty_queue('missing_config')
|
||||
|
||||
try:
|
||||
payload = self._request('GET', '/api/v3/queue')
|
||||
except RuntimeError as exc:
|
||||
result = self._empty_queue('error')
|
||||
result['message'] = redact_message(str(exc))
|
||||
return result
|
||||
|
||||
records = payload.get('records', []) if isinstance(payload, dict) else []
|
||||
if not isinstance(records, list):
|
||||
records = []
|
||||
|
||||
items = [self._map_queue_item(item) for item in records if isinstance(item, dict)]
|
||||
return {
|
||||
'success': True,
|
||||
'status': 'connected',
|
||||
'total': int(payload.get('totalRecords', len(items))) if isinstance(payload, dict) else len(items),
|
||||
'downloading': sum(1 for item in items if item['status'] == 'downloading'),
|
||||
'queued': sum(1 for item in items if item['status'] == 'queued'),
|
||||
'failed': sum(1 for item in items if item['trackedDownloadStatus'] == 'error' or item['status'] == 'failed'),
|
||||
'stalled': sum(1 for item in items if item['status'] == 'stalled'),
|
||||
'items': items,
|
||||
}
|
||||
|
||||
def wanted_missing(self) -> dict[str, Any]:
|
||||
if not self.is_configured():
|
||||
return {'success': False, 'status': 'missing_config', 'message': 'missing_config', 'total': 0, 'items': []}
|
||||
|
||||
try:
|
||||
payload = self._request('GET', '/api/v3/wanted/missing', params={'page': 1, 'pageSize': 20})
|
||||
except RuntimeError as exc:
|
||||
return {'success': False, 'status': 'error', 'message': redact_message(str(exc)), 'total': 0, 'items': []}
|
||||
|
||||
records = payload.get('records', []) if isinstance(payload, dict) else []
|
||||
if not isinstance(records, list):
|
||||
records = []
|
||||
return {
|
||||
'success': True,
|
||||
'status': 'connected',
|
||||
'total': int(payload.get('totalRecords', len(records))) if isinstance(payload, dict) else len(records),
|
||||
'items': records[:20],
|
||||
}
|
||||
|
||||
def add_series(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
config = self._config()
|
||||
if not config:
|
||||
return {'success': False, 'status': 'missing_config', 'message': 'missing_config'}
|
||||
|
||||
title = str(payload.get('title') or '').strip()
|
||||
tvdb_id = _as_int(payload.get('tvdbId') or payload.get('tvdb_id'))
|
||||
if not title or not tvdb_id:
|
||||
return {'success': False, 'status': 'error', 'message': 'title_and_tvdbId_required'}
|
||||
|
||||
root_folder = str(payload.get('rootFolderPath') or config.default_root_folder or '').strip()
|
||||
quality_profile_id = _as_int(payload.get('qualityProfileId'), config.default_quality_profile or 0)
|
||||
if not root_folder or not quality_profile_id:
|
||||
return {'success': False, 'status': 'missing_config', 'message': 'missing_defaults'}
|
||||
|
||||
request_payload = {
|
||||
**payload,
|
||||
'title': title,
|
||||
'tvdbId': tvdb_id,
|
||||
'rootFolderPath': root_folder,
|
||||
'qualityProfileId': quality_profile_id,
|
||||
'monitored': _as_bool(payload.get('monitored'), True),
|
||||
'seasonFolder': _as_bool(payload.get('seasonFolder'), True),
|
||||
'searchForMissingEpisodes': _as_bool(payload.get('searchForMissingEpisodes'), config.search_on_add),
|
||||
'addOptions': {
|
||||
'searchForMissingEpisodes': _as_bool(payload.get('searchForMissingEpisodes'), config.search_on_add),
|
||||
},
|
||||
}
|
||||
|
||||
try:
|
||||
result = self._request('POST', '/api/v3/series', json=request_payload)
|
||||
except RuntimeError as exc:
|
||||
return {'success': False, 'status': 'error', 'message': redact_message(str(exc))}
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'status': 'connected',
|
||||
'message': 'series_added',
|
||||
'data': result if isinstance(result, dict) else {},
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _map_series(raw: dict[str, Any]) -> dict[str, Any]:
|
||||
return {
|
||||
'id': _as_int(raw.get('id')),
|
||||
'tvdbId': _as_int(raw.get('tvdbId')),
|
||||
'title': str(raw.get('title', '')),
|
||||
'year': _as_int(raw.get('year')),
|
||||
'overview': str(raw.get('overview', '')),
|
||||
'status': str(raw.get('status', 'unknown')),
|
||||
'remotePoster': str(raw.get('remotePoster', '')),
|
||||
'existing': bool(raw.get('id') and raw.get('path')),
|
||||
'monitored': _as_bool(raw.get('monitored'), False),
|
||||
'seasons': raw.get('seasons') if isinstance(raw.get('seasons'), list) else [],
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _map_queue_item(raw: dict[str, Any]) -> dict[str, Any]:
|
||||
return {
|
||||
'id': raw.get('id'),
|
||||
'title': str(raw.get('title', raw.get('series', {}).get('title', 'unknown'))),
|
||||
'status': str(raw.get('status', 'unknown')).lower(),
|
||||
'trackedDownloadStatus': str(raw.get('trackedDownloadStatus', 'unknown')).lower(),
|
||||
'size': raw.get('size'),
|
||||
'sizeleft': raw.get('sizeleft'),
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _empty_queue(status: str) -> dict[str, Any]:
|
||||
return {
|
||||
'success': False,
|
||||
'status': status,
|
||||
'message': status,
|
||||
'total': 0,
|
||||
'downloading': 0,
|
||||
'queued': 0,
|
||||
'failed': 0,
|
||||
'stalled': 0,
|
||||
'items': [],
|
||||
}
|
||||
Reference in new issue
Block a user