import peewee
import datetime


def migrate(migrator, database, **kwargs):
    migrator.add_columns(
        "users", valid_tokens_from=peewee.DateTimeField(default=datetime.datetime.now)
    )
    migrator.drop_columns("users", ["api_token"])


def rollback(migrator, database, **kwargs):
    migrator.drop_columns("users", ["valid_tokens_from"])
    migrator.add_columns(
        "users", api_token=peewee.CharField(default="", unique=True, index=True)
    )