infrahub-server 1.6.3__py3-none-any.whl → 1.7.0b0__py3-none-any.whl

infrahub/trigger/models.py

@@ -101,6 +101,7 @@ class TriggerType(StrEnum):
     COMPUTED_ATTR_PYTHON_QUERY = "computed_attr_python_query"
     DISPLAY_LABEL_JINJA2 = "display_label_jinja2"
     HUMAN_FRIENDLY_ID = "human_friendly_id"
+    PROFILE = "profile"
     # OBJECT = "object"
 
     @property

infrahub/trigger/setup.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Awaitable, Callable
+from typing import TYPE_CHECKING, Awaitable, Callable, Sequence
 
 from prefect import get_run_logger, task
 from prefect.automations import AutomationCore
@@ -46,7 +46,7 @@ def compare_automations(
 
 @task(name="trigger-setup-specific", task_run_name="Setup triggers of a specific kind", cache_policy=NONE)  # type: ignore[arg-type]
 async def setup_triggers_specific(
-    gatherer: Callable[[InfrahubDatabase | None], Awaitable[list[TriggerDefinition]]],
+    gatherer: Callable[[InfrahubDatabase | None], Awaitable[Sequence[TriggerDefinition]]],
     trigger_type: TriggerType,
     db: InfrahubDatabase | None = None,
 ) -> TriggerSetupReport:
@@ -69,7 +69,7 @@ async def setup_triggers_specific(
 @task(name="trigger-setup", task_run_name="Setup triggers", cache_policy=NONE)
 async def setup_triggers(
     client: PrefectClient,
-    triggers: list[TriggerDefinition],
+    triggers: Sequence[TriggerDefinition],
     trigger_type: TriggerType | None = None,
     force_update: bool = False,
 ) -> TriggerSetupReport:

infrahub/trigger/tasks.py

@@ -8,6 +8,7 @@ from infrahub.computed_attribute.gather import (
 )
 from infrahub.display_labels.gather import gather_trigger_display_labels_jinja2
 from infrahub.hfid.gather import gather_trigger_hfid
+from infrahub.profiles.gather import gather_trigger_profile_refresh
 from infrahub.trigger.catalogue import builtin_triggers
 from infrahub.webhook.gather import gather_trigger_webhook
 from infrahub.workers.dependencies import get_database
@@ -28,6 +29,7 @@ async def trigger_configure_all() -> None:
             computed_attribute_python_query_triggers,
         ) = await gather_trigger_computed_attribute_python(db=db)
         action_rules = await gather_trigger_action_rules(db=db)
+        profile_refresh_triggers = await gather_trigger_profile_refresh()
         triggers = (
             computed_attribute_j2_triggers
             + computed_attribute_python_triggers
@@ -35,6 +37,7 @@ async def trigger_configure_all() -> None:
             + display_label_triggers
             + human_friendly_id_triggers
             + builtin_triggers
+            + profile_refresh_triggers
             + webhook_trigger
             + action_rules
         )

infrahub/validators/tasks.py

@@ -29,6 +29,7 @@ async def start_validator[ValidatorType: CoreValidator](
         validator = cast("ValidatorType", validator)
     else:
         data["proposed_change"] = proposed_change
+        client.request_context = context.to_request_context()
         validator = await client.create(kind=validator_type, data=data)
         await validator.save()
 

infrahub/webhook/models.py

@@ -233,7 +233,7 @@ class TransformWebhook(Webhook):
             convert_query_response=self.convert_query_response,
             data={"data": {"data": data, **context.model_dump()}},
             client=client,
-        )  # type: ignore[misc]
+        )  # type: ignore[call-overload]
 
     @classmethod
     def from_object(cls, obj: CoreCustomWebhook, transform: CoreTransformPython) -> Self:

infrahub/webhook/tasks.py

@@ -118,7 +118,7 @@ async def configure_webhook_all() -> None:
         triggers = await gather_trigger_webhook(db=db)
 
     log.info(f"{len(triggers)} Webhooks automation configuration completed")
-    await setup_triggers_specific(gatherer=gather_trigger_webhook, db=database, trigger_type=TriggerType.WEBHOOK)  # type: ignore[misc]
+    await setup_triggers_specific(gatherer=gather_trigger_webhook, db=database, trigger_type=TriggerType.WEBHOOK)  # type: ignore[arg-type]
 
 
 @flow(name="webhook-setup-automation-one", flow_run_name="Configurate webhook for {webhook_name}")

infrahub/workers/dependencies.py

