From 95234d9fc98db62a6c19fba50b602f0f77ffb137 Mon Sep 17 00:00:00 2001 From: Dao Nguyen Date: Tue, 17 Sep 2024 00:38:16 +0700 Subject: [PATCH] format py script --- scripts/data/utreexo.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/scripts/data/utreexo.py b/scripts/data/utreexo.py index c405da11..9db65cee 100644 --- a/scripts/data/utreexo.py +++ b/scripts/data/utreexo.py @@ -7,6 +7,7 @@ # The set of leaf nodes in the forest leaf_nodes = dict() + class Node: def __init__(self, key, left=None, right=None): self.val = key @@ -14,6 +15,7 @@ def __init__(self, key, left=None, right=None): self.right = right self.parent = None + def parent_node(root1, root2): # Convert hexadecimal strings to integers val1 = int(root1.val, 16) @@ -27,9 +29,10 @@ def parent_node(root1, root2): root2.parent = root_node return root_node + def utreexo_add(leaf): if leaf in leaf_nodes: - raise Exception('Leaf exists already') + raise Exception("Leaf exists already") n = Node(f"0x{leaf:064x}") leaf_nodes[leaf] = n @@ -44,6 +47,7 @@ def utreexo_add(leaf): root_nodes[h] = n return root_nodes + def utreexo_delete(leaf): leaf_node = leaf_nodes[leaf] del leaf_nodes[leaf] @@ -72,6 +76,7 @@ def utreexo_delete(leaf): proof = list(map(lambda node: node.val, proof)) return proof, leaf_index + def inclusion_proof(node): if node.parent is None: return [], 0 @@ -88,16 +93,22 @@ def inclusion_proof(node): return path, leaf_index + def reset_utreexo(): global root_nodes, leaf_nodes root_nodes = [None] * 27 leaf_nodes = dict() + def print_roots(): - print('Roots:', list(map(lambda node: node.val if node is not None else '', root_nodes))) + print( + "Roots:", + list(map(lambda node: node.val if node is not None else "", root_nodes)), + ) + # Example usage -if __name__ == '__main__': +if __name__ == "__main__": # Add some elements utreexo_add(0x111111111111111111111111) utreexo_add(0x222222222222222222222222) @@ -107,4 +118,4 @@ def print_roots(): # Reset the Utreexo reset_utreexo() - print_roots() \ No newline at end of file + print_roots()