UP_SQL = """
ALTER TABLE users ADD COLUMN `reboot_before` smallint NOT NULL;
ALTER TABLE users ADD COLUMN `reboot_after` smallint NOT NULL;
UPDATE users SET reboot_before=1, reboot_after=2;
"""


DOWN_SQL = """
ALTER TABLE users DROP COLUMN reboot_before;
ALTER TABLE users DROP COLUMN reboot_after;
"""


def migrate_up(manager):
    manager.execute_script(UP_SQL)


def migrate_down(manager):
    manager.execute_script(DOWN_SQL)