try this

Signed-off-by: Andrew Brain <andrewbrain2019@gmail.com>
chaoss · Jul 23, 2024 · 4529a91 · 4529a91
1 parent 0b40bbb
commit 4529a91
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/augur/tasks/github/util/github_graphql_data_access.py b/augur/tasks/github/util/github_graphql_data_access.py
@@ -44,6 +44,8 @@ def paginate_resource(self, query, variables, result_keys):
         result_json = self.make_request_with_retries(query, params).json()
 
         data = self.__extract_data_section(result_keys, result_json)
+        if data is None:
+            raise Exception(f"Error: data section of response is None. Query: {query}. Response: {result_json}")
 
         if self.__get_total_count(data) == 0:
             return
@@ -56,6 +58,8 @@ def paginate_resource(self, query, variables, result_keys):
             result_json = self.make_request_with_retries(query, params).json()
 
             data = self.__extract_data_section(result_keys, result_json)
+            if data is None:
+                raise Exception(f"Error: data section of response is None. Query: {query}. Response: {result_json}")
 
             yield from self.__extract_raw_data_into_list(data)
 
@@ -157,8 +161,6 @@ def __extract_data_section(self, keys, json_response):
 
             core = core[value]
 
-        if core is None:
-            raise Exception(f"Error: data section of response is None. Response: {json_response}")
 
         return core