| 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) |