Merge pull request #885 from Chia-Network/add_merkle_blob_get_key_index

add `MerkleBlob.get_key_index()`

Author: altendky

crates/chia-datalayer/src/merkle.rs

@@ -1326,6 +1326,13 @@ impl MerkleBlob {
 
         Ok(map)
     }
+
+    pub fn get_key_index(&self, key: KvId) -> Result<TreeIndex, Error> {
+        self.key_to_index
+            .get(&key)
+            .copied()
+            .ok_or(Error::UnknownKey(key))
+    }
 }
 
 impl PartialEq for MerkleBlob {
@@ -1529,6 +1536,11 @@ impl MerkleBlob {
 
         Ok(dict.into())
     }
+
+    #[pyo3(name = "get_key_index")]
+    pub fn py_get_key_index(&self, key: KvId) -> PyResult<TreeIndex> {
+        Ok(self.get_key_index(key)?)
+    }
 }
 
 fn try_get_block(blob: &[u8], index: TreeIndex) -> Result<Block, Error> {

wheel/python/chia_rs/datalayer.pyi

@@ -89,6 +89,7 @@ class MerkleBlob:
     def batch_insert(self, keys_values: list[tuple[int64, int64]], hashes: list[bytes32]): ...
     def get_hash_at_index(self, index: uint32): ...
     def get_keys_values(self) -> dict[int64, int64]: ...
+    def get_key_index(self, key: int64) -> uint32: ...
 
     def __len__(self) -> int: ...