infrahub-server 1.2.0rc0__py3-none-any.whl → 1.2.1__py3-none-any.whl

infrahub/message_bus/messages/check_repository_checkdefinition.py

@@ -1,20 +0,0 @@
-from pydantic import ConfigDict, Field
-
-from infrahub.message_bus import InfrahubMessage
-from infrahub.message_bus.types import ProposedChangeBranchDiff
-
-
-class CheckRepositoryCheckDefinition(InfrahubMessage):
-    """Triggers user defined checks to run based on a Check Definition."""
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    check_definition_id: str = Field(..., description="The unique ID of the check definition")
-    commit: str = Field(..., description="The commit to target")
-    repository_id: str = Field(..., description="The unique ID of the Repository")
-    repository_name: str = Field(..., description="The name of the Repository")
-    branch_name: str = Field(..., description="The branch where the check is run")
-    file_path: str = Field(..., description="The path and filename of the check")
-    class_name: str = Field(..., description="The name of the class containing the check")
-    proposed_change: str = Field(..., description="The unique ID of the Proposed Change")
-    branch_diff: ProposedChangeBranchDiff = Field(..., description="The calculated diff between the two branches")

infrahub/message_bus/messages/check_repository_mergeconflicts.py

@@ -1,16 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class CheckRepositoryMergeConflicts(InfrahubMessage):
-    """Runs a check to validate if there are merge conflicts for a proposed change between two branches."""
-
-    validator_id: str = Field(..., description="The id of the validator associated with this check")
-    validator_execution_id: str = Field(..., description="The id of current execution of the associated validator")
-    check_execution_id: str = Field(..., description="The unique ID for the current execution of this check")
-    proposed_change: str = Field(..., description="The unique ID of the Proposed Change")
-    repository_id: str = Field(..., description="The unique ID of the Repository")
-    repository_name: str = Field(..., description="The name of the Repository")
-    source_branch: str = Field(..., description="The source branch")
-    target_branch: str = Field(..., description="The target branch")

infrahub/message_bus/messages/check_repository_usercheck.py

@@ -1,26 +0,0 @@
-from pydantic import ConfigDict, Field
-
-from infrahub.message_bus import InfrahubMessage
-from infrahub.message_bus.types import ProposedChangeBranchDiff
-
-
-class CheckRepositoryUserCheck(InfrahubMessage):
-    """Runs a check as defined within a CoreCheckDefinition within a repository."""
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    validator_id: str = Field(..., description="The id of the validator associated with this check")
-    validator_execution_id: str = Field(..., description="The id of current execution of the associated validator")
-    check_execution_id: str = Field(..., description="The unique ID for the current execution of this check")
-    check_definition_id: str = Field(..., description="The unique ID of the check definition")
-    commit: str = Field(..., description="The commit to target")
-    repository_id: str = Field(..., description="The unique ID of the Repository")
-    repository_name: str = Field(..., description="The name of the Repository")
-    branch_name: str = Field(..., description="The branch where the check is run")
-    file_path: str = Field(..., description="The path and filename of the check")
-    class_name: str = Field(..., description="The name of the class containing the check")
-    proposed_change: str = Field(..., description="The unique ID of the Proposed Change")
-    variables: dict = Field(default_factory=dict, description="Input variables when running the check")
-    name: str = Field(..., description="The name of the check")
-    branch_diff: ProposedChangeBranchDiff = Field(..., description="The calculated diff between the two branches")
-    timeout: int = Field(..., description="The timeout for the check")

infrahub/message_bus/messages/event_branch_create.py

@@ -1,11 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventBranchCreate(InfrahubMessage):
-    """Sent a new branch is created."""
-
-    branch: str = Field(..., description="The branch that was created")
-    branch_id: str = Field(..., description="The unique ID of the branch")
-    sync_with_git: bool = Field(..., description="Indicates if Infrahub should extend this branch to git.")

infrahub/message_bus/messages/event_branch_delete.py

