crafty-4/app/migrations/20240310_fixing_rework_servers_uuid.py

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:
        # 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())),
            ),
        )

    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),
    )