You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
3.5 KiB
77 lines
3.5 KiB
import logging
|
|
|
|
from modules.util import Failed
|
|
|
|
logger = logging.getLogger("Plex Meta Manager")
|
|
|
|
base_url = "https://notifiarr.com/api/v1/"
|
|
dev_url = "https://dev.notifiarr.com/api/v1/"
|
|
|
|
class NotifiarrBase:
|
|
def __init__(self, config, apikey, develop, test, error_notification):
|
|
self.config = config
|
|
self.apikey = apikey
|
|
self.develop = develop
|
|
self.test = test
|
|
self.error_notification = error_notification
|
|
|
|
def _request(self, path, json=None, params=None):
|
|
url = f"{dev_url if self.develop else base_url}" + \
|
|
("notification/test" if self.test else f"{path}{self.apikey}")
|
|
logger.debug(url.replace(self.apikey, "APIKEY"))
|
|
response = self.config.get(url, json=json, params={"event": "pmm"} if self.test else params)
|
|
response_json = response.json()
|
|
if self.develop or self.test:
|
|
logger.debug(json)
|
|
logger.debug("")
|
|
logger.debug(response_json)
|
|
if response.status_code >= 400 or ("result" in response_json and response_json["result"] == "error"):
|
|
raise Failed(f"({response.status_code} [{response.reason}]) {response_json}")
|
|
return response_json
|
|
|
|
def error(self, text, library=None, collection=None, critical=True):
|
|
if self.error_notification:
|
|
json = {"error": str(text), "critical": critical}
|
|
if library:
|
|
json["server_name"] = library.PlexServer.friendlyName
|
|
json["library_name"] = library.name
|
|
if collection:
|
|
json["collection"] = str(collection)
|
|
self._request("notification/plex/", json=json, params={"event": "collections"})
|
|
|
|
class NotifiarrFactory(NotifiarrBase):
|
|
def __init__(self, config, params):
|
|
super().__init__(config, params["apikey"], params["develop"], params["test"], params["error_notification"])
|
|
if not params["test"] and not self._request("user/validate/")["details"]["response"]:
|
|
raise Failed("Notifiarr Error: Invalid apikey")
|
|
|
|
def getNotifiarr(self, library):
|
|
return Notifiarr(self.config, library, self.apikey, self.develop, self.test, self.error_notification)
|
|
|
|
class Notifiarr(NotifiarrBase):
|
|
def __init__(self, config, library, apikey, develop, test, error_notification):
|
|
super().__init__(config, apikey, develop, test, error_notification)
|
|
self.library = library
|
|
|
|
def plex_collection(self, collection, created=False, additions=None, removals=None):
|
|
thumb = None
|
|
if collection.thumb and next((f for f in collection.fields if f.name == "thumb"), None):
|
|
thumb = self.config.get_image_encoded(f"{self.library.url}{collection.thumb}?X-Plex-Token={self.library.token}")
|
|
art = None
|
|
if collection.art and next((f for f in collection.fields if f.name == "art"), None):
|
|
art = self.config.get_image_encoded(f"{self.library.url}{collection.art}?X-Plex-Token={self.library.token}")
|
|
json = {
|
|
"server_name": self.library.PlexServer.friendlyName,
|
|
"library_name": self.library.name,
|
|
"type": "movie" if self.library.is_movie else "show",
|
|
"collection": collection.title,
|
|
"created": created,
|
|
"poster": thumb,
|
|
"background": art
|
|
}
|
|
if additions:
|
|
json["additions"] = additions
|
|
if removals:
|
|
json["removals"] = removals
|
|
self._request("notification/plex/", json=json, params={"event": "collections"})
|