Merge pull request #12 from CodaProtocol/fix/error-handling

[Fix] Error Handling New Block Subscription

Author: yourbuddyconner

CodaClient.py

@@ -76,9 +76,10 @@ def _graphql_request(self, query: str, variables: dict = {}):
     }
     self.logger.debug("Sending a Query: {}".format(payload))
     response = requests.post(self.endpoint, json=payload, headers=headers)
-    if response.status_code == 200:
+    resp_json = response.json()
+    if response.status_code == 200 and "errors" not in resp_json:
       self.logger.debug("Got a Response: {}".format(response.json()))
-      return response.json()
+      return resp_json
     else:
       print(response.text)
       raise Exception(
@@ -97,6 +98,7 @@ async def _graphql_subscription(self, query: str, variables: dict = {}, callback
     self.logger.info("Listening to GraphQL Subscription...")
 
     uri = self.websocket_endpoint
+    self.logger.info(uri)
     async with websockets.client.connect(uri, ping_timeout=None) as websocket:
       # Set up Websocket Connection
       self.logger.debug("WEBSOCKET -- Sending Hello Message: {}".format(hello_message))
@@ -464,15 +466,15 @@ async def listen_block_confirmations(self, callback):
     '''
     await self._graphql_subscription(query, {}, callback)
 
-  async def listen_new_blocks(self, pk: str, callback):
-    """Creates a subscription for new blocks created by a proposer using a particular private key.
+  async def listen_new_blocks(self, callback):
+    """Creates a subscription for new blocks, calls `callback` each time the subscription fires.
     
     Arguments:
-        pk {PublicKey} -- The public key to use to filter blocks
+        callback(block) {coroutine} -- This coroutine is executed with the new block as an argument each time the subscription fires
     """
     query = '''
-    subscription($pk:PublicKey){
-      newBlock(publicKey:$pk){
+    subscription(){
+      newBlock(){
         creator
         stateHash
         protocolState {

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name='CodaClient',
-    version='0.0.13',
+    version='0.0.14',
     python_requires='>=3.5',
     description='A Python wrapper around the Coda Daemon GraphQL API.',
     github='http://github.com/CodaProtocol/coda-python',