Initial commit for backup migration.

Kind of broken :/

app/classes/models/servers.py

@@ -26,7 +26,6 @@ class Servers(BaseModel):
     created = DateTimeField(default=datetime.datetime.now)
     server_name = CharField(default="Server", index=True)
     path = CharField(default="")
-    backup_path = CharField(default="")
     executable = CharField(default="")
     log_path = CharField(default="")
     execution_command = CharField(default="")
@@ -65,7 +64,6 @@ class HelperServers:
         server_id: str,
         name: str,
         server_dir: str,
-        backup_path: str,
         server_command: str,
         server_file: str,
         server_log_file: str,
@@ -81,7 +79,6 @@ class HelperServers:
             name: The name of the server
             server_uuid: This is the UUID of the server
             server_dir: The directory where the server is located
-            backup_path: The path to the backup folder
             server_command: The command to start the server
             server_file: The name of the server file
             server_log_file: The path to the server log file
@@ -111,7 +108,6 @@ class HelperServers:
             server_port=server_port,
             server_ip=server_host,
             stop_command=server_stop,
-            backup_path=backup_path,
             type=server_type,
             created_by=created_by,
         ).server_id

app/migrations/20240308_multi-backup.py

@@ -0,0 +1,105 @@
+import datetime
+import uuid
+import peewee
+import logging
+
+from app.classes.shared.console import Console
+from app.classes.shared.migration import Migrator, MigrateHistory
+from app.classes.models.management import Backups
+
+logger = logging.getLogger(__name__)
+
+
+def migrate(migrator: Migrator, database, **kwargs):
+    """
+    Write your migrations here.
+    """
+    print("pee pee")
+    db = database
+
+    migrator.add_columns("backups", backup_id=peewee.UUIDField(default=uuid.uuid4))
+    migrator.add_columns("backups", backup_name=peewee.CharField(default="Default"))
+    migrator.add_columns("backups", backup_location=peewee.CharField(default=""))
+
+    class Servers(peewee.Model):
+        server_id = peewee.CharField(primary_key=True, default=str(uuid.uuid4()))
+        created = peewee.DateTimeField(default=datetime.datetime.now)
+        server_name = peewee.CharField(default="Server", index=True)
+        path = peewee.CharField(default="")
+        backup_path = peewee.CharField(default="")
+        executable = peewee.CharField(default="")
+        log_path = peewee.CharField(default="")
+        execution_command = peewee.CharField(default="")
+        auto_start = peewee.BooleanField(default=0)
+        auto_start_delay = peewee.IntegerField(default=10)
+        crash_detection = peewee.BooleanField(default=0)
+        stop_command = peewee.CharField(default="stop")
+        executable_update_url = peewee.CharField(default="")
+        server_ip = peewee.CharField(default="127.0.0.1")
+        server_port = peewee.IntegerField(default=25565)
+        logs_delete_after = peewee.IntegerField(default=0)
+        type = peewee.CharField(default="minecraft-java")
+        show_status = peewee.BooleanField(default=1)
+        created_by = peewee.IntegerField(default=-100)
+        shutdown_timeout = peewee.IntegerField(default=60)
+        ignored_exits = peewee.CharField(default="0")
+
+        class Meta:
+            table_name = "servers"
+            database = db
+
+    class NewBackups(peewee.Model):
+        backup_id = peewee.UUIDField(primary_key=True, default=uuid.uuid4)
+        backup_name = peewee.CharField(default="New Backup")
+        backup_location = peewee.CharField(default="")
+        excluded_dirs = peewee.CharField(null=True)
+        max_backups = peewee.IntegerField()
+        server_id = peewee.ForeignKeyField(Servers, backref="backups_server")
+        compress = peewee.BooleanField(default=False)
+        shutdown = peewee.BooleanField(default=False)
+        before = peewee.CharField(default="")
+        after = peewee.CharField(default="")
+
+        class Meta:
+            table_name = "new_backups"
+            database = db
+
+    migrator.create_table(NewBackups)
+
+    migrator.run()
+
+    # Copy data from the existing backups table to the new one
+    for backup in Backups.select():
+        print(backup)
+        # Fetch the related server entry from the Servers table
+        server = Servers.get(Servers.server_id == backup.server_id)
+
+        # Create a new backup entry with data from the old backup entry and related server
+        NewBackups.create(
+            backup_name="Default",
+            backup_location=server.backup_path,  # Set backup_location equal to backup_path
+            excluded_dirs=backup.excluded_dirs,
+            max_backups=backup.max_backups,
+            server_id=server.server_id,
+            compress=backup.compress,
+            shutdown=backup.shutdown,
+            before=backup.before,
+            after=backup.after,
+        )
+
+    # Drop the existing backups table
+    migrator.drop_table("backups")
+
+    # Rename the new table to backups
+    migrator.rename_table("new_backups", "backups")
+    migrator.drop_columns("servers", ["backup_path"])
+
+
+def rollback(migrator: Migrator, database, **kwargs):
+    """
+    Write your rollback migrations here.
+    """
+    db = database
+
+    migrator.drop_columns("backups", ["name", "backup_id", "backup_location"])
+    migrator.add_columns("servers", backup_path=peewee.CharField(default=""))