Add a signal handler to remove the default EE if it gets deleted

This commit is contained in:
Jeff Bradberry 2021-06-09 10:10:10 -04:00
parent 9f1e8a1ae2
commit e6e1f97048

View File

@ -31,6 +31,7 @@ from crum.signals import current_user_getter
# AWX
from awx.main.models import (
ActivityStream,
ExecutionEnvironment,
Group,
Host,
InstanceGroup,
@ -623,6 +624,12 @@ def deny_orphaned_approvals(sender, instance, **kwargs):
approval.deny()
@receiver(pre_delete, sender=ExecutionEnvironment)
def remove_default_ee(sender, instance, **kwargs):
if instance.id == getattr(settings.DEFAULT_EXECUTION_ENVIRONMENT, 'id', None):
settings.DEFAULT_EXECUTION_ENVIRONMENT = None
@receiver(post_save, sender=Session)
def save_user_session_membership(sender, **kwargs):
session = kwargs.get('instance', None)