25 lines
827 B
Python
25 lines
827 B
Python
from autotest_lib.client.common_lib import global_config, host_protections
|
|
|
|
def migrate_up(manager):
|
|
manager.execute_script(ADD_PROTECTION_COLUMN)
|
|
|
|
def migrate_down(manager):
|
|
manager.execute(DROP_COLUMN)
|
|
|
|
default_protection = global_config.global_config.get_config_value(
|
|
'HOSTS', 'default_protection')
|
|
default_protection_value = host_protections.Protection.get_value(
|
|
default_protection)
|
|
|
|
ADD_PROTECTION_COLUMN = """ALTER TABLE hosts
|
|
ADD COLUMN protection INT NOT NULL
|
|
DEFAULT %s;
|
|
|
|
ALTER TABLE hosts
|
|
ALTER COLUMN protection
|
|
DROP DEFAULT;
|
|
""" % default_protection_value
|
|
|
|
DROP_COLUMN = """ALTER TABLE hosts
|
|
DROP COLUMN protection"""
|