error in query_result_id throws exception #3

Author: itzmestar

duneanalytics/__version__.py

@@ -2,7 +2,7 @@
 __title__ = 'duneanalytics'
 __description__ = 'Unofficial library for Dune Analytics.'
 __url__ = 'https://github.com/itzmestar/duneanalytics'
-__version__ = '1.0.2'
+__version__ = '1.0.3'
 __build__ = 0x010001
 __author__ = 'Tarique Anwer'
 __author_email__ = '[email protected]'

duneanalytics/duneanalytics.py

@@ -78,6 +78,8 @@ def fetch_auth_token(self):
         response = self.session.post(session_url)
         if response.status_code == 200:
             self.token = response.json().get('token')
+        else:
+            print(response.text)
 
     def query_result_id(self, query_id):
         """
@@ -98,6 +100,8 @@ def query_result_id(self, query_id):
         if response.status_code == 200:
             data = response.json()
             print(data)
+            if 'error' in data:
+                return None
             result_id = data.get('data').get('get_result').get('result_id')
             return result_id
         else: