42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from typing import Union
|
|
|
|
import requests
|
|
|
|
|
|
class Overseerr(object):
|
|
def __init__(self, host: str, port: int, api_key: str, ssl: bool = True):
|
|
self.host = host
|
|
self.port = port
|
|
self.api_key = api_key
|
|
self.ssl = ssl
|
|
|
|
@property
|
|
def protocol(self):
|
|
return "https://" if self.ssl else "http://"
|
|
|
|
def _get(self, endpoint: str, params: dict = None):
|
|
url = self.protocol + self.host + ":" + str(self.port) + "/api/v1" + endpoint
|
|
return requests.get(url, headers={"X-Api-Key": self.api_key}, params=params)
|
|
|
|
def _post(self, endpoint: str, params: dict = None):
|
|
url = self.protocol + self.host + ":" + str(self.port) + "/api/v1" + endpoint
|
|
return requests.post(url, headers={"X-Api-Key": self.api_key}, json=params)
|
|
|
|
def test_connection(self):
|
|
print("Testing connection to Overseerr @", self.host)
|
|
settings = self._get("/settings/main").json()
|
|
if settings['applicationUrl'] == "":
|
|
return self.host
|
|
return settings['applicationUrl']
|
|
|
|
def get_tv_show(self, id):
|
|
return self._get(f"/tv/{id}").json()
|
|
|
|
def get_movie(self, id):
|
|
return self._get(f"/movie/{id}").json()
|
|
|
|
def update_request_status(self, req_id: int, status: Union["approve", "decline"]):
|
|
return self._post(f"/request/{req_id}/{status}").json()
|
|
|
|
def create_issue(self, message, media_id):
|
|
return self._post(f"/issue", params={"issueType": 0, "message": message, "mediaId": media_id}).json()
|