import peewee as pw


def migrate(migrator, database, fake=False, **kwargs):
    orm_IPList = migrator.orm["iplist"]
    IP_LISTS = ("WHITE", "BLACK", "GRAY", "GRAY_SPLASHSCREEN")
    migrator.change_fields(
        orm_IPList,
        listname=pw.CharField(
            null=False,
            constraints=[
                pw.Check("listname in ('{}')".format("','".join(IP_LISTS)))
            ],
        ),
    )


def rollback(migrator, database, fake=False, **kwargs):
    pass