import peewee as pw


class SecureSite(pw.Model):
    class Meta:
        db_table = "secure_site"

    user = pw.CharField(unique=True)
    subscription_type = pw.TextField(
        null=False,
        constraints=[pw.Check("subscription_type in ('basic','pro')")],
        default="basic",
    )


def migrate(migrator, _db, fake=False, **__):
    if fake:
        return

    migrator.create_model(SecureSite)


def rollback(*_, **__):
    """Not supported"""