# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
from __future__ import annotations
from enum import Enum
from typing import Union
import kubernetes.client as k8s
import kubernetes_asyncio.client as async_k8s
[docs]client_type = Union[k8s.CoreV1Api, async_k8s.CoreV1Api]
[docs]class ExecutionMode(str, Enum):
"""Enum class for execution mode."""
[docs]class KubernetesPodOperatorCallback:
"""
`KubernetesPodOperator` callbacks methods.
Currently, the callbacks methods are not called in the async mode, this support will be added
in the future.
"""
@staticmethod
[docs] def on_sync_client_creation(*, client: k8s.CoreV1Api, **kwargs) -> None:
"""
Invoke this callback after creating the sync client.
:param client: the created `kubernetes.client.CoreV1Api` client.
"""
pass
@staticmethod
[docs] def on_pod_creation(*, pod: k8s.V1Pod, client: client_type, mode: str, **kwargs) -> None:
"""
Invoke this callback after creating the pod.
:param pod: the created pod.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass
@staticmethod
[docs] def on_pod_starting(*, pod: k8s.V1Pod, client: client_type, mode: str, **kwargs) -> None:
"""
Invoke this callback when the pod starts.
:param pod: the started pod.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass
@staticmethod
[docs] def on_pod_completion(*, pod: k8s.V1Pod, client: client_type, mode: str, **kwargs) -> None:
"""
Invoke this callback when the pod completes.
:param pod: the completed pod.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass
@staticmethod
[docs] def on_pod_cleanup(*, pod: k8s.V1Pod, client: client_type, mode: str, **kwargs):
"""
Invoke this callback after cleaning/deleting the pod.
:param pod: the completed pod.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass
@staticmethod
[docs] def on_operator_resuming(
*, pod: k8s.V1Pod, event: dict, client: client_type, mode: str, **kwargs
) -> None:
"""
Invoke this callback when resuming the `KubernetesPodOperator` from deferred state.
:param pod: the current state of the pod.
:param event: the returned event from the Trigger.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass
@staticmethod
[docs] def progress_callback(*, line: str, client: client_type, mode: str, **kwargs) -> None:
"""
Invoke this callback to process pod container logs.
:param line: the read line of log.
:param client: the Kubernetes client that can be used in the callback.
:param mode: the current execution mode, it's one of (`sync`, `async`).
"""
pass