Update base.py

gnh1201 · Nov 12, 2024 · ed91362 · ed91362
1 parent c19a38a
commit ed91362
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/base.py b/base.py
@@ -200,13 +200,16 @@ def get_connector(cls, connection_type):
 
     @classmethod
     def test_connector(cls, data):
-        def test(prelude, data):
-            return data.find(prelude) == 0
+        def test(preludes, data):
+            for prelude in preludes:
+                if data.find(prelude) == 0:
+                    return True
+            return False
 
         for extension in cls.extensions:
             if (
                 extension.type == "connector"
-                and test(extension.prelude, data)
+                and test(extension.preludes, data)
             ):
                 return extension
         return None