Eric Jungbauer
20a8987dd9
Adds 9 read-only tools across three new services: - Lidarr: queue, artist_search, library_stats (music library awareness) - Whisparr: queue, series_search, library_stats (adult content, v3 API) - Overseerr: search, requests, request_counts (cross-library availability, request tracking via X-Api-Key header reusing arr_get helper) Fix _common.py urlencode to use %20 instead of + for query values — Overseerr rejects + as reserved. Safe for all arr services. Pirate catalog: 11 → 20 tools.
73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
"""Shared helpers for pirate tools."""
|
|
|
|
import sys
|
|
import os
|
|
from urllib import request as urlreq, error as urlerror, parse as urlparse
|
|
import json
|
|
|
|
# Add parent agents/ to path so we can import shared.py
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
|
from shared import DASHBOARD_API, api_request
|
|
|
|
|
|
def get_service_config(service_name):
|
|
"""Fetch admin-configured creds for a service (sonarr, radarr, qbittorrent, etc.)."""
|
|
try:
|
|
return api_request(f"{DASHBOARD_API}/api/internal/services/{service_name}", retries=1)
|
|
except Exception as e:
|
|
raise RuntimeError(f"Service '{service_name}' is not configured in admin panel: {e}")
|
|
|
|
|
|
def arr_get(service_name, path, params=None):
|
|
"""GET from an *arr API (Sonarr/Radarr/Lidarr/Whisparr — all use the same v3 API shape)."""
|
|
cfg = get_service_config(service_name)
|
|
base = cfg["base_url"].rstrip("/")
|
|
key = cfg["api_key"]
|
|
if not base or not key:
|
|
raise RuntimeError(f"{service_name} not configured (base_url + api_key required)")
|
|
url = f"{base}{path}"
|
|
if params:
|
|
url += "?" + urlparse.urlencode(params, quote_via=urlparse.quote)
|
|
req = urlreq.Request(url, headers={"X-Api-Key": key})
|
|
with urlreq.urlopen(req, timeout=15) as resp:
|
|
return json.loads(resp.read().decode())
|
|
|
|
|
|
def qbit_request(path, method="GET", params=None, data=None, cookies=None):
|
|
"""qBittorrent Web API helper. Login (if password set) returns SID cookie string."""
|
|
cfg = get_service_config("qbittorrent")
|
|
base = cfg["base_url"].rstrip("/")
|
|
if not base:
|
|
raise RuntimeError("qbittorrent not configured")
|
|
url = f"{base}{path}"
|
|
if params:
|
|
url += "?" + urlparse.urlencode(params, quote_via=urlparse.quote)
|
|
body = None
|
|
headers = {}
|
|
if cookies:
|
|
headers["Cookie"] = cookies
|
|
if data is not None:
|
|
body = urlparse.urlencode(data).encode()
|
|
headers["Content-Type"] = "application/x-www-form-urlencoded"
|
|
req = urlreq.Request(url, data=body, headers=headers, method=method)
|
|
with urlreq.urlopen(req, timeout=15) as resp:
|
|
raw = resp.read().decode()
|
|
try:
|
|
return json.loads(raw), resp.headers
|
|
except json.JSONDecodeError:
|
|
return raw, resp.headers
|
|
|
|
|
|
def qbit_login_if_needed():
|
|
"""If qBit has auth enabled, login and return the SID cookie string. Else return None."""
|
|
cfg = get_service_config("qbittorrent")
|
|
user = cfg.get("username", "")
|
|
pw = cfg.get("password", "")
|
|
if not user or not pw:
|
|
return None # LAN no-auth mode
|
|
_, hdrs = qbit_request("/api/v2/auth/login", method="POST",
|
|
data={"username": user, "password": pw})
|
|
cookies = hdrs.get("Set-Cookie", "")
|
|
# Strip everything after the first semicolon (path/expiry)
|
|
return cookies.split(";")[0] if cookies else None
|