First working notify version
parent
ad3f49a785
commit
419fef8ddc
|
@ -1,4 +1,59 @@
|
||||||
|
from typing import List
|
||||||
|
import os
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
class Notifier:
|
class Notifier:
|
||||||
|
# (positive, negative)
|
||||||
|
_EVENTS = {
|
||||||
|
"backup": (
|
||||||
|
"Backup for {name} succeeded.",
|
||||||
|
"Backup for {name} failed.",
|
||||||
|
),
|
||||||
|
"restore": (
|
||||||
|
"{name} successfully restored.",
|
||||||
|
"Couldn't restore {name}.",
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
# Placeholder
|
# Placeholder
|
||||||
def __init__(*args, **kwargs):
|
def __init__(
|
||||||
pass
|
self, title: str, events: List[str], endpoint: str, api_key: str = None
|
||||||
|
):
|
||||||
|
self.title = title
|
||||||
|
self.events = events
|
||||||
|
self.endpoint = endpoint
|
||||||
|
self.api_key = api_key
|
||||||
|
|
||||||
|
def notify(self, category: str, name: str, status_code: int):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
category: type of notify (e.g. backup or restore)
|
||||||
|
name: name of the spec
|
||||||
|
status_code: exit code of the command
|
||||||
|
"""
|
||||||
|
|
||||||
|
event = "{}_{}".format(
|
||||||
|
category, "success" if status_code == 0 else "failure"
|
||||||
|
)
|
||||||
|
|
||||||
|
# stop if it's not a valid event
|
||||||
|
if event not in self._EVENTS:
|
||||||
|
return
|
||||||
|
|
||||||
|
api_key = self.api_key or os.environ["IFTTT_API_KEY"]
|
||||||
|
|
||||||
|
# Can't do anything without a key
|
||||||
|
if not api_key:
|
||||||
|
return
|
||||||
|
|
||||||
|
url = " https://maker.ifttt.com/trigger/{}/with/key/{}".format(
|
||||||
|
self.endpoint, api_key
|
||||||
|
)
|
||||||
|
|
||||||
|
data = {
|
||||||
|
"value1": self.title,
|
||||||
|
"value2": self._EVENTS[event][int(status_code != 0)],
|
||||||
|
}
|
||||||
|
|
||||||
|
requests.post(url, data=data)
|
||||||
|
|
|
@ -57,4 +57,4 @@ class DirectorySpec(Spec):
|
||||||
)
|
)
|
||||||
|
|
||||||
if self.notifier:
|
if self.notifier:
|
||||||
self.notifier.notify(process.returncode)
|
self.notifier.notify("backup", self.name, process.returncode)
|
||||||
|
|
|
@ -18,6 +18,8 @@ class Spec:
|
||||||
"notify": {
|
"notify": {
|
||||||
"title": "Backup Notification",
|
"title": "Backup Notification",
|
||||||
"events": ["backup_sucess"],
|
"events": ["backup_sucess"],
|
||||||
|
"endpoint": None,
|
||||||
|
"api_key": "",
|
||||||
},
|
},
|
||||||
"extension": "tar.gz",
|
"extension": "tar.gz",
|
||||||
}
|
}
|
||||||
|
@ -67,7 +69,7 @@ class Spec:
|
||||||
"""
|
"""
|
||||||
|
|
||||||
files = sorted(
|
files = sorted(
|
||||||
self.destination.glob(self.extension),
|
self.destination.glob("*." + self.extension),
|
||||||
key=os.path.getmtime,
|
key=os.path.getmtime,
|
||||||
reverse=True,
|
reverse=True,
|
||||||
)
|
)
|
||||||
|
@ -86,7 +88,6 @@ class Spec:
|
||||||
def from_dict(cls, name, obj: Dict, defaults: Dict) -> Spec:
|
def from_dict(cls, name, obj: Dict, defaults: Dict) -> Spec:
|
||||||
# Combine defaults with skeleton, creating new skeleton
|
# Combine defaults with skeleton, creating new skeleton
|
||||||
skel = skeleton.merge(cls.skeleton(), defaults)
|
skel = skeleton.merge(cls.skeleton(), defaults)
|
||||||
print(skel)
|
|
||||||
|
|
||||||
# Then, combine actual values with new skeleton
|
# Then, combine actual values with new skeleton
|
||||||
obj = skeleton.merge_with_skeleton(obj, skel)
|
obj = skeleton.merge_with_skeleton(obj, skel)
|
||||||
|
|
Loading…
Reference in New Issue