removed old method. oops.

polygon-io · Nov 20, 2023 · c0552d9 · c0552d9
1 parent dde5fa5
commit c0552d9
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/polygon/websocket/__init__.py b/polygon/websocket/__init__.py
@@ -197,15 +197,26 @@ async def _unsubscribe(self, topics: Union[List[str], Set[str]]):
             self.json.dumps({"action": "unsubscribe", "params": subs})
         )
 
+
     @staticmethod
     def _parse_subscription(s: str):
         s = s.strip()
         split = s.split(".")
-        if len(split) != 2:
-            logger.warning("invalid subscription:", s)
-            return [None, None]
+        length = len(split)
+
+        match length:
+            case _ if length < 2: 
+                logger.warning("invalid subscription:", s)      
+                return [None, None]
+            case 3:
+                return split[0], split[1] + "." + split[2]
+            case _ if length > 3:
+                logger.warning("invalid subscription:", s)
+                return [None, None]
+            case _:
+                return split[0], split[1]
+
 
-        return split
 
     def subscribe(self, *subscriptions: str):
         """