Update base.py

gnh1201 · Nov 12, 2024 · 5567325 · 5567325
1 parent 75aec1d
commit 5567325
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/base.py b/base.py
@@ -198,6 +198,19 @@ def get_connector(cls, connection_type):
                 return extension
         return None
 
+    @classmethod
+    def test_connector(cls, data):
+        def test(prelude, data):
+            return data.find(prelude) == 0
+
+        for extension in cls.extensions:
+            if (
+                extension.type == "connector"
+                and test(extension.prelude, data)
+            ):
+                return extension
+        return None
+
     @classmethod
     def send_accept(cls, conn, method, success=True):
         if "tcp" in cls.protocols: