crafty-4/app/migrations/20210813111015_init.py

import peewee
import datetime


def migrate(migrator, database, **kwargs):
    db = database

    class Users(peewee.Model):
        user_id = peewee.AutoField()
        created = peewee.DateTimeField(default=datetime.datetime.now)
        last_login = peewee.DateTimeField(default=datetime.datetime.now)
        last_update = peewee.DateTimeField(default=datetime.datetime.now)
        last_ip = peewee.CharField(default="")
        username = peewee.CharField(default="", unique=True, index=True)
        password = peewee.CharField(default="")
        enabled = peewee.BooleanField(default=True)
        superuser = peewee.BooleanField(default=False)
        # we may need to revisit this
        api_token = peewee.CharField(default="", unique=True, index=True)

        class Meta:
            table_name = "users"
            database = db

    class Roles(peewee.Model):
        role_id = peewee.AutoField()
        created = peewee.DateTimeField(default=datetime.datetime.now)
        last_update = peewee.DateTimeField(default=datetime.datetime.now)
        role_name = peewee.CharField(default="", unique=True, index=True)

        class Meta:
            table_name = "roles"
            database = db

    class User_Roles(peewee.Model):
        user_id = peewee.ForeignKeyField(Users, backref="user_role")
        role_id = peewee.ForeignKeyField(Roles, backref="user_role")

        class Meta:
            table_name = "user_roles"
            primary_key = peewee.CompositeKey("user_id", "role_id")
            database = db

    class Audit_Log(peewee.Model):
        audit_id = peewee.AutoField()
        created = peewee.DateTimeField(default=datetime.datetime.now)
        user_name = peewee.CharField(default="")
        user_id = peewee.IntegerField(default=0, index=True)
        source_ip = peewee.CharField(default="127.0.0.1")
        # When auditing global events, use server ID 0
        server_id = peewee.IntegerField(default=None, index=True)
        log_msg = peewee.TextField(default="")

        class Meta:
            database = db

    class Host_Stats(peewee.Model):
        time = peewee.DateTimeField(default=datetime.datetime.now, index=True)
        boot_time = peewee.CharField(default="")
        cpu_usage = peewee.FloatField(default=0)
        cpu_cores = peewee.IntegerField(default=0)
        cpu_cur_freq = peewee.FloatField(default=0)
        cpu_max_freq = peewee.FloatField(default=0)
        mem_percent = peewee.FloatField(default=0)
        mem_usage = peewee.CharField(default="")
        mem_total = peewee.CharField(default="")
        disk_json = peewee.TextField(default="")

        class Meta:
            table_name = "host_stats"
            database = db

    class Servers(peewee.Model):
        server_id = peewee.AutoField()
        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)

        class Meta:
            table_name = "servers"
            database = db

    class User_Servers(peewee.Model):
        user_id = peewee.ForeignKeyField(Users, backref="user_server")
        server_id = peewee.ForeignKeyField(Servers, backref="user_server")

        class Meta:
            table_name = "user_servers"
            primary_key = peewee.CompositeKey("user_id", "server_id")
            database = db

    class Role_Servers(peewee.Model):
        role_id = peewee.ForeignKeyField(Roles, backref="role_server")
        server_id = peewee.ForeignKeyField(Servers, backref="role_server")

        class Meta:
            table_name = "role_servers"
            primary_key = peewee.CompositeKey("role_id", "server_id")
            database = db

    class Server_Stats(peewee.Model):
        stats_id = peewee.AutoField()
        created = peewee.DateTimeField(default=datetime.datetime.now)
        server_id = peewee.ForeignKeyField(Servers, backref="server", index=True)
        started = peewee.CharField(default="")
        running = peewee.BooleanField(default=False)
        cpu = peewee.FloatField(default=0)
        mem = peewee.FloatField(default=0)
        mem_percent = peewee.FloatField(default=0)
        world_name = peewee.CharField(default="")
        world_size = peewee.CharField(default="")
        server_port = peewee.IntegerField(default=25565)
        int_ping_results = peewee.CharField(default="")
        online = peewee.IntegerField(default=0)
        max = peewee.IntegerField(default=0)
        players = peewee.CharField(default="")
        desc = peewee.CharField(default="Unable to Connect")
        version = peewee.CharField(default="")
        updating = peewee.BooleanField(default=False)

        class Meta:
            table_name = "server_stats"
            database = db

    class Commands(peewee.Model):
        command_id = peewee.AutoField()
        created = peewee.DateTimeField(default=datetime.datetime.now)
        server_id = peewee.ForeignKeyField(Servers, backref="server", index=True)
        user = peewee.ForeignKeyField(Users, backref="user", index=True)
        source_ip = peewee.CharField(default="127.0.0.1")
        command = peewee.CharField(default="")
        executed = peewee.BooleanField(default=False)

        class Meta:
            table_name = "commands"
            database = db

    class Webhooks(peewee.Model):
        id = peewee.AutoField()
        name = peewee.CharField(max_length=64, unique=True, index=True)
        method = peewee.CharField(default="POST")
        url = peewee.CharField(unique=True)
        event = peewee.CharField(default="")
        send_data = peewee.BooleanField(default=True)

        class Meta:
            table_name = "webhooks"
            database = db

    class Schedules(peewee.Model):
        schedule_id = peewee.IntegerField(unique=True, primary_key=True)
        server_id = peewee.ForeignKeyField(Servers, backref="schedule_server")
        enabled = peewee.BooleanField()
        action = peewee.CharField()
        interval = peewee.IntegerField()
        interval_type = peewee.CharField()
        start_time = peewee.CharField(null=True)
        command = peewee.CharField(null=True)
        comment = peewee.CharField()

        class Meta:
            table_name = "schedules"
            database = db

    class Backups(peewee.Model):
        directories = peewee.CharField(null=True)
        max_backups = peewee.IntegerField()
        server_id = peewee.ForeignKeyField(Servers, backref="backups_server")
        schedule_id = peewee.ForeignKeyField(Schedules, backref="backups_schedule")

        class Meta:
            table_name = "backups"
            database = db

    migrator.create_table(Backups)
    migrator.create_table(Users)
    migrator.create_table(Roles)
    migrator.create_table(User_Roles)
    migrator.create_table(User_Servers)
    migrator.create_table(Host_Stats)
    migrator.create_table(Webhooks)
    migrator.create_table(Servers)
    migrator.create_table(Role_Servers)
    migrator.create_table(Server_Stats)
    migrator.create_table(Commands)
    migrator.create_table(Audit_Log)
    migrator.create_table(Schedules)


def rollback(migrator, database, **kwargs):
    migrator.drop_table("users")
    migrator.drop_table("roles")
    migrator.drop_table("user_roles")
    migrator.drop_table(
        "audit_log"
    )  # ? Not 100% sure of the table name, please specify in the schema
    migrator.drop_table("host_stats")
    migrator.drop_table("servers")
    migrator.drop_table("user_servers")
    migrator.drop_table("role_servers")
    migrator.drop_table("server_stats")
    migrator.drop_table("commands")
    migrator.drop_table("webhooks")
    migrator.drop_table("schedules")
    migrator.drop_table("backups")