diff options
Diffstat (limited to 'python/secretsprovider.py')
| -rw-r--r-- | python/secretsprovider.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/python/secretsprovider.py b/python/secretsprovider.py index d6815ae6..779b04fa 100644 --- a/python/secretsprovider.py +++ b/python/secretsprovider.py @@ -23,6 +23,7 @@ import ctypes from json import dumps import sys import traceback +from typing import Any, Optional from urllib.parse import urlencode # Binary Ninja Components @@ -58,6 +59,23 @@ class _SecretsProviderMetaclass(type): raise KeyError(f"'{value}' is not a valid secrets provider") return SecretsProvider(provider) + def __contains__(cls: '_SecretsProviderMetaclass', name: object) -> bool: + if not isinstance(name, str): + return False + try: + cls[name] + return True + except KeyError: + return False + + def get(cls: '_SecretsProviderMetaclass', name: str, default: Any = None) -> Optional['SecretsProvider']: + try: + return cls[name] + except KeyError: + if default is not None: + return default + return None + class SecretsProvider(metaclass=_SecretsProviderMetaclass): name = None |