@@ -1,11 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventBranchDelete(InfrahubMessage):
-    """Sent when a branch has been deleted."""
-
-    branch: str = Field(..., description="The branch that was deleted")
-    branch_id: str = Field(..., description="The unique ID of the branch")
-    sync_with_git: bool = Field(..., description="Indicates if the branch was extended to Git")

infrahub/message_bus/messages/event_branch_rebased.py

@@ -1,9 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventBranchRebased(InfrahubMessage):
-    """Sent when a branch has been rebased."""
-
-    branch: str = Field(..., description="The branch that was rebased")

infrahub/message_bus/messages/event_node_mutated.py

@@ -1,15 +0,0 @@
-from typing import Any
-
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventNodeMutated(InfrahubMessage):
-    """Sent when a node has been mutated"""
-
-    branch: str = Field(..., description="The branch that was created")
-    kind: str = Field(..., description="The type of object modified")
-    node_id: str = Field(..., description="The ID of the mutated node")
-    action: str = Field(..., description="The action taken on the node")
-    data: dict[str, Any] = Field(..., description="Data on modified object")

infrahub/message_bus/messages/event_schema_update.py

@@ -1,9 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class EventSchemaUpdate(InfrahubMessage):
-    """Sent when the schema on a branch has been updated."""
-
-    branch: str = Field(..., description="The branch where the update occurred")

infrahub/message_bus/messages/request_repository_checks.py

@@ -1,12 +0,0 @@
-from pydantic import Field
-
-from infrahub.message_bus import InfrahubMessage
-
-
-class RequestRepositoryChecks(InfrahubMessage):
-    """Sent to trigger the checks for a repository to be executed."""
-
-    proposed_change: str = Field(..., description="The unique ID of the Proposed Change")
-    repository: str = Field(..., description="The unique ID of the Repository")
-    source_branch: str = Field(..., description="The source branch")
-    target_branch: str = Field(..., description="The target branch")

infrahub/message_bus/messages/request_repository_userchecks.py

@@ -1,18 +0,0 @@
-from pydantic import ConfigDict, Field
-
-from infrahub.message_bus import InfrahubMessage
-from infrahub.message_bus.types import ProposedChangeBranchDiff
-
-
-class RequestRepositoryUserChecks(InfrahubMessage):
-    """Sent to trigger the user defined checks on a repository."""
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    proposed_change: str = Field(..., description="The unique ID of the Proposed Change")
-    repository_id: str = Field(..., description="The unique ID of the Repository")
-    repository_name: str = Field(..., description="The name of the Repository")
-    source_branch: str = Field(..., description="The source branch")
-    source_branch_sync_with_git: bool = Field(..., description="Indicates if the source branch should sync with git")
-    target_branch: str = Field(..., description="The target branch")
-    branch_diff: ProposedChangeBranchDiff = Field(..., description="The calculated diff between the two branches")

infrahub/message_bus/operations/check/repository.py

