diff options
Diffstat (limited to 'python/typeprinter.py')
| -rw-r--r-- | python/typeprinter.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/python/typeprinter.py b/python/typeprinter.py index be885c07..091cd803 100644 --- a/python/typeprinter.py +++ b/python/typeprinter.py @@ -21,7 +21,7 @@ import abc import ctypes import dataclasses from json import dumps -from typing import List, Tuple, Optional +from typing import List, Tuple, Optional, Any import sys import traceback @@ -66,6 +66,23 @@ class _TypePrinterMetaclass(type): raise KeyError(f"'{value}' is not a valid TypePrinter") return CoreTypePrinter(handle) + def __contains__(cls: '_TypePrinterMetaclass', name: object) -> bool: + if not isinstance(name, str): + return False + try: + cls[name] + return True + except KeyError: + return False + + def get(cls: '_TypePrinterMetaclass', name: str, default: Any = None) -> Optional['TypePrinter']: + try: + return cls[name] + except KeyError: + if default is not None: + return default + return None + @property def default(self): name = binaryninja.Settings().get_string("analysis.types.printerName") |
