Add get and __contains__ methods to all *MetaClass

Fixes: #7588

Author: plafosse

python/architecture.py

@@ -20,7 +20,7 @@
 
 import traceback
 import ctypes
-from typing import Generator, Union, List, Optional, Mapping, Tuple, NewType, Dict, Set
+from typing import Generator, Union, List, Optional, Mapping, Tuple, NewType, Dict, Set, Any
 from dataclasses import dataclass, field
 
 # Binary Ninja components
@@ -514,6 +514,23 @@ def __getitem__(cls: '_ArchitectureMetaClass', name: str) -> 'Architecture':
 			raise KeyError(f"'{name}' is not a valid architecture")
 		return CoreArchitecture._from_cache(arch)
 
+	def __contains__(cls: '_ArchitectureMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_ArchitectureMetaClass', name: str, default: Any = None) -> Optional['Architecture']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class Architecture(metaclass=_ArchitectureMetaClass):
 	"""

python/binaryview.py

@@ -1448,6 +1448,23 @@ def __getitem__(self, value):
 			raise KeyError(f"'{value}' is not a valid view type")
 		return BinaryViewType(view_type)
 
+	def __contains__(cls: '_BinaryViewTypeMetaclass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_BinaryViewTypeMetaclass', name: str, default: Any = None) -> Optional['BinaryViewType']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class BinaryViewType(metaclass=_BinaryViewTypeMetaclass):
 	"""

python/constantrenderer.py

@@ -20,7 +20,7 @@
 
 import traceback
 import ctypes
-from typing import Optional
+from typing import Optional, Any
 
 import binaryninja
 from . import _binaryninjacore as core
@@ -51,6 +51,23 @@ def __getitem__(cls, value):
 			raise KeyError("'%s' is not a valid renderer" % str(value))
 		return CoreConstantRenderer(handle=renderer)
 
+	def __contains__(cls: '_ConstantRendererMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_ConstantRendererMetaClass', name: str, default: Any = None) -> Optional['ConstantRenderer']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class ConstantRenderer(metaclass=_ConstantRendererMetaClass):
 	"""

python/debuginfo.py

@@ -19,7 +19,7 @@
 # IN THE SOFTWARE.
 
 import ctypes
-from typing import Optional, List, Iterator, Callable, Tuple
+from typing import Optional, List, Iterator, Callable, Tuple, Any
 import traceback
 from dataclasses import dataclass, field
 
@@ -78,6 +78,23 @@ def __getitem__(cls, value: str) -> 'DebugInfoParser':
 		assert parser_ref is not None, "core.BNNewDebugInfoParserReference returned None"
 		return DebugInfoParser(parser_ref)
 
+	def __contains__(cls: '_DebugInfoParserMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_DebugInfoParserMetaClass', name: str, default: Any = None) -> Optional['DebugInfoParser']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 	@staticmethod
 	def get_parsers_for_view(view: 'binaryview.BinaryView') -> List['DebugInfoParser']:
 		"""Returns a list of debug-info parsers that are valid for the provided binary view"""

python/demangle.py

@@ -29,7 +29,7 @@
 from .log import log_error_for_exception
 from .architecture import Architecture, CoreArchitecture
 from .platform import Platform
-from typing import Iterable, List, Optional, Union, Tuple
+from typing import Iterable, List, Optional, Union, Tuple, Any
 
 
 def get_qualified_name(names: Iterable[str]):
@@ -300,6 +300,23 @@ def __getitem__(self, value):
 			raise KeyError(f"'{value}' is not a valid Demangler")
 		return CoreDemangler(handle)
 
+	def __contains__(cls: '_DemanglerMetaclass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_DemanglerMetaclass', name: str, default: Any = None) -> Optional['Demangler']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class Demangler(metaclass=_DemanglerMetaclass):
 	"""

python/downloadprovider.py

@@ -23,6 +23,7 @@
 from json import dumps
 import sys
 import traceback
+from typing import Any, Optional
 from urllib.parse import urlencode
 
 # Binary Ninja Components
@@ -269,6 +270,23 @@ def __getitem__(self, value):
 			raise KeyError(f"'{value}' is not a valid download provider")
 		return DownloadProvider(provider)
 
+	def __contains__(cls: '_DownloadProviderMetaclass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_DownloadProviderMetaclass', name: str, default: Any = None) -> Optional['DownloadProvider']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class DownloadProvider(metaclass=_DownloadProviderMetaclass):
 	name = None

python/languagerepresentation.py

@@ -20,7 +20,7 @@
 
 import ctypes
 import traceback
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Any
 
 # Binary Ninja components
 import binaryninja
@@ -664,6 +664,23 @@ def __getitem__(cls, value):
 			raise KeyError("'%s' is not a valid language" % str(value))
 		return CoreLanguageRepresentationFunctionType(handle=lang)
 
+	def __contains__(cls: '_LanguageRepresentationFunctionTypeMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_LanguageRepresentationFunctionTypeMetaClass', name: str, default: Any = None) -> Optional['LanguageRepresentationFunctionType']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class LanguageRepresentationFunctionType(metaclass=_LanguageRepresentationFunctionTypeMetaClass):
 	"""

python/lineformatter.py

@@ -21,7 +21,7 @@
 import ctypes
 import traceback
 from dataclasses import dataclass
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Any
 
 # Binary Ninja components
 import binaryninja
@@ -118,6 +118,23 @@ def __getitem__(cls, value):
             raise KeyError("'%s' is not a valid formatter" % str(value))
         return CoreLineFormatter(handle=lang)
 
+    def __contains__(cls: '_LineFormatterMetaClass', name: object) -> bool:
+        if not isinstance(name, str):
+            return False
+        try:
+            cls[name]
+            return True
+        except KeyError:
+            return False
+
+    def get(cls: '_LineFormatterMetaClass', name: str, default: Any = None) -> Optional['LineFormatter']:
+        try:
+            return cls[name]
+        except KeyError:
+            if default is not None:
+                return default
+            return None
+
 
 class LineFormatter(metaclass=_LineFormatterMetaClass):
     """

python/platform.py

@@ -22,7 +22,7 @@
 import ctypes
 import traceback
 import warnings
-from typing import List, Dict, Optional, Tuple
+from typing import List, Dict, Optional, Tuple, Any
 
 # Binary Ninja components
 import binaryninja
@@ -56,6 +56,23 @@ def __getitem__(cls, value):
 			raise KeyError("'%s' is not a valid platform" % str(value))
 		return CorePlatform(handle=platform)
 
+	def __contains__(cls: '_PlatformMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_PlatformMetaClass', name: str, default: Any = None) -> Optional['Platform']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class Platform(metaclass=_PlatformMetaClass):
 	"""

python/renderlayer.py

@@ -28,7 +28,7 @@
 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 @@ def __getitem__(self, value):
 			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: