From 5880769cbac1362aad46a34faec4c553fa9fcac6 Mon Sep 17 00:00:00 2001 From: Peter LaFosse Date: Thu, 20 Nov 2025 09:58:28 -0500 Subject: Add get and __contains__ methods to all *MetaClass Fixes: https://github.com/Vector35/binaryninja-api/issues/7588 --- python/downloadprovider.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'python/downloadprovider.py') 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 -- cgit v1.3.1