summaryrefslogtreecommitdiff
path: root/python/downloadprovider.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/downloadprovider.py')
-rw-r--r--python/downloadprovider.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/python/downloadprovider.py b/python/downloadprovider.py
index 2fd94ac3..a1ff94bf 100644
--- a/python/downloadprovider.py
+++ b/python/downloadprovider.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
@@ -269,6 +270,23 @@ class _DownloadProviderMetaclass(type):
raise KeyError(f"'{value}' is not a valid download provider")
return DownloadProvider(provider)
+ def __contains__(cls: '_DownloadProviderMetaclass', name: object) -> bool:
+ if not isinstance(name, str):
+ return False
+ try:
+ cls[name]
+ return True
+ except KeyError:
+ return False
+
+ def get(cls: '_DownloadProviderMetaclass', name: str, default: Any = None) -> Optional['DownloadProvider']:
+ try:
+ return cls[name]
+ except KeyError:
+ if default is not None:
+ return default
+ return None
+
class DownloadProvider(metaclass=_DownloadProviderMetaclass):
name = None