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_droneset', 'change_droneset', 'delete_droneset', 'add_drone',
|
'change_drone', 'delete_drone');
|
"""
|
|
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_droneset', 'change_droneset', 'delete_droneset', 'add_drone',
|
'change_drone', 'delete_drone');
|
"""
|
|
|
def migrate_up(manager):
|
"""
|
If the auth tables don't exist, we shouldn't try to set the permissions.
|
|
The auth tables will exist if this is an existing Autotest installation. If
|
they don't, then this is a fresh installation, and the user will run
|
`manage.py syncdb` later, which will add the proper permissions.
|
"""
|
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)
|
