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 ( Webhooks, 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_uuid = peewee.CharField(default="", index=True) 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 Server Table migrator.alter_column_type( Servers, "server_id", peewee.CharField(primary_key=True, default=str(uuid.uuid4())), ) # Changes on Webhook Table migrator.alter_column_type( Webhooks, "server_id", peewee.ForeignKeyField( Servers, backref="webhook_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 (Type Change)") logger.error(ex) Console.error("Error while migrating Data from Int to UUID (Type Change)") 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 Server Table migrator.alter_column_type( "servers", "server_id", peewee.AutoField(), ) # Changes on Webhook Table migrator.alter_column_type( Webhooks, "server_id", peewee.IntegerField(null=True), )