From 3c5ee4434e237c45e22f67575c787ecd4a73144b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Ma=C5=9Blanka?= <piotr.maslanka@henrietta.com.pl>
Date: Thu, 6 Feb 2020 02:29:26 +0100
Subject: [PATCH] conciser code

---
 satella/instrumentation/metrics/json.py | 28 ++++++++++---------------
 1 file changed, 11 insertions(+), 17 deletions(-)

diff --git a/satella/instrumentation/metrics/json.py b/satella/instrumentation/metrics/json.py
index bc8e31f8..cf263e1f 100644
--- a/satella/instrumentation/metrics/json.py
+++ b/satella/instrumentation/metrics/json.py
@@ -8,15 +8,18 @@ from satella.coding import for_argument
 logger = logging.getLogger(__name__)
 
 
-@for_argument(copy.copy, copy.copy)
-def are_equal(tree1, tree2) -> bool:
-    tree1.pop('_', None)
-    tree2.pop('_', None)
-    tree1.pop('_timestamp', None)
-    tree2.pop('_timestamp', None)
+def get_labels_for_node(tree):
+    output = {}
+    for k, v in tree.items():
+        if k in ('_', '_timestamp'):
+            continue
+        if not isinstance(v, (list, dict, tuple)):
+            output[k] = v
+    return output
+
 
-    tree1 = get_labels_for_node(tree1)
-    tree2 = get_labels_for_node(tree2)
+@for_argument(get_labels_for_node, get_labels_for_node)
+def are_equal(tree1, tree2) -> bool:
     return tree1 == tree2
 
 
@@ -94,12 +97,3 @@ def update(tree1, tree2):
 
     return return_tree
 
-
-def get_labels_for_node(tree):
-    output = {}
-    for k, v in tree.items():
-        if k in ('_', '_timestamp'):
-            continue
-        if not isinstance(v, (list, dict, tuple)):
-            output[k] = v
-    return output
\ No newline at end of file
-- 
GitLab