Adding migration to fix the rework servers uuid issue

app/migrations/20240310_fixing_rework_servers_uuid.py

@@ -0,0 +1,132 @@
+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 Schedules, Backups
+from app.classes.models.server_permissions import RoleServers
+
+logger = logging.getLogger(__name__)
+
+
+def migrate(migrator: Migrator, database, **kwargs):
+    """
+    Write your migrations here.
+    """
+    db = database
+
+    # **********************************************************************************
+    #          Servers New Model from Old (easier to migrate without dunmping Database)
+    # **********************************************************************************
+    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
+
+    try:
+        logger.info("Migrating Data from Int to UUID (Fixing Issue)")
+        Console.info("Migrating Data from Int to UUID (Fixing Issue)")
+
+        # Changes on Servers Roles Table
+        migrator.alter_column_type(
+            RoleServers,
+            "server_id",
+            peewee.ForeignKeyField(
+                Servers,
+                backref="role_server",
+                null=True,
+                field=peewee.CharField(primary_key=True, default=str(uuid.uuid4())),
+            ),
+        )
+
+        # Changes on Backups Table
+        migrator.alter_column_type(
+            Backups,
+            "server_id",
+            peewee.ForeignKeyField(
+                Servers,
+                backref="backup_server",
+                null=True,
+                field=peewee.CharField(primary_key=True, default=str(uuid.uuid4())),
+            ),
+        )
+
+        # Changes on SChedule Table
+        migrator.alter_column_type(
+            Schedules,
+            "server_id",
+            peewee.ForeignKeyField(
+                Servers,
+                backref="schedule_server",
+                null=True,
+                field=peewee.CharField(primary_key=True, default=str(uuid.uuid4())),
+            ),
+        )
+
+        migrator.run()
+
+        logger.info("Migrating Data from Int to UUID (Fixing Issue) : SUCCESS")
+        Console.info("Migrating Data from Int to UUID (Fixing Issue) : SUCCESS")
+
+    except Exception as ex:
+        logger.error("Error while migrating Data from Int to UUID (Fixing Issue)")
+        logger.error(ex)
+        Console.error("Error while migrating Data from Int to UUID (Fixing Issue)")
+        Console.error(ex)
+        last_migration = MigrateHistory.get_by_id(MigrateHistory.select().count())
+        last_migration.delete()
+        return
+
+    return
+
+
+def rollback(migrator: Migrator, database, **kwargs):
+    """
+    Write your rollback migrations here.
+    """
+    db = database
+
+    # Changes on Webhook Table
+    migrator.alter_column_type(
+        RoleServers,
+        "server_id",
+        peewee.IntegerField(null=True),
+    )
+
+    # Changes on Webhook Table
+    migrator.alter_column_type(
+        Backups,
+        "server_id",
+        peewee.IntegerField(null=True),
+    )
+
+    # Changes on Webhook Table
+    migrator.alter_column_type(
+        Schedules,
+        "server_id",
+        peewee.IntegerField(null=True),
+    )