infrahub-server 1.1.3__py3-none-any.whl → 1.1.5__py3-none-any.whl

infrahub/transformations/models.py

@@ -10,7 +10,8 @@ class TransformPythonData(BaseModel):
     data: dict = Field(..., description="Input data for the template")
     branch: str = Field(..., description="The branch to target")
     transform_location: str = Field(..., description="Location of the transform within the repository")
-    commit: str = Field(..., description="The commit id to use when rendering the template")
+    commit: str = Field(..., description="The commit id to use when generating the artifact")
+    timeout: int = Field(..., description="The timeout value to use when generating the artifact")
 
 
 class TransformJinjaTemplateData(BaseModel):
@@ -23,3 +24,4 @@ class TransformJinjaTemplateData(BaseModel):
     branch: str = Field(..., description="The branch to target")
     template_location: str = Field(..., description="Location of the template within the repository")
     commit: str = Field(..., description="The commit id to use when rendering the template")
+    timeout: int = Field(..., description="The timeout value to use when rendering the template")

infrahub/transformations/tasks.py

@@ -25,7 +25,7 @@ async def transform_python(message: TransformPythonData) -> Any:
         commit=message.commit,
     )
 
-    transformed_data = await repo.execute_python_transform(
+    transformed_data = await repo.execute_python_transform.with_options(timeout_seconds=message.timeout)(
         branch_name=message.branch,
         commit=message.commit,
         location=message.transform_location,
@@ -49,7 +49,7 @@ async def transform_render_jinja2_template(message: TransformJinjaTemplateData)
         commit=message.commit,
     )
 
-    rendered_template = await repo.render_jinja2_template(
+    rendered_template = await repo.render_jinja2_template.with_options(timeout_seconds=message.timeout)(
         commit=message.commit, location=message.template_location, data={"data": message.data}
     )
 

infrahub/webhook/models.py

@@ -6,11 +6,13 @@ from math import floor
 from typing import Any, Optional, Union
 from uuid import uuid4
 
+from infrahub_sdk.protocols import CoreTransformPython
 from pydantic import BaseModel, ConfigDict, Field
 
 from infrahub.core.constants import InfrahubKind
 from infrahub.git.repository import InfrahubReadOnlyRepository, InfrahubRepository
 from infrahub.services import InfrahubServices
+from infrahub.transformations.constants import DEFAULT_TRANSFORM_TIMEOUT
 
 
 class SendWebhookData(BaseModel):
@@ -95,7 +97,13 @@ class TransformWebhook(Webhook):
         default_branch = repo.default_branch
         commit = repo.get_commit_value(branch_name=default_branch)
 
-        self._payload = await repo.execute_python_transform(
+        timeout = DEFAULT_TRANSFORM_TIMEOUT
+        if transform := await self.service.client.get(
+            kind=CoreTransformPython, name__value=self.transform_name, raise_when_missing=False
+        ):
+            timeout = transform.timeout.value
+
+        self._payload = await repo.execute_python_transform.with_options(timeout_seconds=timeout)(
             branch_name=default_branch,
             commit=commit,
             location=f"{self.transform_file}::{self.transform_class}",

infrahub/workflows/catalogue.py

@@ -57,7 +57,7 @@ TRIGGER_ARTIFACT_DEFINITION_GENERATE = WorkflowDefinition(
 )
 
 TRIGGER_GENERATOR_DEFINITION_RUN = WorkflowDefinition(
-    name="generator_definition_run",
+    name="generator-definition-run",
     type=WorkflowType.CORE,
     module="infrahub.generators.tasks",
     function="run_generator_definition",
@@ -81,7 +81,7 @@ REQUEST_GENERATOR_RUN = WorkflowDefinition(
 )
 
 REQUEST_GENERATOR_DEFINITION_RUN = WorkflowDefinition(
-    name="request_generator_definition_run",
+    name="request-generator-definition-run",
     type=WorkflowType.CORE,
     module="infrahub.generators.tasks",
     function="request_generator_definition_run",

infrahub_sdk/analyzer.py

@@ -91,7 +91,7 @@ class GraphQLQueryAnalyzer:
                     else:
                         data["default_value"] = variable.default_value.value
 
-                if not data.get("default_value", None) and non_null:
+                if not data.get("default_value") and non_null:
                     data["required"] = True
 
                 response.append(GraphQLQueryVariable(**data))

infrahub_sdk/checks.py

@@ -8,9 +8,10 @@ from abc import abstractmethod
 from typing import TYPE_CHECKING, Any
 
 import ujson
-from git.repo import Repo
 from pydantic import BaseModel, Field
 
+from infrahub_sdk.repository import GitRepoManager
+
 from .exceptions import UninitializedError
 
 if TYPE_CHECKING:
@@ -43,7 +44,7 @@ class InfrahubCheck:
         params: dict | None = None,
         client: InfrahubClient | None = None,
     ):
-        self.git: Repo | None = None
+        self.git: GitRepoManager | None = None
         self.initializer = initializer or InfrahubCheckInitializer()
 
         self.logs: list[dict[str, Any]] = []
@@ -137,10 +138,9 @@ class InfrahubCheck:
             return self.branch
 
         if not self.git:
-            self.git = Repo(self.root_directory)
+            self.git = GitRepoManager(self.root_directory)
 
         self.branch = str(self.git.active_branch)
-
         return self.branch
 
     @abstractmethod

infrahub_sdk/client.py

@@ -547,8 +547,10 @@ class InfrahubClient(BaseClient):
         at: Timestamp | None = None,
         branch: str | None = None,
         timeout: int | None = None,
+        **kwargs: Any,
     ) -> int:
         """Return the number of nodes of a given kind."""
+        filters = kwargs
         schema = await self.schema.get(kind=kind, branch=branch)
 
         branch = branch or self.default_branch
@@ -556,7 +558,10 @@ class InfrahubClient(BaseClient):
             at = Timestamp(at)
 
         response = await self.execute_graphql(
-            query=Query(query={schema.kind: {"count": None}}).render(), branch_name=branch, at=at, timeout=timeout
+            query=Query(query={schema.kind: {"count": None, "@filters": filters}}).render(),
+            branch_name=branch,
+            at=at,
+            timeout=timeout,
         )
         return int(response.get(schema.kind, {}).get("count", 0))
 
@@ -781,7 +786,7 @@ class InfrahubClient(BaseClient):
             nodes = []
             related_nodes = []
             batch_process = await self.create_batch()
-            count = await self.count(kind=schema.kind)
+            count = await self.count(kind=schema.kind, **filters)
             total_pages = (count + pagination_size - 1) // pagination_size
 
             for page_number in range(1, total_pages + 1):
@@ -1181,7 +1186,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_address(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -1196,7 +1201,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_address(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -1211,7 +1216,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_address(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -1328,7 +1333,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,
@@ -1344,7 +1349,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,
@@ -1360,7 +1365,7 @@ class InfrahubClient(BaseClient):
     async def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNode,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,
@@ -1651,8 +1656,10 @@ class InfrahubClientSync(BaseClient):
         at: Timestamp | None = None,
         branch: str | None = None,
         timeout: int | None = None,
+        **kwargs: Any,
     ) -> int:
         """Return the number of nodes of a given kind."""
+        filters = kwargs
         schema = self.schema.get(kind=kind, branch=branch)
 
         branch = branch or self.default_branch
@@ -1660,7 +1667,10 @@ class InfrahubClientSync(BaseClient):
             at = Timestamp(at)
 
         response = self.execute_graphql(
-            query=Query(query={schema.kind: {"count": None}}).render(), branch_name=branch, at=at, timeout=timeout
+            query=Query(query={schema.kind: {"count": None, "@filters": filters}}).render(),
+            branch_name=branch,
+            at=at,
+            timeout=timeout,
         )
         return int(response.get(schema.kind, {}).get("count", 0))
 
@@ -1920,7 +1930,7 @@ class InfrahubClientSync(BaseClient):
             related_nodes = []
             batch_process = self.create_batch()
 
-            count = self.count(kind=schema.kind)
+            count = self.count(kind=schema.kind, **filters)
             total_pages = (count + pagination_size - 1) // pagination_size
 
             for page_number in range(1, total_pages + 1):
@@ -2296,7 +2306,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_address(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -2311,7 +2321,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_address(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -2326,7 +2336,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_address(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         address_type: str | None = ...,
@@ -2439,7 +2449,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,
@@ -2455,7 +2465,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,
@@ -2471,7 +2481,7 @@ class InfrahubClientSync(BaseClient):
     def allocate_next_ip_prefix(
         self,
         resource_pool: CoreNodeSync,
-        kind: Literal[None] = ...,
+        kind: None = ...,
         identifier: str | None = ...,
         prefix_length: int | None = ...,
         member_type: str | None = ...,

infrahub_sdk/ctl/cli_commands.py

@@ -130,12 +130,12 @@ async def run(
     debug: bool = False,
     _: str = CONFIG_PARAM,
     branch: str = typer.Option("main", help="Branch on which to run the script."),
-    concurrent: int = typer.Option(
-        4,
+    concurrent: int | None = typer.Option(
+        None,
         help="Maximum number of requests to execute at the same time.",
-        envvar="INFRAHUBCTL_CONCURRENT_EXECUTION",
+        envvar="INFRAHUB_MAX_CONCURRENT_EXECUTION",
     ),
-    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUBCTL_TIMEOUT"),
+    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUB_TIMEOUT"),
     variables: list[str] | None = typer.Argument(
         None, help="Variables to pass along with the query. Format key=value key=value."
     ),

infrahub_sdk/ctl/exporter.py

@@ -26,9 +26,9 @@ def dump(
     concurrent: int = typer.Option(
         4,
         help="Maximum number of requests to execute at the same time.",
-        envvar="INFRAHUBCTL_CONCURRENT_EXECUTION",
+        envvar="INFRAHUB_MAX_CONCURRENT_EXECUTION",
     ),
-    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUBCTL_TIMEOUT"),
+    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUB_TIMEOUT"),
     exclude: list[str] = typer.Option(
         ["CoreAccount"],
         help="Prevent node kind(s) from being exported, CoreAccount is excluded by default",

infrahub_sdk/ctl/importer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from asyncio import run as aiorun
 from pathlib import Path
 
@@ -24,12 +26,12 @@ def load(
     quiet: bool = typer.Option(False, help="No console output"),
     _: str = CONFIG_PARAM,
     branch: str = typer.Option("main", help="Branch from which to export"),
-    concurrent: int = typer.Option(
-        4,
+    concurrent: int | None = typer.Option(
+        None,
         help="Maximum number of requests to execute at the same time.",
-        envvar="INFRAHUBCTL_CONCURRENT_EXECUTION",
+        envvar="INFRAHUB_MAX_CONCURRENT_EXECUTION",
     ),
-    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUBCTL_TIMEOUT"),
+    timeout: int = typer.Option(60, help="Timeout in sec", envvar="INFRAHUB_TIMEOUT"),
 ) -> None:
     """Import nodes and their relationships into the database."""
     console = Console()

infrahub_sdk/ctl/repository.py

@@ -6,13 +6,14 @@ import typer
 import yaml
 from pydantic import ValidationError
 from rich.console import Console
+from rich.table import Table
 
 from infrahub_sdk.ctl.client import initialize_client
 
 from ..async_typer import AsyncTyper
 from ..ctl.exceptions import FileNotValidError
 from ..ctl.utils import init_logging
-from ..graphql import Mutation
+from ..graphql import Mutation, Query
 from ..schema.repository import InfrahubRepositoryConfig
 from ._file import read_file
 from .parameters import CONFIG_PARAM
@@ -102,3 +103,57 @@ async def add(
     )
 
     await client.execute_graphql(query=query.render(), branch_name=branch, tracker="mutation-repository-create")
+
+
+@app.command()
+async def list(
+    branch: str | None = None,
+    debug: bool = False,
+    _: str = CONFIG_PARAM,
+) -> None:
+    init_logging(debug=debug)
+
+    client = initialize_client(branch=branch)
+
+    repo_status_query = {
+        "CoreGenericRepository": {
+            "edges": {
+                "node": {
+                    "__typename": None,
+                    "name": {"value": None},
+                    "operational_status": {"value": None},
+                    "sync_status": {"value": None},
+                    "internal_status": {"value": None},
+                    "... on CoreReadOnlyRepository": {
+                        "ref": {"value": None},
+                    },
+                }
+            }
+        },
+    }
+
+    query = Query(name="GetRepositoryStatus", query=repo_status_query)
+    resp = await client.execute_graphql(query=query.render(), branch_name=branch, tracker="query-repository-list")
+
+    table = Table(title="List of all Repositories")
+
+    table.add_column("Name", justify="right", style="cyan", no_wrap=True)
+    table.add_column("Type")
+    table.add_column("Operational status")
+    table.add_column("Sync status")
+    table.add_column("Internal status")
+    table.add_column("Ref")
+
+    for repository_node in resp["CoreGenericRepository"]["edges"]:
+        repository = repository_node["node"]
+
+        table.add_row(
+            repository["name"]["value"],
+            repository["__typename"],
+            repository["operational_status"]["value"],
+            repository["sync_status"]["value"],
+            repository["internal_status"]["value"],
+            repository["ref"]["value"] if "ref" in repository else "",
+        )
+
+    console.print(table)

infrahub_sdk/generator.py

@@ -4,7 +4,7 @@ import os
 from abc import abstractmethod
 from typing import TYPE_CHECKING
 
-from git.repo import Repo
+from infrahub_sdk.repository import GitRepoManager
 
 from .exceptions import UninitializedError
 
@@ -30,7 +30,7 @@ class InfrahubGenerator:
     ) -> None:
         self.query = query
         self.branch = branch
-        self.git: Repo | None = None
+        self.git: GitRepoManager | None = None
         self.params = params or {}
         self.root_directory = root_directory or os.getcwd()
         self.generator_instance = generator_instance
@@ -81,7 +81,7 @@ class InfrahubGenerator:
             return self.branch
 
         if not self.git:
-            self.git = Repo(self.root_directory)
+            self.git = GitRepoManager(self.root_directory)
 
         self.branch = str(self.git.active_branch)
 

infrahub_sdk/node.py

@@ -1074,7 +1074,7 @@ class InfrahubNode(InfrahubNodeBase):
         timeout: int | None = None,
     ) -> Self:
         if not schema:
-            node_kind = data.get("__typename", None) or data.get("node", {}).get("__typename", None)
+            node_kind = data.get("__typename") or data.get("node", {}).get("__typename", None)
             if not node_kind:
                 raise ValueError("Unable to determine the type of the node, __typename not present in data")
             schema = await client.schema.get(kind=node_kind, branch=branch, timeout=timeout)
@@ -1594,7 +1594,7 @@ class InfrahubNodeSync(InfrahubNodeBase):
         timeout: int | None = None,
     ) -> Self:
         if not schema:
-            node_kind = data.get("__typename", None) or data.get("node", {}).get("__typename", None)
+            node_kind = data.get("__typename") or data.get("node", {}).get("__typename", None)
             if not node_kind:
                 raise ValueError("Unable to determine the type of the node, __typename not present in data")
             schema = client.schema.get(kind=node_kind, branch=branch, timeout=timeout)

infrahub_sdk/pytest_plugin/items/base.py

@@ -6,7 +6,6 @@ from typing import TYPE_CHECKING, Any
 
 import pytest
 import ujson
-from git.exc import InvalidGitRepositoryError
 
 from ..exceptions import InvalidResourceConfigError
 from ..models import InfrahubInputOutputTest
@@ -28,7 +27,6 @@ class InfrahubItem(pytest.Item):
         **kwargs: dict[str, Any],
     ):
         super().__init__(*args, **kwargs)  # type: ignore[arg-type]
-
         self.resource_name: str = resource_name
         self.resource_config: InfrahubRepositoryConfigElement = resource_config
         self.test: InfrahubTest = test
@@ -68,9 +66,6 @@ class InfrahubItem(pytest.Item):
         """Run the test logic."""
 
     def repr_failure(self, excinfo: pytest.ExceptionInfo, style: str | None = None) -> str:  # noqa: ARG002
-        if isinstance(excinfo.value, InvalidGitRepositoryError):
-            return f"Invalid Git repository at {excinfo.value}"
-
         return str(excinfo.value)
 
     def reportinfo(self) -> tuple[Path | str, int | None, str]:

infrahub_sdk/pytest_plugin/items/graphql_query.py

@@ -40,7 +40,7 @@ class InfrahubGraphQLQueryItem(InfrahubItem):
             )
 
         if isinstance(excinfo.value, OutputMatchError):
-            return "\n".join([excinfo.value.message, excinfo.value.differences])
+            return f"{excinfo.value.message}\n{excinfo.value.differences}"
 
         return super().repr_failure(excinfo, style=style)
 

infrahub_sdk/pytest_plugin/items/jinja2_transform.py

@@ -79,7 +79,7 @@ class InfrahubJinja2Item(InfrahubItem):
             return "\n".join(["Syntax error detected in the template", excinfo.value.message or ""])
 
         if isinstance(excinfo.value, OutputMatchError):
-            return "\n".join([excinfo.value.message, excinfo.value.differences])
+            return f"{excinfo.value.message}\n{excinfo.value.differences}"
 
         return super().repr_failure(excinfo, style=style)
 

infrahub_sdk/pytest_plugin/items/python_transform.py

@@ -62,7 +62,7 @@ class InfrahubPythonTransformItem(InfrahubItem):
             )
 
         if isinstance(excinfo.value, OutputMatchError):
-            return "\n".join([excinfo.value.message, excinfo.value.differences])
+            return f"{excinfo.value.message}\n{excinfo.value.differences}"
 
         return super().repr_failure(excinfo, style=style)
 

infrahub_sdk/repository.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from dulwich import porcelain
+from dulwich.repo import Repo
+
+
+class GitRepoManager:
+    def __init__(self, root_directory: str, branch: str = "main"):
+        self.root_directory = root_directory
+        self.branch = branch
+        self.git: Repo = self.initialize_repo()
+
+    def initialize_repo(self) -> Repo:
+        # Check if the directory already has a repository
+
+        root_path = Path(self.root_directory)
+
+        if root_path.exists() and (root_path / ".git").is_dir():
+            repo = Repo(self.root_directory)  # Open existing repo
+        else:
+            repo = Repo.init(self.root_directory, default_branch=self.branch.encode("utf-8"))
+
+        if not repo:
+            raise ValueError("Failed to initialize or open a repository.")
+
+        return repo
+
+    @property
+    def active_branch(self) -> str | None:
+        active_branch = porcelain.active_branch(self.root_directory).decode("utf-8")
+        return active_branch

infrahub_sdk/testing/repository.py

@@ -7,10 +7,11 @@ from enum import Enum
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-from git.repo import Repo
+from dulwich import porcelain
 
 from infrahub_sdk.graphql import Mutation
 from infrahub_sdk.protocols import CoreGenericRepository
+from infrahub_sdk.repository import GitRepoManager
 
 if TYPE_CHECKING:
     from infrahub_sdk import InfrahubClient
@@ -37,14 +38,14 @@ class GitRepo:
 
     type: GitRepoType = GitRepoType.INTEGRATED
 
-    _repo: Repo | None = None
+    _repo: GitRepoManager | None = None
     initial_branch: str = "main"
     directories_to_ignore: list[str] = field(default_factory=list)
     remote_directory_name: str = "/remote"
     _branches: list[str] = field(default_factory=list)
 
     @property
-    def repo(self) -> Repo:
+    def repo(self) -> GitRepoManager:
         if self._repo:
             return self._repo
         raise ValueError("Repo hasn't been initialized yet")
@@ -62,12 +63,17 @@ class GitRepo:
             dst=self.dst_directory / self.name,
             ignore=shutil.ignore_patterns(".git"),
         )
-        self._repo = Repo.init(self.dst_directory / self.name, initial_branch=self.initial_branch)
-        for untracked in self.repo.untracked_files:
-            self.repo.index.add(untracked)
-        self.repo.index.commit("First commit")
 
-        self.repo.git.checkout(self.initial_branch)
+        self._repo = GitRepoManager(str(Path(self.dst_directory / self.name)), branch=self.initial_branch)
+
+        files = list(
+            porcelain.get_untracked_paths(self._repo.git.path, self._repo.git.path, self._repo.git.open_index())
+        )
+        files_to_add = [str(Path(self._repo.git.path) / t) for t in files]
+        if files_to_add:
+            porcelain.add(repo=self._repo.git.path, paths=files_to_add)
+            porcelain.commit(repo=self._repo.git.path, message="First commit")
+            porcelain.checkout_branch(self._repo.git, self.initial_branch.encode("utf-8"))
 
     async def add_to_infrahub(self, client: InfrahubClient, branch: str | None = None) -> dict:
         input_data = {

infrahub_sdk/transforms.py

@@ -5,7 +5,7 @@ import os
 from abc import abstractmethod
 from typing import TYPE_CHECKING, Any
 
-from git import Repo
+from infrahub_sdk.repository import GitRepoManager
 
 from .exceptions import UninitializedError
 
@@ -27,7 +27,7 @@ class InfrahubTransform:
         server_url: str = "",
         client: InfrahubClient | None = None,
     ):
-        self.git: Repo
+        self.git: GitRepoManager
 
         self.branch = branch
         self.server_url = server_url or os.environ.get("INFRAHUB_URL", "http://127.0.0.1:8000")
@@ -56,7 +56,7 @@ class InfrahubTransform:
             return self.branch
 
         if not hasattr(self, "git") or not self.git:
-            self.git = Repo(self.root_directory)
+            self.git = GitRepoManager(self.root_directory)
 
         self.branch = str(self.git.active_branch)