openhound-github 0.3.3__tar.gz → 0.3.4__tar.gz

{openhound_github-0.3.3 → openhound_github-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openhound-github
-Version: 0.3.3
+Version: 0.3.4
 Summary: Github collector for OpenHound
 Author-email: "jdreijer@specterops.io" <jdreijer@specterops.io>, Jared Atkinson <jatkinson@specterops.io>
 License-File: LICENSE.md

{openhound_github-0.3.3 → openhound_github-0.3.4}/src/openhound_github/resources/enterprise.py

@@ -1,3 +1,4 @@
+import logging
 from dataclasses import dataclass
 
 from dlt.sources.helpers.rest_client.client import RESTClient
@@ -28,6 +29,8 @@ from openhound_github.models import (
     EnterpriseUser,
 )
 
+logger = logging.getLogger(__name__)
+
 
 @dataclass
 class SourceContext:
@@ -51,17 +54,24 @@ def enterprise(ctx: SourceContext):
         "variables": {"slug": ctx.enterprise_name, "after": None},
     }
 
-    for page_data in ctx.client.paginate(
-        "/graphql",
-        method="POST",
-        json=data,
-        paginator=paginator,
-        data_selector="data",
-    ):
-        page_enterprise = page_data[0].get("enterprise")
-
-        if page_enterprise:
-            yield page_enterprise
+    try:
+        for page_data in ctx.client.paginate(
+            "/graphql",
+            method="POST",
+            json=data,
+            paginator=paginator,
+            data_selector="data",
+        ):
+            page_enterprise = page_data[0].get("enterprise")
+
+            if page_enterprise:
+                yield page_enterprise
+    except Exception as e:
+        logger.error(
+            f"Error in resource 'enterprise' processing enterprise '{ctx.enterprise_name}': {e}",
+            extra={"resource": "enterprise", "phase": "resource_iteration"},
+        )
+        return
 
 
 @app.transformer(