from json import JSONDecodeError from modules import util from modules.util import Failed logger = util.logger base_url = "https://notifiarr.com/api/v1/" dev_url = "https://dev.notifiarr.com/api/v1/" class Notifiarr: def __init__(self, config, params): self.config = config self.apikey = params["apikey"] self.develop = params["develop"] self.test = params["test"] logger.secret(self.apikey) logger.debug(f"Environment: {'Test' if self.test else 'Develop' if self.develop else 'Production'}") url, _ = self.get_url("user/validate/") response = self.config.get(url) 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 params["test"] and not response_json["details"]["response"]: raise Failed("Notifiarr Error: Invalid apikey") def get_url(self, path): url = f"{dev_url if self.develop else base_url}{'notification/test' if self.test else f'{path}{self.apikey}'}" logger.trace(url) params = {"event": "pmm"} if self.test else None return url, params