From 5badd0a2c09d815cd890e6a50e25a3bc4f8e861b Mon Sep 17 00:00:00 2001
From: Robert Latta <robert.latta@geant.org>
Date: Mon, 7 Jan 2019 14:03:48 +0000
Subject: [PATCH] Changed update_service_hierarchy function to use defaultdict

---
 inventory_provider/storage/external_inventory.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/inventory_provider/storage/external_inventory.py b/inventory_provider/storage/external_inventory.py
index e31e43bb..cc7bccf6 100644
--- a/inventory_provider/storage/external_inventory.py
+++ b/inventory_provider/storage/external_inventory.py
@@ -36,16 +36,11 @@ def update_interfaces_to_services(services):
 
 def update_service_hierarchy(records):
     r = db.get_redis()
-    children_to_parents = {}
-    parents_to_children = {}
+    children_to_parents = defaultdict(list)
+    parents_to_children = defaultdict(list)
     for relation in records:
         parent_id = relation["parent_circuit_id"]
         child_id = relation["child_circuit_id"]
-        if child_id not in children_to_parents:
-            children_to_parents[child_id] = []
-        if parent_id not in parents_to_children:
-            parents_to_children[parent_id] = []
-
         parents_to_children[parent_id].append(relation)
         children_to_parents[child_id].append(relation)
 
-- 
GitLab