@@ -60,9 +60,15 @@ def get_installation_type(installation_type: str = Depends(build_installation_ty
     return installation_type
 
 
-async def build_database() -> InfrahubDatabase:
-    if "database" not in _singletons:
-        _singletons["database"] = InfrahubDatabase(driver=await get_db(retry=5))
+async def build_database(singleton: bool = True) -> InfrahubDatabase:
+    if not singleton or "database" not in _singletons:
+        db = InfrahubDatabase(driver=await get_db(retry=5))
+
+        if singleton:
+            _singletons["database"] = db
+
+        return db
+
     return _singletons["database"]
 
 

infrahub/workers/infrahub_async.py

@@ -24,6 +24,7 @@ from infrahub.core import registry
 from infrahub.core.initialization import initialization
 from infrahub.database.graph import validate_graph_version
 from infrahub.dependencies.registry import build_component_registry
+from infrahub.exceptions import InitializationError
 from infrahub.git import initialize_repositories_directory
 from infrahub.lock import initialize_lock
 from infrahub.services import InfrahubServices
@@ -182,9 +183,17 @@ class InfrahubWorkerAsync(BaseWorker):
     async def _init_infrahub_client(self, client: InfrahubClient | None = None) -> InfrahubClient:
         if not client:
             self._logger.debug(f"Using Infrahub API at {config.SETTINGS.main.internal_address}")
-            client = InfrahubClient(
-                config=Config(address=config.SETTINGS.main.internal_address, retry_on_failure=True, log=self._logger)
-            )
+            try:
+                client = InfrahubClient(
+                    config=Config(
+                        address=config.SETTINGS.main.infrahub_address, retry_on_failure=True, log=self._logger
+                    )
+                )
+            except InitializationError as err:
+                self._logger.error(
+                    "Infrahub client initialization failed due to missing configuration for internal_address."
+                )
+                raise typer.Exit(1) from err
 
         try:
             await client.branch.all()
@@ -194,7 +203,7 @@ class InfrahubWorkerAsync(BaseWorker):
 
         return client
 
-    async def _init_services(self, client: InfrahubClient) -> None:
+    async def _init_services(self, client: InfrahubClient | None) -> None:
         client = await self._init_infrahub_client(client=client)
 
         service = await InfrahubServices.new(

infrahub/workflows/catalogue.py

@@ -623,6 +623,23 @@ PROFILE_REFRESH = WorkflowDefinition(
 )
 
 
+PROFILE_REFRESH_SETUP = WorkflowDefinition(
+    name="profile-refresh-setup",
+    type=WorkflowType.CORE,
+    module="infrahub.profiles.tasks",
+    function="profile_refresh_setup",
+)
+
+
+PROFILE_REFRESH_PROCESS = WorkflowDefinition(
+    name="profile-refresh-process",
+    type=WorkflowType.CORE,
+    module="infrahub.profiles.tasks",
+    function="profile_refresh_process",
+    tags=[WorkflowTag.DATABASE_CHANGE],
+)
+
+
 CLEAN_UP_DEADLOCKS = WorkflowDefinition(
     name="clean-up-deadlocks",
     type=WorkflowType.INTERNAL,
@@ -685,6 +702,8 @@ WORKFLOWS = [
     IPAM_RECONCILIATION,
     PROFILE_REFRESH,
     PROFILE_REFRESH_MULTIPLE,
+    PROFILE_REFRESH_PROCESS,
+    PROFILE_REFRESH_SETUP,
     PROPOSED_CHANGE_MERGE,
     QUERY_COMPUTED_ATTRIBUTE_TRANSFORM_TARGETS,
     REMOVE_ADD_NODE_FROM_GROUP,

infrahub_sdk/node/constants.py

@@ -20,3 +20,4 @@ ARTIFACT_DEFINITION_GENERATE_FEATURE_NOT_SUPPORTED_MESSAGE = (
 HIERARCHY_FETCH_FEATURE_NOT_SUPPORTED_MESSAGE = "Hierarchical fields are not supported for this node."
 
 HFID_STR_SEPARATOR = "__"
+PROFILE_KIND_PREFIX = "Profile"

infrahub_sdk/node/related_node.py

@@ -1,12 +1,11 @@
 from __future__ import annotations
 
+import re
 from typing import TYPE_CHECKING, Any
 
-from ..exceptions import (
-    Error,
-)
+from ..exceptions import Error
 from ..protocols_base import CoreNodeBase
-from .constants import PROPERTIES_FLAG, PROPERTIES_OBJECT
+from .constants import PROFILE_KIND_PREFIX, PROPERTIES_FLAG, PROPERTIES_OBJECT
 
 if TYPE_CHECKING:
     from ..client import InfrahubClient, InfrahubClientSync
@@ -40,6 +39,7 @@ class RelatedNodeBase:
         self._display_label: str | None = None
         self._typename: str | None = None
         self._kind: str | None = None
+        self._source_typename: str | None = None
 
         if isinstance(data, (CoreNodeBase)):
             self._peer = data
@@ -74,6 +74,8 @@ class RelatedNodeBase:
                 prop_data = properties_data.get(prop, properties_data.get(f"_relation__{prop}", None))
                 if prop_data and isinstance(prop_data, dict) and "id" in prop_data:
                     setattr(self, prop, prop_data["id"])
+                    if prop == "source" and "__typename" in prop_data:
+                        self._source_typename = prop_data["__typename"]
                 elif prop_data and isinstance(prop_data, (str, bool)):
                     setattr(self, prop, prop_data)
                 else:
@@ -125,6 +127,13 @@ class RelatedNodeBase:
             return self._peer.get_kind()
         return self._kind
 
+    @property
+    def is_from_profile(self) -> bool:
+        """Return whether this relationship was set from a profile. Done by checking if the source is of a profile kind."""
+        if not self._source_typename:
+            return False
+        return bool(re.match(rf"^{PROFILE_KIND_PREFIX}[A-Z]", self._source_typename))
+
     def _generate_input_data(self, allocate_from_pool: bool = False) -> dict[str, Any]:
         data: dict[str, Any] = {}
 

infrahub_sdk/node/relationship.py

@@ -56,6 +56,14 @@ class RelationshipManagerBase:
     def has_update(self) -> bool:
         return self._has_update
 
+    @property
+    def is_from_profile(self) -> bool:
+        """Return whether this relationship was set from a profile. All its peers must be from a profile."""
+        if not self.peers:
+            return False
+        all_profiles = [p.is_from_profile for p in self.peers]
+        return bool(all_profiles) and all(all_profiles)
+
     def _generate_input_data(self, allocate_from_pool: bool = False) -> list[dict]:
         return [peer._generate_input_data(allocate_from_pool=allocate_from_pool) for peer in self.peers]
 

{infrahub_server-1.6.3.dist-info → infrahub_server-1.7.0b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: infrahub-server
-Version: 1.6.3
+Version: 1.7.0b0
 Summary: Infrahub is taking a new approach to Infrastructure Management by providing a new generation of datastore to organize and control all the data that defines how an infrastructure should run.
 Project-URL: Homepage, https://opsmill.com
 Project-URL: Repository, https://github.com/opsmill/infrahub
@@ -11,9 +11,10 @@ License-File: LICENSE.txt
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: <3.13,>=3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: <3.14,>=3.12
 Requires-Dist: aio-pika<9.5,>=9.4
-Requires-Dist: aiodataloader==0.4.0
+Requires-Dist: aiodataloader==0.4.3
 Requires-Dist: ariadne-codegen==0.15.3
 Requires-Dist: asgi-correlation-id==4.2.0
 Requires-Dist: authlib==1.6.5
@@ -24,7 +25,7 @@ Requires-Dist: click==8.1.7
 Requires-Dist: deepdiff==8.6.1
 Requires-Dist: dulwich==0.22.7
 Requires-Dist: email-validator<2.2,>=2.1
-Requires-Dist: fast-depends==2.4.12
+Requires-Dist: fast-depends==3.0.5
 Requires-Dist: fastapi-storages<0.4,>=0.3
 Requires-Dist: fastapi==0.121.1
 Requires-Dist: gitpython<4,>=3
@@ -33,22 +34,22 @@ Requires-Dist: gunicorn<24,>=23.0.0
 Requires-Dist: jinja2<4,>=3
 Requires-Dist: lunr<0.8,>=0.7.0.post1
 Requires-Dist: nats-py==2.7.2
-Requires-Dist: neo4j-rust-ext<5.29,>=5.28
-Requires-Dist: neo4j<5.29,>=5.28
+Requires-Dist: neo4j-rust-ext==6.0.3.0
+Requires-Dist: neo4j==6.0.3
 Requires-Dist: netaddr==1.3.0
 Requires-Dist: netutils==1.12.0
-Requires-Dist: numpy==1.26.2
-Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.28.1
-Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.28.1
-Requires-Dist: opentelemetry-instrumentation-aio-pika==0.49b1
-Requires-Dist: opentelemetry-instrumentation-fastapi==0.49b1
-Requires-Dist: prefect-redis==0.2.6
-Requires-Dist: prefect==3.5.0
-Requires-Dist: pyarrow<15,>=14
+Requires-Dist: numpy>=1.26.2
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.39.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.39.0
+Requires-Dist: opentelemetry-instrumentation-aio-pika==0.60b0
+Requires-Dist: opentelemetry-instrumentation-fastapi==0.60b0
+Requires-Dist: prefect-redis==0.2.8
+Requires-Dist: prefect==3.6.7
+Requires-Dist: pyarrow>=14
 Requires-Dist: pydantic-settings<2.9,>=2.8
-Requires-Dist: pydantic<2.11,>=2.10
+Requires-Dist: pydantic<2.13,>=2.12
 Requires-Dist: pyjwt<2.9,>=2.8
-Requires-Dist: pytest<7.5,>=7.4
+Requires-Dist: pytest<9.1,>=9.0
 Requires-Dist: python-multipart==0.0.18
 Requires-Dist: pyyaml<7,>=6
 Requires-Dist: redis[hiredis]==6.0.0