Skip to content

prefect_kubernetes.flows

A module to define flows interacting with Kubernetes resources.

run_namespaced_job async

Flow for running a namespaced Kubernetes job.

Parameters:

Name Type Description Default
kubernetes_job KubernetesJob

The KubernetesJob block that specifies the job to run.

required
print_func Optional[Callable]

A function to print the logs from the job pods.

None

Returns:

Type Description
Dict[str, Any]

A dict of logs from each pod in the job, e.g. {'pod_name': 'pod_log_str'}.

Raises:

Type Description
RuntimeError

If the created Kubernetes job attains a failed status.

```python
from prefect_kubernetes import KubernetesJob, run_namespaced_job
from prefect_kubernetes.credentials import KubernetesCredentials

run_namespaced_job(
    kubernetes_job=KubernetesJob.from_yaml_file(
        credentials=KubernetesCredentials.load("k8s-creds"),
        manifest_path="path/to/job.yaml",
    )
)
```
Source code in prefect_kubernetes/flows.py
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
@flow
async def run_namespaced_job(
    kubernetes_job: KubernetesJob, print_func: Optional[Callable] = None
) -> Dict[str, Any]:
    """Flow for running a namespaced Kubernetes job.

    Args:
        kubernetes_job: The `KubernetesJob` block that specifies the job to run.
        print_func: A function to print the logs from the job pods.

    Returns:
        A dict of logs from each pod in the job, e.g. {'pod_name': 'pod_log_str'}.

    Raises:
        RuntimeError: If the created Kubernetes job attains a failed status.

    Example:

        ```python
        from prefect_kubernetes import KubernetesJob, run_namespaced_job
        from prefect_kubernetes.credentials import KubernetesCredentials

        run_namespaced_job(
            kubernetes_job=KubernetesJob.from_yaml_file(
                credentials=KubernetesCredentials.load("k8s-creds"),
                manifest_path="path/to/job.yaml",
            )
        )
        ```
    """
    kubernetes_job_run = task(kubernetes_job.trigger)()

    await task(kubernetes_job_run.wait_for_completion.aio)(
        kubernetes_job_run, print_func
    )

    return task(kubernetes_job_run.fetch_result)()