Fix result paging issue

log2timeline · Oct 8, 2024 · 882c8e2 · 882c8e2
1 parent 5637f40
commit 882c8e2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dftimewolf/lib/collectors/gcp_logging.py b/dftimewolf/lib/collectors/gcp_logging.py
@@ -79,7 +79,11 @@ def ListPages(self, logging_client: Any) -> Any:
           order_by=logging.DESCENDING,
           filter_=self._filter_expression,
           page_size=1000)
-    return results.pages
+
+    if hasattr(results, 'pages'):
+      return results.pages
+
+    yield results
 
   def ProcessPages(self, pages: Any, backoff_multiplier: int,
     output_file: Any, output_path: str) -> str: