80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
from typing import Any, Dict, Optional
|
|
import httpx
|
|
from .base import ApiClient
|
|
|
|
|
|
class JellyseerrClient(ApiClient):
|
|
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]]:
|
|
return await self.get(
|
|
"/api/v1/search",
|
|
params={
|
|
"query": query,
|
|
"page": page,
|
|
},
|
|
)
|
|
|
|
async def create_request(
|
|
self,
|
|
*,
|
|
media_type: str,
|
|
media_id: int,
|
|
seasons: Optional[list[int]] = None,
|
|
is_4k: Optional[bool] = 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
|
|
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
|