fix an error in headers

Author: IAmTomahawkx

twitchio/ext/eventsub/models.py

@@ -121,8 +121,14 @@ def __init__(self, frame: dict):
         self.message_type: Literal["notification", "revocation", "reconnect", "session_keepalive"] = meta["message_type"]
         self.message_retry: int = 0 # don't make breaking changes with the Header class
         self.signature: str = ""
-        self.subscription_type: str = frame["payload"]["subscription"]["type"]
-        self.subscription_version: str = frame["payload"]["subscription"]["version"]
+        self.subscription_type: Optional[str]
+        self.subscription_version: Optional[str]
+        if frame["payload"]:
+            self.subscription_type = frame["payload"]["subscription"]["type"]
+            self.subscription_version = frame["payload"]["subscription"]["version"]
+        else:
+            self.subscription_type = None
+            self.subscription_version = None
 
 
 class BaseEvent:
@@ -157,16 +163,19 @@ def __init__(self, client: Union[EventSubClient, EventSubWSClient], _data: Union
             data = _data
 
         self.headers: Union[Headers, WebsocketHeaders]
-        self.subscription: Subscription
 
         if request:
             data: dict = _loads(_data)
             self.headers = Headers(request)
-            self.subscription = Subscription(data["subscription"])
+            self.subscription: Subscription = Subscription(data["subscription"])
             self.setup(data)
         else:
             self.headers = WebsocketHeaders(data)
-            self.subscription = Subscription(data["payload"]["subscription"])
+            self.subscription: Optional[Subscription]
+            if data["payload"]:
+                self.subscription = Subscription(data["payload"]["subscription"])
+            else:
+                self.subscription = None
             self.setup(data["payload"])
 
 

twitchio/ext/eventsub/websocket.py

@@ -180,6 +180,9 @@ async def pump(self) -> None:
                 await self.connect()
                 return
 
+            except Exception as e:
+                logger.error("Exception in the pump function!", exc_info=e)
+
     def parse_frame(self, frame: dict) -> _messages:
         type_: str = frame["metadata"]["message_type"]
         return _message_types[type_](self, frame, None)