subscribe updates

headers are added to the websockets connect method
keep alive message is handled without crashing

Author: DrWacker

python_graphql_client/graphql_client.py

@@ -92,13 +92,17 @@ async def subscribe(
         )
 
         async with websockets.connect(
-            self.endpoint, subprotocols=["graphql-ws"]
+            self.endpoint,
+            subprotocols=["graphql-ws"],
+            extra_headers=self.__request_headers(headers)
         ) as websocket:
             await websocket.send(connection_init_message)
             await websocket.send(request_message)
             async for response_message in websocket:
                 response_body = json.loads(response_message)
                 if response_body["type"] == "connection_ack":
                     logging.info("the server accepted the connection")
+                elif response_body["type"] == "ka":
+                    logging.info("the server sent a keep alive message")
                 else:
                     handle(response_body["payload"])