From 0ab44e70f9679b2bb2bfdba41f7500b72289a96e Mon Sep 17 00:00:00 2001 From: Ryan Petrello Date: Thu, 17 Oct 2019 22:27:25 -0400 Subject: [PATCH] properly migrate the CyberArk AIM type to its new name --- ..._v360_rename_cyberark_aim_credential_type.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/awx/main/migrations/0098_v360_rename_cyberark_aim_credential_type.py b/awx/main/migrations/0098_v360_rename_cyberark_aim_credential_type.py index f63524d5c1..0bd03b94ba 100644 --- a/awx/main/migrations/0098_v360_rename_cyberark_aim_credential_type.py +++ b/awx/main/migrations/0098_v360_rename_cyberark_aim_credential_type.py @@ -1,12 +1,23 @@ # Generated by Django 2.2.4 on 2019-10-16 19:51 from django.db import migrations +from awx.main.models import CredentialType def update_cyberark_aim_name(apps, schema_editor): - apps.get_model('main', 'CredentialType').objects.filter(namespace='aim').update( - name='CyberArk AIM Central Credential Provider Lookup' - ) + CredentialType.setup_tower_managed_defaults() + 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() class Migration(migrations.Migration):