@@ -1,293 +0,0 @@
-from infrahub_sdk.protocols import CoreCheckDefinition
-from infrahub_sdk.uuidt import UUIDT
-from prefect import flow
-from prefect.logging import get_run_logger
-
-from infrahub import lock
-from infrahub.core.constants import InfrahubKind
-from infrahub.core.manager import NodeManager
-from infrahub.core.timestamp import Timestamp
-from infrahub.exceptions import CheckError
-from infrahub.git.repository import InfrahubRepository
-from infrahub.message_bus import InfrahubMessage, messages
-from infrahub.message_bus.types import KVTTL
-from infrahub.services import InfrahubServices
-from infrahub.workflows.utils import add_tags
-
-
-@flow(
-    name="git-repository-check-definition",
-    flow_run_name="Run user defined checks for repository {message.repository_name}",
-)
-async def check_definition(message: messages.CheckRepositoryCheckDefinition, service: InfrahubServices) -> None:
-    await add_tags(branches=[message.branch_name], nodes=[message.proposed_change])
-    log = get_run_logger()
-
-    definition = await service.client.get(
-        kind=CoreCheckDefinition, id=message.check_definition_id, branch=message.branch_name
-    )
-    proposed_change = await service.client.get(kind=InfrahubKind.PROPOSEDCHANGE, id=message.proposed_change)
-    validator_execution_id = str(UUIDT())
-    check_execution_ids: list[str] = []
-    await proposed_change.validations.fetch()
-    validator = None
-    events: list[InfrahubMessage] = []
-
-    for relationship in proposed_change.validations.peers:
-        existing_validator = relationship.peer
-
-        if (
-            existing_validator.typename == InfrahubKind.USERVALIDATOR
-            and existing_validator.repository.id == message.repository_id
-            and existing_validator.check_definition.id == message.check_definition_id
-        ):
-            validator = existing_validator
-            service.log.info("Found the same validator", validator=validator)
-
-    if validator:
-        validator.conclusion.value = "unknown"
-        validator.state.value = "queued"
-        validator.started_at.value = ""
-        validator.completed_at.value = ""
-        await validator.save()
-    else:
-        validator = await service.client.create(
-            kind=InfrahubKind.USERVALIDATOR,
-            data={
-                "label": f"Check: {definition.name.value}",
-                "proposed_change": message.proposed_change,
-                "repository": message.repository_id,
-                "check_definition": message.check_definition_id,
-            },
-        )
-        await validator.save()
-
-    if definition.targets.id:
-        # Check against a group of targets
-        await definition.targets.fetch()
-        group = definition.targets.peer
-        await group.members.fetch()
-        for relationship in group.members.peers:
-            member = relationship.peer
-
-            check_execution_id = str(UUIDT())
-            check_execution_ids.append(check_execution_id)
-            events.append(
-                messages.CheckRepositoryUserCheck(
-                    name=member.display_label,
-                    validator_id=validator.id,
-                    validator_execution_id=validator_execution_id,
-                    check_execution_id=check_execution_id,
-                    repository_id=message.repository_id,
-                    repository_name=message.repository_name,
-                    commit=message.commit,
-                    file_path=message.file_path,
-                    class_name=message.class_name,
-                    branch_name=message.branch_name,
-                    check_definition_id=message.check_definition_id,
-                    proposed_change=message.proposed_change,
-                    variables=member.extract(params=definition.parameters.value),
-                    branch_diff=message.branch_diff,
-                    timeout=definition.timeout.value,
-                )
-            )
-
-    else:
-        check_execution_id = str(UUIDT())
-        check_execution_ids.append(check_execution_id)
-        events.append(
-            messages.CheckRepositoryUserCheck(
-                name=definition.name.value,
-                validator_id=validator.id,
-                validator_execution_id=validator_execution_id,
-                check_execution_id=check_execution_id,
-                repository_id=message.repository_id,
-                repository_name=message.repository_name,
-                commit=message.commit,
-                file_path=message.file_path,
-                class_name=message.class_name,
-                branch_name=message.branch_name,
-                check_definition_id=message.check_definition_id,
-                proposed_change=message.proposed_change,
-                branch_diff=message.branch_diff,
-                timeout=definition.timeout.value,
-            )
-        )
-
-    checks_in_execution = ",".join(check_execution_ids)
-    log.info(f"Checks in execution {checks_in_execution}")
-    await service.cache.set(
-        key=f"validator_execution_id:{validator_execution_id}:checks",
-        value=checks_in_execution,
-        expires=KVTTL.TWO_HOURS,
-    )
-    events.append(
-        messages.FinalizeValidatorExecution(
-            start_time=Timestamp().to_string(),
-            validator_id=validator.id,
-            validator_execution_id=validator_execution_id,
-            validator_type=InfrahubKind.USERVALIDATOR,
-        )
-    )
-
-    for event in events:
-        event.assign_meta(parent=message)
-        await service.message_bus.send(message=event)
-
-
-@flow(
-    name="git-repository-check-merge-conflict",
-    flow_run_name="Check for merge conflicts between {message.source_branch} and {message.target_branch}",
-)
-async def merge_conflicts(message: messages.CheckRepositoryMergeConflicts, service: InfrahubServices) -> None:
-    """Runs a check to see if there are merge conflicts between two branches."""
-    await add_tags(branches=[message.source_branch], nodes=[message.proposed_change])
-
-    success_condition = "-"
-    validator = await service.client.get(kind=InfrahubKind.REPOSITORYVALIDATOR, id=message.validator_id)
-    await validator.checks.fetch()
-
-    repo = await InfrahubRepository.init(id=message.repository_id, name=message.repository_name, service=service)
-    async with lock.registry.get(name=message.repository_name, namespace="repository"):
-        conflicts = await repo.get_conflicts(source_branch=message.source_branch, dest_branch=message.target_branch)
-
-    existing_checks = {}
-    for relationship in validator.checks.peers:
-        existing_check = relationship.peer
-        if existing_check.typename == InfrahubKind.FILECHECK and existing_check.kind.value == "MergeConflictCheck":
-            check_key = ""
-            if existing_check.files.value:
-                check_key = "".join(existing_check.files.value)
-            check_key = f"-{check_key}"
-            existing_checks[check_key] = existing_check
-
-    validator_conclusion = "success"
-    check = None
-    if conflicts:
-        validator_conclusion = "failure"
-        for conflict in conflicts:
-            conflict_key = f"-{conflict}"
-            if conflict_key in existing_checks:
-                existing_checks[conflict_key].created_at.value = Timestamp().to_string()
-                await existing_checks[conflict_key].save()
-                existing_checks.pop(conflict_key)
-            else:
-                check = await service.client.create(
-                    kind=InfrahubKind.FILECHECK,
-                    data={
-                        "name": conflict,
-                        "origin": "ConflictCheck",
-                        "kind": "MergeConflictCheck",
-                        "validator": message.validator_id,
-                        "created_at": Timestamp().to_string(),
-                        "files": [conflict],
-                        "conclusion": "failure",
-                        "severity": "critical",
-                    },
-                )
-                await check.save()
-
-    elif success_condition in existing_checks:
-        existing_checks[success_condition].created_at.value = Timestamp().to_string()
-        await existing_checks[success_condition].save()
-        existing_checks.pop(success_condition)
-
-    else:
-        check = await service.client.create(
-            kind=InfrahubKind.FILECHECK,
-            data={
-                "name": "Merge Conflict Check",
-                "origin": "ConflictCheck",
-                "kind": "MergeConflictCheck",
-                "validator": message.validator_id,
-                "created_at": Timestamp().to_string(),
-                "conclusion": "success",
-                "severity": "info",
-            },
-        )
-        await check.save()
-
-    async with service.database.start_session() as db:
-        await NodeManager.delete(db=db, nodes=list(existing_checks.values()))
-
-    await service.cache.set(
-        key=f"validator_execution_id:{message.validator_execution_id}:check_execution_id:{message.check_execution_id}",
-        value=validator_conclusion,
-        expires=KVTTL.TWO_HOURS,
-    )
-
-
-@flow(name="git-repository-user-check", flow_run_name="Execute user defined Check '{message.name}'")
-async def user_check(message: messages.CheckRepositoryUserCheck, service: InfrahubServices) -> None:
-    await add_tags(branches=[message.branch_name], nodes=[message.proposed_change])
-    log = get_run_logger()
-
-    validator = await service.client.get(kind=InfrahubKind.USERVALIDATOR, id=message.validator_id)
-    await validator.checks.fetch()
-
-    repo = await InfrahubRepository.init(
-        id=message.repository_id, name=message.repository_name, commit=message.commit, service=service
-    )
-    conclusion = "failure"
-    severity = "critical"
-    log_entries = ""
-    try:
-        check_run = await repo.execute_python_check.with_options(timeout_seconds=message.timeout)(
-            branch_name=message.branch_name,
-            location=message.file_path,
-            class_name=message.class_name,
-            client=service.client,
-            commit=message.commit,
-            params=message.variables,
-        )
-        if check_run.passed:
-            conclusion = "success"
-            severity = "info"
-            log.info("The check passed")
-        else:
-            log.warning("The check reported failures")
-            for log_entry in check_run.log_entries:
-                log.warning(log_entry)
-        log_entries = check_run.log_entries
-    except CheckError as exc:
-        log.warning("The check failed to run")
-        log.error(exc.message)
-        log_entries = f"FATAL Error/n:{exc.message}"
-
-    check = None
-    for relationship in validator.checks.peers:
-        existing_check = relationship.peer
-        if (
-            existing_check.typename == InfrahubKind.STANDARDCHECK
-            and existing_check.kind.value == "CheckDefinition"
-            and existing_check.name.value == message.name
-        ):
-            check = existing_check
-
-    if check:
-        check.created_at.value = Timestamp().to_string()
-        check.message.value = log_entries
-        check.conclusion.value = conclusion
-        check.severity.value = severity
-        await check.save()
-    else:
-        check = await service.client.create(
-            kind=InfrahubKind.STANDARDCHECK,
-            data={
-                "name": message.name,
-                "origin": message.repository_id,
-                "kind": "CheckDefinition",
-                "validator": message.validator_id,
-                "created_at": Timestamp().to_string(),
-                "message": log_entries,
-                "conclusion": conclusion,
-                "severity": severity,
-            },
-        )
-        await check.save()
-
-    await service.cache.set(
-        key=f"validator_execution_id:{message.validator_execution_id}:check_execution_id:{message.check_execution_id}",
-        value=conclusion,
-        expires=KVTTL.TWO_HOURS,
-    )

