From 3b5f1da4b25978bd97ebab7762122370dcb6d5a0 Mon Sep 17 00:00:00 2001 From: Idan Hefetz Date: Sun, 4 Mar 2018 10:03:53 +0000 Subject: [PATCH] bugfix for log warnings Change-Id: I0e59c44e29e350c9613004fc97afeaff36c2f562 --- vitrage/entity_graph/processor/processor_utils.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/vitrage/entity_graph/processor/processor_utils.py b/vitrage/entity_graph/processor/processor_utils.py index ee75ce75b..0c292a79e 100644 --- a/vitrage/entity_graph/processor/processor_utils.py +++ b/vitrage/entity_graph/processor/processor_utils.py @@ -83,12 +83,10 @@ def find_neighbor_types(neighbors): def get_vertex_types(vertex): - props = vertex.items() - if VProps.VITRAGE_CATEGORY not in props: + vitrage_category = vertex.get(VProps.VITRAGE_CATEGORY) + vitrage_type = vertex.get(VProps.VITRAGE_TYPE) + if not vitrage_category: LOG.warning('no vitrage_category in vertex: %s', str(vertex)) - - vitrage_category = vertex[VProps.VITRAGE_CATEGORY] - vitrage_type = vertex[VProps.VITRAGE_TYPE] return vitrage_category, vitrage_type