summaryrefslogtreecommitdiff
path: root/python/typelibrary.py
diff options
context:
space:
mode:
authorPeter LaFosse <peter@vector35.com>2026-01-16 10:36:58 -0500
committerPeter LaFosse <peter@vector35.com>2026-01-16 11:51:23 -0500
commitdd31558a0a1600342da23e2d0745be01ad19202b (patch)
tree7f04ed5998d798aeb518d4956fac495e2b7c4a1a /python/typelibrary.py
parentf327d9565f49b047d38bc1a7583e5f4e3776261a (diff)
Add get_metadata() method and make query_metadata() raise KeyError consistently
Introduces a Pythonic get_metadata() method to BinaryView, Function, Project, TypeArchive, and TypeLibrary classes. This method behaves like dict.get(), returning a default value (None by default) when a key doesn't exist, instead of raising a KeyError. Additionally, updates query_metadata() in TypeArchive and TypeLibrary to raise KeyError when a key is not found, making them consistent with BinaryView, Function, and Project. Previously these two classes returned None on missing keys. This breaking change is documented in the method docstrings. This provides a more consistent and Pythonic API for querying metadata across all metadata-supporting classes. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Diffstat (limited to 'python/typelibrary.py')
-rw-r--r--python/typelibrary.py36
1 files changed, 33 insertions, 3 deletions
diff --git a/python/typelibrary.py b/python/typelibrary.py
index 612375e8..457cbd99 100644
--- a/python/typelibrary.py
+++ b/python/typelibrary.py
@@ -19,7 +19,7 @@
# IN THE SOFTWARE.
import ctypes
-from typing import Optional, List, Dict, Union
+from typing import Any, Optional, List, Dict, Union
import uuid
# Binary Ninja components
@@ -233,12 +233,15 @@ class TypeLibrary:
"""
return core.BNFinalizeTypeLibrary(self.handle)
- def query_metadata(self, key: str) -> Optional['metadata.MetadataValueType']:
+ def query_metadata(self, key: str) -> 'metadata.MetadataValueType':
"""
`query_metadata` retrieves a metadata associated with the given key stored in the type library
:param string key: key to query
:rtype: metadata associated with the key
+ ..note: As of Binary Ninja 5.3 this API now raises KeyError on failure. Please use `get_metadata`
+ for a non-raising version of the API.
+
:Example:
>>> lib.store_metadata("ordinals", {"9": "htons"})
@@ -247,7 +250,34 @@ class TypeLibrary:
"""
md_handle = core.BNTypeLibraryQueryMetadata(self.handle, key)
if md_handle is None:
- return None
+ raise KeyError(key)
+ return metadata.Metadata(handle=md_handle).value
+
+ def get_metadata(self, key: str, default: Any = None) -> 'metadata.MetadataValueType | Any':
+ """
+ `get_metadata` retrieves a metadata value associated with the given key stored in the current BinaryView.
+
+ This method behaves like `dict.get()`:
+ - If the key exists, its metadata value is returned.
+ - If the key does not exist and `default` is not provided, `None` is returned.
+ - If the key does not exist and `default` is provided, `default` is returned.
+
+ :param str key: key to query
+ :param default: value to return if the key does not exist (defaults to None)
+ :rtype: metadata associated with the key or the default value
+ :Example:
+
+ >>> tl.store_metadata("integer", 1337)
+ >>> tl.get_metadata("integer")
+ 1337L
+ >>> tl.get_metadata("missing")
+ None
+ >>> tl.get_metadata("missing", 42)
+ 42
+ """
+ md_handle = core.BNTypeLibraryQueryMetadata(self.handle, key)
+ if md_handle is None:
+ return default
return metadata.Metadata(handle=md_handle).value
def store_metadata(self, key: str, md: metadata.Metadata) -> None: