Add can_create_x lookup

Author: d3vzer0

src/openhound_github/lookup.py

@@ -1,7 +1,8 @@
 from functools import lru_cache
 
+import duckdb
 from duckdb import DuckDBPyConnection
-from openhound.core.lookup import LookupManager
+from openhound.core.lookup import LookupManager, logger
 
 
 class GithubLookup(LookupManager):
@@ -10,6 +11,20 @@ def __init__(self, client: DuckDBPyConnection, schema: str = "github"):
         self.schema = schema
         self.client = client
 
+    def _find_single_row(self, *args):
+        try:
+            self.client.execute(*args)
+            result = self.client.fetchone()
+            return result if result else None
+
+        except duckdb.CatalogException as err:
+            logger.error("DuckDB lookup failed, missing table: %s", err)
+            return None
+
+        except duckdb.Error as err:
+            logger.error("DuckDB lookup query failed: %s", err)
+            return None
+
     @lru_cache
     def org_id(self) -> str | None:
         res = self._find_single_object(
@@ -124,6 +139,18 @@ def role_can_create_branch(self, role_id: str, repository_node_id: str):
             [role_id, repository_node_id],
         )
 
+    @lru_cache
+    def members_can_create_repository(self, org_login: str):
+        return self._find_single_row(
+            f"""SELECT
+                members_can_create_repositories,
+                members_can_create_public_repositories,
+                members_can_create_internal_repositories,
+                members_can_create_private_repositories
+            FROM {self.schema}.organizations WHERE login = ?""",
+            [org_login],
+        )
+
     @lru_cache
     def bypass_pull_request_allowances(self, actor_id: str):
         """Returns the node_ids of users/teams that bypass PR review requirements on branches in a repository (GH_BypassPullRequestAllowances)"""