You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

31 lines
953 B

import common
from autotest_lib.database import db_utils
UP_SQL = """
SET @group_id = (SELECT id FROM auth_group WHERE name = 'Basic Admin');
INSERT IGNORE INTO auth_group_permissions (group_id, permission_id)
SELECT @group_id, id FROM auth_permission WHERE codename IN (
'add_hostattribute', 'change_hostattribute', 'delete_hostattribute');
"""
DOWN_SQL = """
DELETE auth_group_permissions.* FROM
auth_group INNER JOIN auth_group_permissions ON (
auth_group.id = auth_group_permissions.group_id)
INNER JOIN auth_permission ON (
auth_group_permissions.permission_id = auth_permission.id)
WHERE auth_group.name = 'Basic Admin' AND codename IN (
'add_hostattribute', 'change_hostattribute', 'delete_hostattribute');
"""
def migrate_up(manager):
if db_utils.auth_tables_exist(manager):
manager.execute_script(UP_SQL)
def migrate_down(manager):
if db_utils.auth_tables_exist(manager):
manager.execute_script(DOWN_SQL)