Skip to content
Snippets Groups Projects
Commit 3c5ee443 authored by Piotr Maślanka's avatar Piotr Maślanka
Browse files

conciser code

parent da39a316
No related branches found
No related tags found
No related merge requests found
...@@ -8,15 +8,18 @@ from satella.coding import for_argument ...@@ -8,15 +8,18 @@ from satella.coding import for_argument
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@for_argument(copy.copy, copy.copy) def get_labels_for_node(tree):
def are_equal(tree1, tree2) -> bool: output = {}
tree1.pop('_', None) for k, v in tree.items():
tree2.pop('_', None) if k in ('_', '_timestamp'):
tree1.pop('_timestamp', None) continue
tree2.pop('_timestamp', None) if not isinstance(v, (list, dict, tuple)):
output[k] = v
return output
tree1 = get_labels_for_node(tree1) @for_argument(get_labels_for_node, get_labels_for_node)
tree2 = get_labels_for_node(tree2) def are_equal(tree1, tree2) -> bool:
return tree1 == tree2 return tree1 == tree2
...@@ -94,12 +97,3 @@ def update(tree1, tree2): ...@@ -94,12 +97,3 @@ def update(tree1, tree2):
return return_tree 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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment