External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-08 01:47:35 -02:30
Code Issues Packages Projects Releases Wiki Activity

Heavily refactor group recursive delete provides closure for AC

Browse Source
This commit is contained in:
Matthew Jones
2014-05-28 00:34:37 -04:00
parent 7160bff363
commit fe925f1925
1 changed files with 49 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
awx/main/models/inventory.py
View File

@@ -11,7 +11,7 @@ import os
import re import re
import shlex import shlex
import uuid import uuid
import copy
# PyYAML # PyYAML
import yaml import yaml
@@ -534,41 +534,61 @@ class Group(CommonModelNameNotUnique):
from awx.main.tasks import update_inventory_computed_fields, bulk_inventory_element_delete from awx.main.tasks import update_inventory_computed_fields, bulk_inventory_element_delete
from awx.main.utils import ignore_inventory_computed_fields from awx.main.utils import ignore_inventory_computed_fields
from awx.main.signals import disable_activity_stream from awx.main.signals import disable_activity_stream
def remove_host_from_group(host, group):
#host.inventory_sources.through.objects.filter(inventorysource__group=group).delete()
return host.groups.count() < 2
def mark_actual(): def mark_actual():
initial_hosts = self.hosts.all().prefetch_related('groups', 'inventory_sources') all_group_hosts = Group.hosts.through.objects.select_related("host", "group").filter(group__inventory=self.inventory)
linked_children = [(self, c) for c in self.children.all().prefetch_related('parents', 'hosts', 'inventory_sources', 'children')] group_hosts = {'groups': {}, 'hosts': {}}
all_group_parents = Group.parents.through.objects.select_related("parent", "group").filter(from_group__inventory=self.inventory)
group_children = {}
group_parents = {}
marked_hosts = [] marked_hosts = []
marked_groups = [self] marked_groups = [self.id]
for host in initial_hosts:
is_last_group = remove_host_from_group(host, self) for pairing in all_group_hosts:
if is_last_group: if pairing.group_id not in group_hosts['groups']:
marked_hosts.append(host) group_hosts['groups'][pairing.group_id] = []
self.hosts.through.objects.filter(group=self).delete() if pairing.host_id not in group_hosts['hosts']:
self.children.through.objects.filter(to_group=self).delete() group_hosts['hosts'][pairing.host_id] = []
self.inventory_sources.through.objects.filter(group=self).delete() group_hosts['groups'][pairing.group_id].append(pairing.host_id)
group_hosts['hosts'][pairing.host_id].append(pairing.group_id)
for pairing in all_group_parents:
if pairing.to_group_id not in group_children:
group_children[pairing.to_group_id] = []
if pairing.from_group_id not in group_parents:
group_parents[pairing.from_group_id] = []
group_children[pairing.to_group_id].append(pairing.from_group_id)
group_parents[pairing.from_group_id].append(pairing.to_group_id)
linked_children = [(self.id, g) for g in group_children[self.id]] if self.id in group_children else []
if self.id in group_hosts['groups']:
for host in copy.copy(group_hosts['groups'][self.id]):
group_hosts['hosts'][host].remove(self.id)
group_hosts['groups'][self.id].remove(host)
if len(group_hosts['hosts'][host]) < 1:
marked_hosts.append(host)
for subgroup in linked_children: for subgroup in linked_children:
parent, group = subgroup parent, group = subgroup
if group.parents.count() > 1: group_parents[group].remove(parent)
group_children[parent].remove(group)
if len(group_parents[group]) > 0:
continue continue
all_group_hosts = group.hosts.all() for host in copy.copy(group_hosts['groups'].get(group, [])):
for host in group.hosts.all(): group_hosts['hosts'][host].remove(group)
is_last_group = remove_host_from_group(host, group) group_hosts['groups'][group].remove(host)
if is_last_group: if len(group_hosts['hosts'][host]) < 1:
marked_hosts.append(host) marked_hosts.append(host)
group.hosts.through.objects.filter(group=group).delete() if group in group_children:
for childgroup in group.children.all().prefetch_related('parents', 'hosts', 'inventory_sources', 'children'): for direct_child in group_children[group]:
linked_children.append((group, childgroup)) linked_children.append((group, direct_child))
marked_groups.append(group) marked_groups.append(group)
group.children.through.objects.filter(to_group=group).delete() Group.objects.filter(id__in=marked_groups).update(active=False)
group.inventory_sources.through.objects.filter(group=group).delete() Host.objects.filter(id__in=marked_hosts).update(active=False)
all_groups = [g.id for g in marked_groups] Group.parents.through.objects.filter(to_group__id__in=marked_groups)
all_hosts = [h.id for h in marked_hosts] Group.hosts.through.objects.filter(group__id__in=marked_groups)
Group.objects.filter(id__in=all_groups).update(active=False) Group.inventory_sources.through.objects.filter(group__id__in=marked_groups).delete()
Host.objects.filter(id__in=all_hosts).update(active=False) bulk_inventory_element_delete.delay(self.inventory.id, groups=marked_groups, hosts=marked_hosts)
bulk_inventory_element_delete.delay(self.inventory.id, groups=all_groups, hosts=all_hosts)
with ignore_inventory_computed_fields(): with ignore_inventory_computed_fields():
with disable_activity_stream(): with disable_activity_stream():
mark_actual() mark_actual()