infrahub/message_bus/operations/event/node.py

@@ -1,20 +0,0 @@
-from infrahub.message_bus import messages
-from infrahub.services import InfrahubServices
-
-
-async def mutated(
-    message: messages.EventNodeMutated,
-    service: InfrahubServices,
-) -> None:
-    """Event posted when a node is mutated"""
-    # Note for now this is only kept to facilitate publishing to other queues in the future
-    # This operation doesn't have a flow defined to avoid having the workers need to register
-    # the results in Prefect when they don't actually do anything aside from add noise.
-    service.log.debug(
-        "Mutation on node",
-        branch=message.branch,
-        node_id=message.node_id,
-        action=message.action,
-        kind=message.kind,
-        data=message.data,
-    )

infrahub/message_bus/operations/event/schema.py

@@ -1,17 +0,0 @@
-from prefect import flow
-
-from infrahub.log import get_logger
-from infrahub.message_bus import messages
-from infrahub.services import InfrahubServices
-
-log = get_logger()
-
-
-@flow(name="event-schema-update")
-async def update(message: messages.EventSchemaUpdate, service: InfrahubServices) -> None:
-    log.info("run_message", branch=message.branch)
-
-    msg = messages.RefreshRegistryBranches()
-
-    msg.assign_meta(parent=message)
-    await service.message_bus.send(message=msg)

