meisnate12
10 months ago
parent
fe9d0749bd
commit
33b92f2a2d
@ -1,42 +1,99 @@
|
|||||||
from json import JSONDecodeError
|
from json import JSONDecodeError
|
||||||
from modules import util
|
from modules import util
|
||||||
from modules.util import Failed
|
from modules.util import Failed
|
||||||
from retrying import retry
|
|
||||||
|
|
||||||
logger = util.logger
|
logger = util.logger
|
||||||
|
|
||||||
class Gotify:
|
class Gotify:
|
||||||
def __init__(self, config, params):
|
def __init__(self, config, params):
|
||||||
self.config = config
|
self.config = config
|
||||||
self.apikey = params["apikey"]
|
self.token = params["token"]
|
||||||
self.url = params["url"]
|
self.url = params["url"].rstrip("/")
|
||||||
logger.secret(self.apikey)
|
logger.secret(self.url)
|
||||||
|
logger.secret(self.token)
|
||||||
try:
|
try:
|
||||||
self.request(path="message")
|
logger.info(f"Gotify Version: {self._request(path='version', post=False)['version']}")
|
||||||
except JSONDecodeError:
|
except Exception:
|
||||||
raise Failed("Gotify Error: Invalid JSON response received")
|
logger.stacktrace()
|
||||||
|
raise Failed("Gotify Error: Invalid URL")
|
||||||
|
|
||||||
def notification(self, json):
|
def _request(self, path="message", json=None, post=True):
|
||||||
return self.request(json=json)
|
if post:
|
||||||
|
response = self.config.post(f"{self.url}/{path}", headers={"X-Gotify-Key": self.token}, json=json)
|
||||||
@retry(stop_max_attempt_number=6, wait_fixed=10000, retry_on_exception=util.retry_if_not_failed)
|
else:
|
||||||
def request(self, json=None, path="message"):
|
response = self.config.get(f"{self.url}/{path}")
|
||||||
if not json:
|
|
||||||
json = {
|
|
||||||
"message": "Well hello there.",
|
|
||||||
"priority": 1,
|
|
||||||
"title": "This is first contact"
|
|
||||||
}
|
|
||||||
response = self.config.post(f"{self.url}{path}?token={self.apikey}", json=json)
|
|
||||||
try:
|
try:
|
||||||
response_json = response.json()
|
response_json = response.json()
|
||||||
except JSONDecodeError as e:
|
except JSONDecodeError as e:
|
||||||
logger.error(response.content)
|
logger.error(response.content)
|
||||||
logger.debug(e)
|
logger.debug(e)
|
||||||
raise e
|
raise e
|
||||||
if response.status_code >= 400 or ("result" in response_json and response_json["result"] == "error"):
|
if response.status_code >= 400:
|
||||||
logger.debug(f"Response: {response_json}")
|
raise Failed(f"({response.status_code} [{response.reason}]) {response_json['errorDescription']}")
|
||||||
raise Failed(f"({response.status_code} [{response.reason}]) {response_json}")
|
return response_json
|
||||||
if not response_json["id"]:
|
|
||||||
raise Failed("Gotify Error: Invalid apikey")
|
def notification(self, json):
|
||||||
return response
|
message = ""
|
||||||
|
if json["event"] == "run_end":
|
||||||
|
title = "Run Completed"
|
||||||
|
message = f"Start Time: {json['start_time']}\n" \
|
||||||
|
f"End Time: {json['end_time']}\n" \
|
||||||
|
f"Run Time: {json['run_time']}\n" \
|
||||||
|
f"Collections Created: {json['collections_created']}\n" \
|
||||||
|
f"Collections Modified: {json['collections_modified']}\n" \
|
||||||
|
f"Collections Deleted: {json['collections_deleted']}\n" \
|
||||||
|
f"Items Added: {json['items_added']}\n" \
|
||||||
|
f"Items Removed: {json['items_removed']}"
|
||||||
|
if json["added_to_radarr"]:
|
||||||
|
message += f"\n{json['added_to_radarr']} Movies Added To Radarr"
|
||||||
|
if json["added_to_sonarr"]:
|
||||||
|
message += f"\n{json['added_to_sonarr']} Movies Added To Sonarr"
|
||||||
|
elif json["event"] == "run_start":
|
||||||
|
title = "Run Started"
|
||||||
|
message = json["start_time"]
|
||||||
|
elif json["event"] == "version":
|
||||||
|
title = "New Version Available"
|
||||||
|
message = f"Current: {json['current']}\n" \
|
||||||
|
f"Latest: {json['latest']}\n" \
|
||||||
|
f"Notes: {json['notes']}"
|
||||||
|
elif json["event"] == "delete":
|
||||||
|
if "library_name" in json:
|
||||||
|
title = "Collection Deleted"
|
||||||
|
else:
|
||||||
|
title = "Playlist Deleted"
|
||||||
|
message = json["message"]
|
||||||
|
else:
|
||||||
|
new_line = "\n"
|
||||||
|
if "server_name" in json:
|
||||||
|
message += f"{new_line if message else ''}Server: {json['server_name']}"
|
||||||
|
if "library_name" in json:
|
||||||
|
message += f"{new_line if message else ''}Library: {json['library_name']}"
|
||||||
|
if "collection" in json:
|
||||||
|
message += f"{new_line if message else ''}Collection: {json['collection']}"
|
||||||
|
if "playlist" in json:
|
||||||
|
message += f"{new_line if message else ''}Playlist: {json['playlist']}"
|
||||||
|
if json["event"] == "error":
|
||||||
|
if "collection" in json:
|
||||||
|
title_name = "Collection"
|
||||||
|
elif "playlist" in json:
|
||||||
|
title_name = "Playlist"
|
||||||
|
elif "library_name" in json:
|
||||||
|
title_name = "Library"
|
||||||
|
else:
|
||||||
|
title_name = "Global"
|
||||||
|
title = f"{'Critical ' if json['critical'] else ''}{title_name} Error"
|
||||||
|
message += f"{new_line if message else ''}Error Message: {json['error']}"
|
||||||
|
else:
|
||||||
|
title = f"{'Collection' if 'collection' in json else 'Playlist'} {'Created' if json['created'] else 'Modified'}"
|
||||||
|
if json['radarr_adds']:
|
||||||
|
message += f"{new_line if message else ''}{len(json['radarr_adds'])} Radarr Additions:"
|
||||||
|
if json['sonarr_adds']:
|
||||||
|
message += f"{new_line if message else ''}{len(json['sonarr_adds'])} Sonarr Additions:"
|
||||||
|
message += f"{new_line if message else ''}{len(json['additions'])} Additions:"
|
||||||
|
for add_dict in json['additions']:
|
||||||
|
message += f"\n{add_dict['title']}"
|
||||||
|
message += f"{new_line if message else ''}{len(json['removals'])} Removals:"
|
||||||
|
for add_dict in json['removals']:
|
||||||
|
message += f"\n{add_dict['title']}"
|
||||||
|
|
||||||
|
self._request(json={"message": message, "title": title})
|
||||||
|
|||||||
Loading…
Reference in new issue