mirror of
https://github.com/StuffAnThings/qbit_manage.git
synced 2024-11-13 04:46:35 +08:00
42 lines
1.6 KiB
Python
Executable file
42 lines
1.6 KiB
Python
Executable file
import time
|
|
from json import JSONDecodeError
|
|
|
|
from modules import util
|
|
from modules.util import Failed
|
|
|
|
logger = util.logger
|
|
|
|
|
|
class Notifiarr:
|
|
"""Notifiarr API"""
|
|
|
|
BASE_URL = "https://notifiarr.com/api"
|
|
API_VERSION = "v1"
|
|
|
|
def __init__(self, config, params):
|
|
"""Initialize Notifiarr API"""
|
|
self.config = config
|
|
self.apikey = params["apikey"]
|
|
self.header = {"X-API-Key": self.apikey}
|
|
self.instance = params["instance"]
|
|
self.url = f"{self.BASE_URL}/{self.API_VERSION}/"
|
|
logger.secret(self.apikey)
|
|
response = self.config.get(f"{self.url}user/qbitManage/", headers=self.header, params={"fetch": "settings"})
|
|
response_json = None
|
|
try:
|
|
response_json = response.json()
|
|
except JSONDecodeError as e:
|
|
logger.debug(e)
|
|
raise Failed("Notifiarr Error: Invalid response")
|
|
if response.status_code >= 400 or ("result" in response_json and response_json["result"] == "error"):
|
|
logger.debug(f"Response: {response_json}")
|
|
raise Failed(f"({response.status_code} [{response.reason}]) {response_json}")
|
|
if not response_json["details"]["response"]:
|
|
raise Failed("Notifiarr Error: Invalid apikey")
|
|
|
|
def notification(self, json):
|
|
"""Send notification to Notifiarr"""
|
|
params = {"qbit_client": self.config.data["qbt"]["host"], "instance": self.instance}
|
|
response = self.config.get(f"{self.url}notification/qbitManage/", json=json, headers=self.header, params=params)
|
|
time.sleep(1) # Pause for 1 second before sending the next request
|
|
return response
|