126 lines
4.8 KiB
Python
126 lines
4.8 KiB
Python
from typing import Any, Dict, Optional
|
|
from urllib.parse import quote, unquote, urlsplit
|
|
import httpx
|
|
from .base import ApiClient
|
|
|
|
|
|
class JellyseerrClient(ApiClient):
|
|
async def _send_request(
|
|
self,
|
|
client: httpx.AsyncClient,
|
|
method: str,
|
|
url: str,
|
|
*,
|
|
headers: Dict[str, str],
|
|
params: Optional[Dict[str, Any]],
|
|
payload: Optional[Dict[str, Any]],
|
|
) -> httpx.Response:
|
|
request_headers = dict(headers)
|
|
if method.upper() in {"POST", "PUT", "PATCH", "DELETE"} and self.base_url:
|
|
# Seerr's optional CSRF protection also applies to API-key writes.
|
|
# Seed its secret/token cookie pair, then echo the readable token in
|
|
# the header Seerr's own web client uses.
|
|
csrf_response = await client.get(
|
|
f"{self.base_url}/api/v1/auth/me",
|
|
headers=self.headers(),
|
|
)
|
|
csrf_response.raise_for_status()
|
|
csrf_token = client.cookies.get("XSRF-TOKEN")
|
|
if csrf_token:
|
|
request_headers["XSRF-TOKEN"] = unquote(csrf_token)
|
|
parsed_base = urlsplit(self.base_url)
|
|
request_headers["Origin"] = f"{parsed_base.scheme}://{parsed_base.netloc}"
|
|
return await super()._send_request(
|
|
client,
|
|
method,
|
|
url,
|
|
headers=request_headers,
|
|
params=params,
|
|
payload=payload,
|
|
)
|
|
|
|
async def get_status(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v1/status")
|
|
|
|
async def get_request(self, request_id: str) -> Optional[Dict[str, Any]]:
|
|
return await self.get(f"/api/v1/request/{request_id}")
|
|
|
|
async def get_recent_requests(self, take: int = 10, skip: int = 0) -> Optional[Dict[str, Any]]:
|
|
return await self.get(
|
|
"/api/v1/request",
|
|
params={
|
|
"take": take,
|
|
"skip": skip,
|
|
},
|
|
)
|
|
|
|
async def get_movie(self, tmdb_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.get(f"/api/v1/movie/{tmdb_id}")
|
|
|
|
async def get_tv(self, tmdb_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.get(f"/api/v1/tv/{tmdb_id}")
|
|
|
|
async def search(self, query: str, page: int = 1) -> Optional[Dict[str, Any]]:
|
|
# Seerr rejects the `+` encoding that standard query builders use for
|
|
# spaces. Build this query explicitly so multi-word titles are sent as
|
|
# percent-encoded values.
|
|
encoded_query = quote(query, safe="")
|
|
return await self.get(f"/api/v1/search?query={encoded_query}&page={page}")
|
|
|
|
async def get_service_settings(self, media_type: str) -> Optional[Any]:
|
|
service = "sonarr" if media_type == "tv" else "radarr"
|
|
return await self.get(f"/api/v1/settings/{service}")
|
|
|
|
async def create_request(
|
|
self,
|
|
*,
|
|
media_type: str,
|
|
media_id: int,
|
|
seasons: Optional[list[int]] = None,
|
|
is_4k: Optional[bool] = None,
|
|
server_id: Optional[int] = None,
|
|
profile_id: Optional[int] = None,
|
|
root_folder: Optional[str] = None,
|
|
) -> Optional[Dict[str, Any]]:
|
|
payload: Dict[str, Any] = {
|
|
"mediaType": media_type,
|
|
"mediaId": media_id,
|
|
}
|
|
if isinstance(seasons, list) and seasons:
|
|
payload["seasons"] = seasons
|
|
if isinstance(is_4k, bool):
|
|
payload["is4k"] = is_4k
|
|
if isinstance(server_id, int):
|
|
payload["serverId"] = server_id
|
|
if isinstance(profile_id, int):
|
|
payload["profileId"] = profile_id
|
|
if isinstance(root_folder, str) and root_folder.strip():
|
|
payload["rootFolder"] = root_folder.strip()
|
|
return await self.post("/api/v1/request", payload=payload)
|
|
|
|
async def get_users(self, take: int = 50, skip: int = 0) -> Optional[Dict[str, Any]]:
|
|
return await self.get(
|
|
"/api/v1/user",
|
|
params={
|
|
"take": take,
|
|
"skip": skip,
|
|
},
|
|
)
|
|
|
|
async def get_user(self, user_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.get(f"/api/v1/user/{user_id}")
|
|
|
|
async def delete_user(self, user_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.delete(f"/api/v1/user/{user_id}")
|
|
|
|
async def login_local(self, email: str, password: str) -> Optional[Dict[str, Any]]:
|
|
payload = {"email": email, "password": password}
|
|
try:
|
|
return await self.post("/api/v1/auth/local", payload=payload)
|
|
except httpx.HTTPStatusError as exc:
|
|
# Backward compatibility for older Seerr/Overseerr deployments
|
|
# that still expose /auth/login instead of /auth/local.
|
|
if exc.response is not None and exc.response.status_code in {404, 405}:
|
|
return await self.post("/api/v1/auth/login", payload=payload)
|
|
raise
|