# Generated by database migrator
from peewee import *
from app.classes.models.users import Users
from app.classes.models.servers import Servers


def migrate(migrator, database, **kwargs):
    migrator.drop_table("user_servers")  # Can be model class OR table name
    """
    Write your migrations here.
    """


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

    class User_Servers(Model):
        user_id = ForeignKeyField(Users, backref="user_server")
        server_id = ForeignKeyField(Servers, backref="user_server")
        permissions = CharField(default="00000000")

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

    migrator.create_table(User_Servers)
    """
    Write your rollback migrations here.
    """