diff options
Diffstat (limited to 'python/websocketprovider.py')
| -rw-r--r-- | python/websocketprovider.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/python/websocketprovider.py b/python/websocketprovider.py index 0d77ae0b..f990d859 100644 --- a/python/websocketprovider.py +++ b/python/websocketprovider.py @@ -22,6 +22,7 @@ import abc import ctypes import sys import traceback +from typing import Any, Optional # Binary Ninja Components import binaryninja._binaryninjacore as core @@ -249,6 +250,23 @@ class _WebsocketProviderMetaclass(type): raise KeyError("'%s' is not a valid websocket provider" % str(value)) return WebsocketProvider(provider) + def __contains__(cls: '_WebsocketProviderMetaclass', name: object) -> bool: + if not isinstance(name, str): + return False + try: + cls[name] + return True + except KeyError: + return False + + def get(cls: '_WebsocketProviderMetaclass', name: str, default: Any = None) -> Optional['WebsocketProvider']: + try: + return cls[name] + except KeyError: + if default is not None: + return default + return None + def __setattr__(self, name, value): try: type.__setattr__(self, name, value) |