Added Command to Backup Databases

Author: yubarajshrestha

src/backup/Backup.py

@@ -6,6 +6,8 @@
 from shutil import make_archive
 from datetime import datetime
 from masonite.utils.location import base_path
+import subprocess
+import gzip
 
 
 class Backup:
@@ -23,7 +25,44 @@ def database(self):
         """
         Backup the database.
         """
-        pass
+        
+        db_config = config("database.databases")
+        default = db_config.get("default")
+        connection = db_config.get(default)
+        driver = connection.get("driver")
+
+        db_file_path = base_path("{}.gz".format("database-" + str(datetime.timestamp(datetime.now()))))
+
+        if driver == "sqlite":
+            pass
+        elif driver == "mysql":
+            command_str = f"mysqldump -u {connection.get('user')} -p{connection.get('password')} {connection.get('database')}"
+            
+        elif driver == "postgres":
+            command_str = f"pg_dump -U{connection.get('user')} -h{connection.get('host')} -p{connection.get('port')} -d{connection.get('database')}"
+
+        elif driver == "mssql":
+            command_str = f"sqlcmd -S{connection.get('host')} -U{connection.get('user')} -P{connection.get('port')} -d{connection.get('database')}"
+
+        elif driver == "sqlserver":
+            command_str = f"sqlcmd -S{connection.get('host')} -U{connection.get('user')} -P{connection.get('port')} -d{connection.get('database')}"
+
+        elif driver == "oracle":
+            command_str = f"sqlplus -S{connection.get('user')}/{connection.get('password')}@{connection.get('host')}:{connection.get('port')}/{connection.get('database')}"
+      
+        if command_str:
+            with gzip.open(db_file_path, "wb") as f:
+                popen = subprocess.Popen(
+                    [command_str],
+                    stdout=subprocess.PIPE,
+                    shell=True,
+                    universal_newlines=True,
+                )
+                for stdout_line in iter(popen.stdout.readline, ""):
+                    f.write(stdout_line.encode('utf-8'))
+                popen.stdout.close()
+                popen.wait()
+        return db_file_path
 
     def files(self):
         """
@@ -45,9 +84,10 @@ def files(self):
         with tempfile.TemporaryDirectory() as tmp:
             shutil.copytree(
                 self.backup_config.get("source").get("root"),
-                pathlib.Path(tmp).joinpath("temporary_backup"),
+                pathlib.Path(tmp).joinpath("backup"),
                 ignore=shutil.ignore_patterns(*self.backup_config.get("source").get("excludes")),
             )
 
             with patch("os.path.isfile", side_effect=self.accept):
                 make_archive(path_to_archive, "zip", tmp)
+        return path_to_archive

src/backup/commands/BackupRunCommand.py

@@ -1,3 +1,4 @@
+import os
 from masonite.commands import Command
 
 
@@ -19,14 +20,20 @@ def handle(self):
         if not self.validate_options():
             return
 
-        self.info("Starting backup...")
-
         if self.option("only-db"):
             self.app.make("backup").database()
         elif self.option("only-files"):
             self.app.make("backup").files()
+        else:
+            self.info("Backuping database...")
+            database_file_path = self.app.make("backup").database()
+            self.info("Backuping files...")
+            self.app.make("backup").files()
+
+            # delete the database file
+            os.remove(database_file_path)
 
-        self.info("Backup complete...")
+        self.info("============ Backup complete ============")
 
     def validate_options(self):
         if self.option("only-db") and self.option("only-files"):

src/backup/config/backup.py

@@ -11,8 +11,8 @@
 |
 """
 
-FILENAME = "backup"
-DIRECTORY = "backup"
+FILENAME = "backup" # The filename of the backup file. (without the extension)
+DIRECTORY = "backup" # storage/backup
 SOURCE = {
     "root": base_path(),
     "excludes": [