Starting work on backup system

modoboa_installer/scripts/backup.py

@@ -0,0 +1,114 @@
+"""Backup script for pre-installed instance"""
+
+import shutil
+import utils
+import os
+
+#TODO: have version of each modoboa componenent saved into the config file to restore the same version  
+
+class Backup():
+
+
+    def __init__(self, config):
+        self.config = config
+        self.destinationPath = ""
+        self.BACKUPDIRECTORY = ["mails", "custom", "databases"]
+
+
+    def preparePath(self):
+        for dir in self.BACKUPDIRECTORY:
+            os.mkdir(self.destinationPath + dir)
+
+
+    def validatePath(self, path):
+        """Check basic condition for backup directory"""
+
+        if os.path.isfile(path):
+            print("Error, you provided a file instead of a directory!")
+            return False
+
+        if not os.path.exists(path):
+            createDir = input(f"\"{path}\" doesn't exists, would you like to create it ? [Y/n]\n").lower()
+
+            if createDir == "y" or createDir == "yes":
+                os.mkdir(path)
+            else:
+                return False
+
+        if len(os.listdir(path)) != 0:
+            delDir = input("Warning : backup folder is not empty, it will be purged if you continue... [Y/n]").lower()
+            if delDir == "y" or delDir == "yes":
+                shutil.rmtree(path)
+            else:
+                return False
+
+        self.destinationPath = path
+
+        if self.destinationPath[-1] != "/":
+            self.destinationPath += "/"
+
+        self.preparePath()
+        return True
+
+
+    def setPath(self):
+        """Setup backup directory"""
+        user_value = None
+        while (user_value != '' and not self.validatePath(user_value)):
+            print("Enter backup path, please provide an empty folder.")
+            user_value = utils.user_input("-> ")
+    
+
+    def backupConfigFile(self):
+        utils.copy_file("installer.cfg", self.destinationPath)
+
+
+    def backupMails(self):
+
+        utils.printcolor("Backing up mails", utils.MAGENTA)
+
+        home_path = self.config.get("dovecot", "home_dir")
+
+        if not os.path.exists(home_path) or os.path.isfile(home_path):
+            utils.printcolor("Error backing up Email, provided path "
+            f" ({home_path}) seems not right...", utils.RED)
+        
+        else:
+            shutil.copytree(home_path, self.destinationPath+"mails/")
+            utils.printcolor("Mail backup complete!", utils.GREEN)
+
+
+    def backupCustomConfig(self):
+        """Custom config :
+        - Amavis : /etc/amavis/conf.d/99-custom
+        - Postscreen : /etc/postfix/custom_whitelist.cidr
+        Feel free to suggest to add others!"""
+        utils.printcolor("Backing up some custom configuration...", utils.MAGENTA)
+
+        custom_path = self.destinationPath+"custom/"
+
+        """AMAVIS"""
+        amavis_custom = "/etc/amavis/conf.d/99-custom"
+        if os.path.isfile(amavis_custom):
+            utils.copy_file(amavis_custom, custom_path)
+            utils.printcolor("Amavis custom configuration saved!", utils.GREEN)
+
+        """POSTSCREEN"""
+        postscreen_custom = "/etc/postfix/custom_whitelist.cidr"
+        if os.path.isfile(postscreen_custom):
+            utils.copy_file(postscreen_custom, custom_path)
+            utils.printcolor("Postscreen whitelist custom configuration saved!", utils.GREEN)
+
+
+    def backupDBs(self):
+        """Backing up databases"""
+
+
+    def run(self):
+        self.setPath()
+        self.backupConfigFile()
+        self.backupMails()
+        self.backupCustomConfig()
+        self.backupDBs()
+
+