From 2e3e6cbde5f9e2d04f359cbe4f3a32d20b31bd5a Mon Sep 17 00:00:00 2001 From: Lila Yasin Date: Mon, 10 Jul 2023 10:41:34 -0400 Subject: [PATCH] hop node migration file updates(#14196) rename migration function set_peers_from_control_nodes_true to automatically_peer_from_control_plane import settings and only run function if settings.IS_K8S is true set listener_port for control nodes to None --- awx/main/migrations/0186_hop_nodes.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/awx/main/migrations/0186_hop_nodes.py b/awx/main/migrations/0186_hop_nodes.py index c157b2c952..4375c5cc6b 100644 --- a/awx/main/migrations/0186_hop_nodes.py +++ b/awx/main/migrations/0186_hop_nodes.py @@ -2,11 +2,14 @@ import django.core.validators from django.db import migrations, models +from django.conf import settings -def set_peers_from_control_nodes_true(apps, schema_editor): - Instance = apps.get_model('main', 'Instance') - Instance.objects.filter(node_type='execution').update(peers_from_control_nodes=True) +def automatically_peer_from_control_plane(apps, schema_editor): + if settings.IS_K8S: + Instance = apps.get_model('main', 'Instance') + Instance.objects.filter(node_type='execution').update(peers_from_control_nodes=True) + Instance.objects.filter(node_type='control').update(listener_port=None) class Migration(migrations.Migration): @@ -49,5 +52,5 @@ class Migration(migrations.Migration): validators=[django.core.validators.MinValueValidator(1), django.core.validators.MaxValueValidator(65535)], ), ), - migrations.RunPython(set_peers_from_control_nodes_true), + migrations.RunPython(automatically_peer_from_control_plane), ]