Add get and __contains__ methods to all *MetaClass

Fixes: https://github.com/Vector35/binaryninja-api/issues/7588

1 files changed, 18 insertions, 1 deletions

diff --git a/python/architecture.py b/python/architecture.py
index 278b3317..734864a4 100644
--- a/python/architecture.py
+++ b/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 @@ class _ArchitectureMetaClass(type):
 			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):
 	"""