Add check_success.py

src/check_success.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+import requests
+import os
+
+import config
+
+
+def raiseWebhook(message, file=None):
+    data = {
+        "username": "Backup script",
+        "embeds": 
+        [{
+            "title": "Erreur lors de la sauvegarde",
+            "description": message,
+            "color": color_red,
+            "fields": []
+        }]
+    }
+
+    if file is not None:
+        data[files]={"file": open(file, "rb")}
+
+    requests.post(config.ERROR_WEBHOOK, json=data)
+
+with open("latest-success", "r") as f:
+    latest_success = f.read().strip()
+
+today = datetime.today().strftime('%Y-%m-%d')
+if latest_success != today:
+    log_file = None
+    if os.path.exists(f"logs/{today}.txt"):
+        log_file = f"logs/{today}.txt"
+
+    raiseWebhook(f"No successful backup today", file=log_file)

src/example_config.py

@@ -8,3 +8,5 @@ REMOTE_USER = "myuser"
 BACKUP_MAP = [
     ("/home/remote/MyData", "backup-it-here") # myuser@192.168.1.5:/home/remote/MyData will be backed up to /mnt/{backup-date and "latest"}/backup-it-here
 ]
+
+ERROR_WEBHOOK = "https://discord.com/api/webhooks/...."

src/main.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 import subprocess
 import requests
+import sys
 import os
 
 import config
@@ -9,9 +10,19 @@ def log(*args, **kwargs):
     date = datetime.now().strftime('%d-%b-%Y %H:%M:%S')
     print(f"[{date}]", *args, **kwargs)
 
+def log_error(*args, **kwargs):
+    date = datetime.now().strftime('%d-%b-%Y %H:%M:%S')
+    print(f"[{date}]", *args, **kwargs, file=sys.stderr)
+
+def check_return_code(ret, *args, **kwargs):
+    if ret != 0:
+        log_error(*args, **kwargs)
+        exit(1)
+
 
 if not os.path.exists(config.DISK):
     log_error(f"Drive {config.DISK} not present on the system")
+    exit(1)
 
 passwd = requests.get(config.PWD_URL).text.strip()
 log("Got password :", passwd)
@@ -20,26 +31,46 @@ ssh_key_passwd = requests.get(config.SSH_KEY_PWD_URL).text.strip()
 log("Got password :", ssh_key_passwd)
 
 
-subprocess.run([
+class Drive():
+    def __init__(self, disk, passwd, mount_location):
+        self.disk = disk
+        self.passwd = passwd
+        self.mount_location = mount_location
+
+    def __enter__(self):
+        subprocess.run([
             "cryptsetup", "luksOpen",
-    config.DISK, "backup"
+            self.disk, "backup"
-], input=passwd, text=True)
+        ], input=self.passwd, text=True)
 
-subprocess.call([
+        subprocess.call([
             "mount", "/dev/mapper/backup",
-    config.MNT_LOCATION
+            self.mount_location
-])
+        ])
+        return self
 
-log("Successfully mounted, doing the backup stuff")
+    def __exit__(self, exception_type, exception_value, exception_traceback):
+        subprocess.call([
+            "umount", self.mount_location
+        ])
 
-os.makedirs(
+        subprocess.call([
+            "cryptsetup", "luksClose",
+            "/dev/mapper/backup"
+        ])
+
+
+with Drive(config.DISK, passwd, config.MNT_LOCATION) as backup_drive:
+    log("Successfully mounted, doing the backup stuff")
+
+    os.makedirs(
         os.path.join(config.MNT_LOCATION, "latest"),
         exist_ok=True
-)
+    )
 
-for src, dest in config.BACKUP_MAP:
+    for src, dest in config.BACKUP_MAP:
         # We could also use a rsync daemon on the remote machine for pwd(-less) auth
-    subprocess.call(
+        ret = subprocess.call(
             [
                 "sshpass", "-P", "passphrase", "-e",
                     "rsync",
@@ -52,29 +83,25 @@ for src, dest in config.BACKUP_MAP:
             ],
             env={**os.environ, "SSHPASS":ssh_key_passwd}
         )
+        check_return_code(ret, f"Error when syncing {config.REMOTE_LOCATION}")
 
-log("Rsync ok. Creating snapshot.")
+    log("Rsync ok. Creating snapshot.")
 
-subprocess.call([
+    ret = subprocess.call([
         "cp", "-a",
         "--reflink=always", #! The partition must be Btrfs or XFS
         os.path.join(config.MNT_LOCATION, "latest"),
         os.path.join(config.MNT_LOCATION, datetime.today().strftime('%Y-%m-%d'))
-])
+    ])
+    check_return_code(ret, f"Error when creating snapshot")
 
-log("Snapshot ok. Deleting unneeded backups.")
+    log("Snapshot ok. Deleting unneeded backups.")
 
-#TODO
+    #TODO
 
-log("Unmounting.")
+    log("Unmounting.")
 
-subprocess.call([
+log("Everything okay, done !")
-    "umount", config.MNT_LOCATION
-])
 
-subprocess.call([
+with open("latest-success", "w") as f:
-    "cryptsetup", "luksClose",
+    f.write(datetime.today().strftime('%Y-%m-%d'))
-    "/dev/mapper/backup"
-])
-
-log("Done !")