From 5880769cbac1362aad46a34faec4c553fa9fcac6 Mon Sep 17 00:00:00 2001
From: Peter LaFosse <peter@vector35.com>
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/websocketprovider.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'python/websocketprovider.py')

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)
-- 
cgit v1.3.1