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)