summaryrefslogtreecommitdiff
path: root/python/constantrenderer.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/constantrenderer.py')
-rw-r--r--python/constantrenderer.py19
1 files changed, 18 insertions, 1 deletions
diff --git a/python/constantrenderer.py b/python/constantrenderer.py
index 7f4c251c..a0c2c98c 100644
--- a/python/constantrenderer.py
+++ b/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 @@ class _ConstantRendererMetaClass(type):
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):
"""