From 2d114a4d1668db1ee405b5cf3121351b016928c0 Mon Sep 17 00:00:00 2001 From: Oleksii Baranov Date: Tue, 14 Mar 2023 19:32:56 +0100 Subject: [PATCH] Add migration for new cyberark plugin names --- .../0177_change_cyberark_plugin_names.py | 31 +++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 awx/main/migrations/0177_change_cyberark_plugin_names.py diff --git a/awx/main/migrations/0177_change_cyberark_plugin_names.py b/awx/main/migrations/0177_change_cyberark_plugin_names.py new file mode 100644 index 0000000000..4a6c0452e8 --- /dev/null +++ b/awx/main/migrations/0177_change_cyberark_plugin_names.py @@ -0,0 +1,31 @@ +# Generated by Django 3.2.16 on 2023-03-14 18:25 + +from django.db import migrations + +from awx.main.models import CredentialType + + +class Migration(migrations.Migration): + def update_cyberark_ccp_name(apps, schema_editor): + CredentialType.setup_tower_managed_defaults(apps) + aim_types = apps.get_model('main', 'CredentialType').objects.filter(namespace='aim').order_by('id') + + if aim_types.count() == 2: + original, renamed = aim_types.all() + apps.get_model('main', 'Credential').objects.filter(credential_type_id=original.id).update(credential_type_id=renamed.id) + original.delete() + + def update_cyberark_conjur_name(apps, schema_editor): + CredentialType.setup_tower_managed_defaults(apps) + conjur_types = apps.get_model('main', 'CredentialType').objects.filter(namespace='conjur').order_by('id') + + if conjur_types.count() == 2: + original, renamed = conjur_types.all() + apps.get_model('main', 'Credential').objects.filter(credential_type_id=original.id).update(credential_type_id=renamed.id) + original.delete() + + dependencies = [ + ('main', '0176_inventorysource_scm_branch'), + ] + + operations = [migrations.RunPython(update_cyberark_ccp_name), migrations.RunPython(update_cyberark_conjur_name)]