58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
from typing import Any, Dict, Optional
|
|
from .base import ApiClient
|
|
|
|
|
|
class RadarrClient(ApiClient):
|
|
async def get_system_status(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/system/status")
|
|
|
|
async def get_movie_by_tmdb_id(self, tmdb_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/movie", params={"tmdbId": tmdb_id})
|
|
|
|
async def get_movies(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/movie")
|
|
|
|
async def get_root_folders(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/rootfolder")
|
|
|
|
async def get_quality_profiles(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/qualityprofile")
|
|
|
|
async def get_queue(self, movie_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/queue", params={"movieId": movie_id})
|
|
|
|
async def get_indexers(self) -> Optional[Dict[str, Any]]:
|
|
return await self.get("/api/v3/indexer")
|
|
|
|
async def search(self, movie_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.post("/api/v3/command", payload={"name": "MoviesSearch", "movieIds": [movie_id]})
|
|
|
|
async def add_movie(
|
|
self,
|
|
tmdb_id: int,
|
|
quality_profile_id: int,
|
|
root_folder: str,
|
|
monitored: bool = True,
|
|
search_for_movie: bool = True,
|
|
) -> Optional[Dict[str, Any]]:
|
|
payload = {
|
|
"tmdbId": tmdb_id,
|
|
"qualityProfileId": quality_profile_id,
|
|
"rootFolderPath": root_folder,
|
|
"monitored": monitored,
|
|
"addOptions": {"searchForMovie": search_for_movie},
|
|
}
|
|
return await self.post("/api/v3/movie", payload=payload)
|
|
|
|
async def grab_release(self, guid: str, indexer_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.post("/api/v3/release", payload={"guid": guid, "indexerId": indexer_id})
|
|
|
|
async def push_release(self, payload: Dict[str, Any]) -> Optional[Dict[str, Any]]:
|
|
return await self.post("/api/v3/release/push", payload=payload)
|
|
|
|
async def download_release(self, guid: str, indexer_id: int) -> Optional[Dict[str, Any]]:
|
|
return await self.post(
|
|
"/api/v3/command",
|
|
payload={"name": "DownloadRelease", "guid": guid, "indexerId": indexer_id},
|
|
)
|