Plex-Meta-Manager/modules/notifiarr.py

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