Restore workflow done

modoboa_installer/scripts/modoboa.py

@@ -1,5 +1,6 @@
 """Modoboa related tasks."""
 
+from genericpath import isfile
 import json
 import os
 import pwd
@@ -176,6 +177,18 @@ class Modoboa(base.Installer):
         self.backend.grant_access(
             self.config.get("amavis", "dbname"), self.dbuser)
 
+
+    def get_sql_schema_path(self):
+        if self.restore:
+            utils.printcolor("Trying to restore modoboa database from backup", utils.MAGENTA)
+            modoboaDbBackupPath = self.restore + "databases/modoboa.sql"
+            if os.path.isfile(modoboaDbBackupPath):
+                utils.printcolor("Modoboa database backup found ! Restoring...", utils.GREEN)
+                return modoboaDbBackupPath
+            utils.printcolor("Modoboa database backup not found, creating empty database", utils.RED)
+
+        return super().get_sql_schema_path()()
+
     def get_packages(self):
         """Include extra packages if needed."""
         packages = super(Modoboa, self).get_packages()