infrahub-server 1.2.4__py3-none-any.whl → 1.2.5__py3-none-any.whl

infrahub_testcontainers/performance_test.py

@@ -1,6 +1,6 @@
 import hashlib
 import json
-from datetime import UTC, datetime
+from datetime import datetime, timezone
 from types import TracebackType
 from typing import Any
 
@@ -35,7 +35,7 @@ class InfrahubPerformanceTest:
         self.env_vars = {}
         self.project_name = ""
         self.test_info = {}
-        self.start_time = datetime.now(UTC)
+        self.start_time = datetime.now(timezone.utc)
         self.end_time: datetime | None = None
         self.results_url = results_url
         self.scraper_endpoint = ""
@@ -57,7 +57,7 @@ class InfrahubPerformanceTest:
 
     def finalize(self, session: pytest.Session) -> None:
         if self.initialized:
-            self.end_time = datetime.now(UTC)
+            self.end_time = datetime.now(timezone.utc)
             self.extract_test_session_information(session)
             self.send_results()
 
@@ -129,7 +129,7 @@ class InfrahubPerformanceTest:
         if not exc_type and self.active_measurements:
             self.add_measurement(
                 definition=self.active_measurements.definition,
-                value=(datetime.now(UTC) - self.active_measurements.start_time).total_seconds() * 1000,
+                value=(datetime.now(timezone.utc) - self.active_measurements.start_time).total_seconds() * 1000,
                 context=self.active_measurements.context,
             )
 

infrahub/message_bus/messages/event_branch_merge.py

@@ -1,13 +0,0 @@
-from pydantic import Field
-
-from infrahub.context import InfrahubContext
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventBranchMerge(InfrahubMessage):
-    """Sent when a branch has been merged."""
-
-    source_branch: str = Field(..., description="The source branch")
-    target_branch: str = Field(..., description="The target branch")
-
-    context: InfrahubContext = Field(..., description="The context of the event")

infrahub/message_bus/operations/event/__init__.py

@@ -1,3 +0,0 @@
-from . import branch
-
-__all__ = ["branch"]

infrahub/message_bus/operations/event/branch.py

@@ -1,61 +0,0 @@
-from __future__ import annotations
-
-from prefect import flow
-
-from infrahub.core import registry
-from infrahub.core.diff.model.path import BranchTrackingId
-from infrahub.core.diff.models import RequestDiffUpdate
-from infrahub.core.diff.repository.repository import DiffRepository
-from infrahub.dependencies.registry import get_component_registry
-from infrahub.log import get_logger
-from infrahub.message_bus import InfrahubMessage, messages
-from infrahub.services import InfrahubServices  # noqa: TC001  needed for prefect flow
-from infrahub.workflows.catalogue import (
-    DIFF_UPDATE,
-    TRIGGER_ARTIFACT_DEFINITION_GENERATE,
-    TRIGGER_GENERATOR_DEFINITION_RUN,
-)
-
-log = get_logger()
-
-
-@flow(name="branch-event-merge")
-async def merge(message: messages.EventBranchMerge, service: InfrahubServices) -> None:
-    async with service.database.start_session() as db:
-        log.info("Branch merged", source_branch=message.source_branch, target_branch=message.target_branch)
-
-        events: list[InfrahubMessage] = [
-            messages.RefreshRegistryBranches(),
-        ]
-        component_registry = get_component_registry()
-        default_branch = registry.get_branch_from_registry()
-        diff_repository = await component_registry.get_component(DiffRepository, db=db, branch=default_branch)
-        # send diff update requests for every branch-tracking diff
-        branch_diff_roots = await diff_repository.get_roots_metadata(base_branch_names=[message.target_branch])
-
-        await service.workflow.submit_workflow(
-            workflow=TRIGGER_ARTIFACT_DEFINITION_GENERATE,
-            context=message.context,
-            parameters={"branch": message.target_branch},
-        )
-
-        await service.workflow.submit_workflow(
-            workflow=TRIGGER_GENERATOR_DEFINITION_RUN,
-            context=message.context,
-            parameters={"branch": message.target_branch},
-        )
-
-        for diff_root in branch_diff_roots:
-            if (
-                diff_root.base_branch_name != diff_root.diff_branch_name
-                and diff_root.tracking_id
-                and isinstance(diff_root.tracking_id, BranchTrackingId)
-            ):
-                request_diff_update_model = RequestDiffUpdate(branch_name=diff_root.diff_branch_name)
-                await service.workflow.submit_workflow(
-                    workflow=DIFF_UPDATE, context=message.context, parameters={"model": request_diff_update_model}
-                )
-
-        for event in events:
-            event.assign_meta(parent=message)
-            await service.message_bus.send(message=event)