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/transform.py b/python/transform.py
index 9512b24b..b647a8a9 100644
--- a/python/transform.py
+++ b/python/transform.py
@@ -23,7 +23,7 @@ import ctypes
 import abc
 import io
 import zipfile
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Any
 
 # Binary Ninja components
 import binaryninja
@@ -58,6 +58,23 @@ class _TransformMetaClass(type):
 			raise KeyError("'%s' is not a valid transform" % str(name))
 		return Transform(xform)
 
+	def __contains__(cls: '_TransformMetaClass', name: object) -> bool:
+		if not isinstance(name, str):
+			return False
+		try:
+			cls[name]
+			return True
+		except KeyError:
+			return False
+
+	def get(cls: '_TransformMetaClass', name: str, default: Any = None) -> Optional['Transform']:
+		try:
+			return cls[name]
+		except KeyError:
+			if default is not None:
+				return default
+			return None
+
 
 class TransformParameter:
 	def __init__(self, name, long_name=None, fixed_length=0):