diff options
| author | Peter LaFosse <peter@vector35.com> | 2025-11-20 09:58:28 -0500 |
|---|---|---|
| committer | Peter LaFosse <peter@vector35.com> | 2025-11-21 10:02:50 -0500 |
| commit | 5880769cbac1362aad46a34faec4c553fa9fcac6 (patch) | |
| tree | 02db205b7aa915900cbfc10384940b3750eb7d69 /python/renderlayer.py | |
| parent | f52693d50de6a5c97afa0274277edca18cfa00ea (diff) | |
Add get and __contains__ methods to all *MetaClass
Fixes: https://github.com/Vector35/binaryninja-api/issues/7588
Diffstat (limited to 'python/renderlayer.py')
| -rw-r--r-- | python/renderlayer.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/python/renderlayer.py b/python/renderlayer.py index 0570ac3f..3731684d 100644 --- a/python/renderlayer.py +++ b/python/renderlayer.py @@ -28,7 +28,7 @@ from .enums import LinearDisassemblyLineType, RenderLayerDefaultEnableState from . import binaryview from . import types from .log import log_error_for_exception -from typing import Iterable, List, Optional, Union, Tuple +from typing import Iterable, List, Optional, Union, Tuple, Any class _RenderLayerMetaclass(type): @@ -49,6 +49,23 @@ class _RenderLayerMetaclass(type): raise KeyError(f"'{value}' is not a valid RenderLayer") return self._handle_to_instance(handle) + def __contains__(cls: '_RenderLayerMetaclass', name: object) -> bool: + if not isinstance(name, str): + return False + try: + cls[name] + return True + except KeyError: + return False + + def get(cls: '_RenderLayerMetaclass', name: str, default: Any = None) -> Optional['RenderLayer']: + try: + return cls[name] + except KeyError: + if default is not None: + return default + return None + def _handle_to_instance(self, handle): handle_ptr = ctypes.cast(handle, ctypes.c_void_p) if handle_ptr.value in RenderLayer._registered_instances: |