infrahub/message_bus/operations/requests/repository.py

@@ -1,133 +0,0 @@
-from infrahub_sdk.uuidt import UUIDT
-from prefect import flow
-from prefect.logging import get_run_logger
-
-from infrahub.core.constants import InfrahubKind
-from infrahub.core.timestamp import Timestamp
-from infrahub.message_bus import InfrahubMessage, messages
-from infrahub.message_bus.types import KVTTL
-from infrahub.services import InfrahubServices
-from infrahub.workflows.utils import add_tags
-
-
-@flow(name="repository-check", flow_run_name="Running repository checks for repository {message.repository}")
-async def checks(message: messages.RequestRepositoryChecks, service: InfrahubServices) -> None:
-    """Request to start validation checks on a specific repository."""
-    await add_tags(branches=[message.source_branch], nodes=[message.proposed_change])
-    log = get_run_logger()
-
-    events: list[InfrahubMessage] = []
-
-    repository = await service.client.get(
-        kind=InfrahubKind.GENERICREPOSITORY, id=message.repository, branch=message.source_branch
-    )
-    proposed_change = await service.client.get(kind=InfrahubKind.PROPOSEDCHANGE, id=message.proposed_change)
-
-    validator_execution_id = str(UUIDT())
-    check_execution_ids: list[str] = []
-    await proposed_change.validations.fetch()
-    await repository.checks.fetch()
-
-    validator_name = f"Repository Validator: {repository.name.value}"
-    validator = None
-    for relationship in proposed_change.validations.peers:
-        existing_validator = relationship.peer
-
-        if (
-            existing_validator.typename == InfrahubKind.REPOSITORYVALIDATOR
-            and existing_validator.repository.id == message.repository
-        ):
-            validator = existing_validator
-
-    if validator:
-        validator.conclusion.value = "unknown"
-        validator.state.value = "queued"
-        validator.started_at.value = ""
-        validator.completed_at.value = ""
-        await validator.save()
-    else:
-        validator = await service.client.create(
-            kind=InfrahubKind.REPOSITORYVALIDATOR,
-            data={
-                "label": validator_name,
-                "proposed_change": message.proposed_change,
-                "repository": message.repository,
-            },
-        )
-        await validator.save()
-
-    check_execution_id = str(UUIDT())
-    check_execution_ids.append(check_execution_id)
-    log.info("Adding check for merge conflict")
-
-    events.append(
-        messages.CheckRepositoryMergeConflicts(
-            validator_id=validator.id,
-            validator_execution_id=validator_execution_id,
-            check_execution_id=check_execution_id,
-            proposed_change=message.proposed_change,
-            repository_id=message.repository,
-            repository_name=repository.name.value,
-            source_branch=message.source_branch,
-            target_branch=message.target_branch,
-        )
-    )
-
-    checks_in_execution = ",".join(check_execution_ids)
-    log.info(f"Checks in execution {checks_in_execution}")
-    await service.cache.set(
-        key=f"validator_execution_id:{validator_execution_id}:checks",
-        value=checks_in_execution,
-        expires=KVTTL.TWO_HOURS,
-    )
-    events.append(
-        messages.FinalizeValidatorExecution(
-            start_time=Timestamp().to_string(),
-            validator_id=validator.id,
-            validator_execution_id=validator_execution_id,
-            validator_type=InfrahubKind.REPOSITORYVALIDATOR,
-        )
-    )
-
-    for event in events:
-        event.assign_meta(parent=message)
-        await service.message_bus.send(message=event)
-
-
-@flow(
-    name="repository-users-check",
-    flow_run_name="Evaluating user-defined checks on repository {message.repository_name}",
-)
-async def user_checks(message: messages.RequestRepositoryUserChecks, service: InfrahubServices) -> None:
-    """Request to start validation checks on a specific repository for User-defined checks."""
-
-    await add_tags(branches=[message.source_branch], nodes=[message.proposed_change])
-    log = get_run_logger()
-
-    events: list[InfrahubMessage] = []
-
-    repository = await service.client.get(
-        kind=InfrahubKind.GENERICREPOSITORY, id=message.repository_id, branch=message.source_branch, fragment=True
-    )
-    await repository.checks.fetch()
-
-    for relationship in repository.checks.peers:
-        log.info("Adding check for user defined check")
-        check_definition = relationship.peer
-        events.append(
-            messages.CheckRepositoryCheckDefinition(
-                check_definition_id=check_definition.id,
-                repository_id=repository.id,
-                repository_name=repository.name.value,
-                commit=repository.commit.value,
-                file_path=check_definition.file_path.value,
-                class_name=check_definition.class_name.value,
-                branch_name=message.source_branch,
-                proposed_change=message.proposed_change,
-                branch_diff=message.branch_diff,
-            )
-        )
-
-    for event in events:
-        event.assign_meta(parent=message)
-        await service.message_bus.send(message=event)

infrahub/webhook/constants.py

@@ -1 +0,0 @@
-AUTOMATION_NAME_RUN = "Trigger-webhook-execution"