20 lines
834 B
Python
20 lines
834 B
Python
from autotest_lib.client.common_lib import global_config
|
|
|
|
def migrate_up(manager):
|
|
# Add the column with a default first, and then drop the default.
|
|
# We cannot add the column, populate the values, and then specify NOT NULL
|
|
# because a record added while this is executing could enter a null value
|
|
# into the table before NOT NULL is specified.
|
|
manager.execute(ADD_COLUMN)
|
|
manager.execute(DROP_DEFAULT)
|
|
|
|
def migrate_down(manager):
|
|
manager.execute(DROP_COLUMN)
|
|
|
|
job_timeout_default = global_config.global_config.get_config_value(
|
|
'AUTOTEST_WEB', 'job_timeout_default')
|
|
ADD_COLUMN = ('ALTER TABLE jobs ADD COLUMN timeout INT NOT NULL DEFAULT %s'
|
|
% job_timeout_default)
|
|
DROP_DEFAULT = 'ALTER TABLE jobs ALTER COLUMN timeout DROP DEFAULT'
|
|
DROP_COLUMN = 'ALTER TABLE jobs DROP COLUMN timeout'
|