Merge pull request #90 from cirosantilli/master

Add the blk*.data file path basename to Block.blk_file
alecalve · Jan 14, 2021 · d763ca6 · d763ca6
2 parents e3a9750 + b1ca91e
commit d763ca6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/blockchain_parser/block.py b/blockchain_parser/block.py
@@ -46,14 +46,15 @@ class Block(object):
     Represents a Bitcoin block, contains its header and its transactions.
     """
 
-    def __init__(self, raw_hex, height=None):
+    def __init__(self, raw_hex, height=None, blk_file=None):
         self.hex = raw_hex
         self._hash = None
         self._transactions = None
         self._header = None
         self._n_transactions = None
         self.size = len(raw_hex)
         self.height = height
+        self.blk_file = blk_file
 
     def __repr__(self):
         return "Block(%s)" % self.hash

diff --git a/blockchain_parser/blockchain.py b/blockchain_parser/blockchain.py
@@ -94,7 +94,7 @@ def get_unordered_blocks(self):
         """
         for blk_file in get_files(self.path):
             for raw_block in get_blocks(blk_file):
-                yield Block(raw_block)
+                yield Block(raw_block, None, os.path.split(blk_file)[1])
 
     def __getBlockIndexes(self, index):
         """There is no method of leveldb to close the db (and release the lock).