Add get and __contains__ methods to all *MetaClass

Fixes: https://github.com/Vector35/binaryninja-api/issues/7588

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