From 5880769cbac1362aad46a34faec4c553fa9fcac6 Mon Sep 17 00:00:00 2001 From: Peter LaFosse Date: Thu, 20 Nov 2025 09:58:28 -0500 Subject: Add get and __contains__ methods to all *MetaClass Fixes: https://github.com/Vector35/binaryninja-api/issues/7588 --- python/architecture.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) (limited to 'python/architecture.py') 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): """ -- cgit v1.3.1