apache-airflow-providers-cncf-kubernetes 10.0.0__py3-none-any.whl → 10.0.1__py3-none-any.whl

airflow/providers/cncf/kubernetes/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "10.0.0"
+__version__ = "10.0.1"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.8.0"

airflow/providers/cncf/kubernetes/backcompat/__init__.py

@@ -14,3 +14,20 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+
+from __future__ import annotations
+
+
+def get_logical_date_key() -> str:
+    """
+    Get the key for execution/logical date for the appropriate Airflow version.
+
+    This is done by detecting the CLI argument name. There are various ways to
+    do this, but the CLI key name has a very small import footprint (especially
+    compared to importing ORM models).
+    """
+    from airflow.cli import cli_config
+
+    if hasattr(cli_config, "ARG_LOGICAL_DATE"):
+        return "logical_date"
+    return "execution_date"

airflow/providers/cncf/kubernetes/cli/kubernetes_command.py

@@ -25,7 +25,9 @@ from datetime import datetime, timedelta
 from kubernetes import client
 from kubernetes.client.api_client import ApiClient
 from kubernetes.client.rest import ApiException
+from packaging.version import Version
 
+from airflow import __version__ as airflow_version
 from airflow.models import DagRun, TaskInstance
 from airflow.providers.cncf.kubernetes import pod_generator
 from airflow.providers.cncf.kubernetes.executors.kubernetes_executor import KubeConfig
@@ -36,15 +38,21 @@ from airflow.utils import cli as cli_utils, yaml
 from airflow.utils.cli import get_dag
 from airflow.utils.providers_configuration_loader import providers_configuration_loaded
 
+AIRFLOW_VERSION = Version(airflow_version)
+AIRFLOW_V_3_0_PLUS = Version(AIRFLOW_VERSION.base_version) >= Version("3.0.0")
+
 
 @cli_utils.action_cli
 @providers_configuration_loaded
 def generate_pod_yaml(args):
     """Generate yaml files for each task in the DAG. Used for testing output of KubernetesExecutor."""
-    execution_date = args.execution_date
+    logical_date = args.logical_date if AIRFLOW_V_3_0_PLUS else args.execution_date
     dag = get_dag(subdir=args.subdir, dag_id=args.dag_id)
     yaml_output_path = args.output_path
-    dr = DagRun(dag.dag_id, execution_date=execution_date)
+    if AIRFLOW_V_3_0_PLUS:
+        dr = DagRun(dag.dag_id, logical_date=logical_date)
+    else:
+        dr = DagRun(dag.dag_id, execution_date=logical_date)
     kube_config = KubeConfig()
     for task in dag.tasks:
         ti = TaskInstance(task, None)
@@ -55,7 +63,7 @@ def generate_pod_yaml(args):
             pod_id=create_unique_id(args.dag_id, ti.task_id),
             try_number=ti.try_number,
             kube_image=kube_config.kube_image,
-            date=ti.execution_date,
+            date=ti.logical_date if AIRFLOW_V_3_0_PLUS else ti.execution_date,
             args=ti.command_as_list(),
             pod_override_object=PodGenerator.from_obj(ti.executor_config),
             scheduler_job_id="worker-config",
@@ -64,7 +72,7 @@ def generate_pod_yaml(args):
             with_mutation_hook=True,
         )
         api_client = ApiClient()
-        date_string = pod_generator.datetime_to_label_safe_datestring(execution_date)
+        date_string = pod_generator.datetime_to_label_safe_datestring(logical_date)
         yaml_file_name = f"{args.dag_id}_{ti.task_id}_{date_string}.yml"
         os.makedirs(os.path.dirname(yaml_output_path + "/airflow_yaml_output/"), exist_ok=True)
         with open(yaml_output_path + "/airflow_yaml_output/" + yaml_file_name, "w") as output:

airflow/providers/cncf/kubernetes/decorators/kubernetes.py

@@ -20,9 +20,10 @@ import base64
 import os
 import pickle
 import uuid
+from collections.abc import Sequence
 from shlex import quote
 from tempfile import TemporaryDirectory
-from typing import TYPE_CHECKING, Callable, Sequence
+from typing import TYPE_CHECKING, Callable
 
 import dill
 from kubernetes.client import models as k8s

airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py

@@ -30,17 +30,24 @@ import logging
 import multiprocessing
 import time
 from collections import Counter, defaultdict
+from collections.abc import Sequence
 from contextlib import suppress
 from datetime import datetime
 from queue import Empty, Queue
-from typing import TYPE_CHECKING, Any, Sequence
+from typing import TYPE_CHECKING, Any
 
+from deprecated import deprecated
 from kubernetes.dynamic import DynamicClient
 from sqlalchemy import or_, select, update
 
+try:
+    from airflow.cli.cli_config import ARG_LOGICAL_DATE
+except ImportError:  # 2.x compatibility.
+    from airflow.cli.cli_config import (  # type: ignore[attr-defined, no-redef]
+        ARG_EXECUTION_DATE as ARG_LOGICAL_DATE,
+    )
 from airflow.cli.cli_config import (
     ARG_DAG_ID,
-    ARG_EXECUTION_DATE,
     ARG_OUTPUT_PATH,
     ARG_SUBDIR,
     ARG_VERBOSE,
@@ -51,6 +58,7 @@ from airflow.cli.cli_config import (
     positive_int,
 )
 from airflow.configuration import conf
+from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.executors.base_executor import BaseExecutor
 from airflow.executors.executor_constants import KUBERNETES_EXECUTOR
 from airflow.providers.cncf.kubernetes.executors.kubernetes_executor_types import (
@@ -118,7 +126,7 @@ KUBERNETES_COMMANDS = (
         help="Generate YAML files for all tasks in DAG. Useful for debugging tasks without "
         "launching into a cluster",
         func=lazy_load_command("airflow.providers.cncf.kubernetes.cli.kubernetes_command.generate_pod_yaml"),
-        args=(ARG_DAG_ID, ARG_EXECUTION_DATE, ARG_SUBDIR, ARG_OUTPUT_PATH, ARG_VERBOSE),
+        args=(ARG_DAG_ID, ARG_LOGICAL_DATE, ARG_SUBDIR, ARG_OUTPUT_PATH, ARG_VERBOSE),
     ),
 )
 
@@ -605,6 +613,10 @@ class KubernetesExecutor(BaseExecutor):
             tis_to_flush.extend(_iter_tis_to_flush())
             return tis_to_flush
 
+    @deprecated(
+        reason="Replaced by function `revoke_task`. Upgrade airflow core to make this go away.",
+        category=AirflowProviderDeprecationWarning,
+    )
     def cleanup_stuck_queued_tasks(self, tis: list[TaskInstance]) -> list[str]:
         """
         Handle remnants of tasks that were failed because they were stuck in queued.
@@ -616,28 +628,39 @@ class KubernetesExecutor(BaseExecutor):
         :param tis: List of Task Instances to clean up
         :return: List of readable task instances for a warning message
         """
+        reprs = []
+        for ti in tis:
+            reprs.append(repr(ti))
+            self.revoke_task(ti=ti)
+            self.fail(ti.key)
+        return reprs
+
+    def revoke_task(self, *, ti: TaskInstance):
+        """
+        Revoke task that may be running.
+
+        :param ti: task instance to revoke
+        """
         if TYPE_CHECKING:
             assert self.kube_client
             assert self.kube_scheduler
-        readable_tis: list[str] = []
-        if not tis:
-            return readable_tis
+        self.running.discard(ti.key)
+        self.queued_tasks.pop(ti.key, None)
         pod_combined_search_str_to_pod_map = self.get_pod_combined_search_str_to_pod_map()
-        for ti in tis:
-            # Build the pod selector
-            base_label_selector = f"dag_id={ti.dag_id},task_id={ti.task_id}"
-            if ti.map_index >= 0:
-                # Old tasks _couldn't_ be mapped, so we don't have to worry about compat
-                base_label_selector += f",map_index={ti.map_index}"
-
-            search_str = f"{base_label_selector},run_id={ti.run_id}"
-            pod = pod_combined_search_str_to_pod_map.get(search_str, None)
-            if not pod:
-                self.log.warning("Cannot find pod for ti %s", ti)
-                continue
-            readable_tis.append(repr(ti))
-            self.kube_scheduler.delete_pod(pod_name=pod.metadata.name, namespace=pod.metadata.namespace)
-        return readable_tis
+        # Build the pod selector
+        base_label_selector = f"dag_id={ti.dag_id},task_id={ti.task_id}"
+        if ti.map_index >= 0:
+            # Old tasks _couldn't_ be mapped, so we don't have to worry about compat
+            base_label_selector += f",map_index={ti.map_index}"
+
+        search_str = f"{base_label_selector},run_id={ti.run_id}"
+        pod = pod_combined_search_str_to_pod_map.get(search_str, None)
+        if not pod:
+            self.log.warning("Cannot find pod for ti %s", ti)
+            return
+
+        self.kube_scheduler.patch_pod_revoked(pod_name=pod.metadata.name, namespace=pod.metadata.namespace)
+        self.kube_scheduler.delete_pod(pod_name=pod.metadata.name, namespace=pod.metadata.namespace)
 
     def adopt_launched_task(
         self,

airflow/providers/cncf/kubernetes/executors/kubernetes_executor_types.py

@@ -16,7 +16,7 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Any, Optional, Union
 
 ADOPTED = "adopted"
 if TYPE_CHECKING:
@@ -25,13 +25,23 @@ if TYPE_CHECKING:
     from airflow.utils.state import TaskInstanceState
 
     # TaskInstance key, command, configuration, pod_template_file
-    KubernetesJobType = Tuple[TaskInstanceKey, CommandType, Any, Optional[str]]
+    KubernetesJobType = tuple[TaskInstanceKey, CommandType, Any, Optional[str]]
 
     # key, pod state, pod_name, namespace, resource_version
-    KubernetesResultsType = Tuple[TaskInstanceKey, Optional[Union[TaskInstanceState, str]], str, str, str]
+    KubernetesResultsType = tuple[TaskInstanceKey, Optional[Union[TaskInstanceState, str]], str, str, str]
 
     # pod_name, namespace, pod state, annotations, resource_version
-    KubernetesWatchType = Tuple[str, str, Optional[Union[TaskInstanceState, str]], Dict[str, str], str]
+    KubernetesWatchType = tuple[str, str, Optional[Union[TaskInstanceState, str]], dict[str, str], str]
 
 ALL_NAMESPACES = "ALL_NAMESPACES"
 POD_EXECUTOR_DONE_KEY = "airflow_executor_done"
+
+POD_REVOKED_KEY = "airflow_pod_revoked"
+"""Label to indicate pod revoked by executor.
+
+When executor the executor revokes a task, the pod deletion is the result of
+the revocation.  So we don't want it to process that as an external deletion.
+So we want events on a revoked pod to be ignored.
+
+:meta private:
+"""

airflow/providers/cncf/kubernetes/executors/kubernetes_executor_utils.py

@@ -28,10 +28,12 @@ from kubernetes.client.rest import ApiException
 from urllib3.exceptions import ReadTimeoutError
 
 from airflow.exceptions import AirflowException
+from airflow.providers.cncf.kubernetes.backcompat import get_logical_date_key
 from airflow.providers.cncf.kubernetes.executors.kubernetes_executor_types import (
     ADOPTED,
     ALL_NAMESPACES,
     POD_EXECUTOR_DONE_KEY,
+    POD_REVOKED_KEY,
 )
 from airflow.providers.cncf.kubernetes.kube_client import get_kube_client
 from airflow.providers.cncf.kubernetes.kubernetes_helper_functions import (
@@ -150,6 +152,7 @@ class KubernetesJobWatcher(multiprocessing.Process, LoggingMixin):
         kwargs["_request_timeout"] = client_timeout
         kwargs["timeout_seconds"] = server_conn_timeout
 
+        logical_date_key = get_logical_date_key()
         for event in self._pod_events(kube_client=kube_client, query_kwargs=kwargs):
             task = event["object"]
             self.log.debug("Event: %s had an event of type %s", task.metadata.name, event["type"])
@@ -159,7 +162,7 @@ class KubernetesJobWatcher(multiprocessing.Process, LoggingMixin):
             task_instance_related_annotations = {
                 "dag_id": annotations["dag_id"],
                 "task_id": annotations["task_id"],
-                "execution_date": annotations.get("execution_date"),
+                logical_date_key: annotations.get(logical_date_key),
                 "run_id": annotations.get("run_id"),
                 "try_number": annotations["try_number"],
             }
@@ -203,9 +206,13 @@ class KubernetesJobWatcher(multiprocessing.Process, LoggingMixin):
         resource_version: str,
         event: Any,
     ) -> None:
+        """Process status response."""
         pod = event["object"]
+
+        if POD_REVOKED_KEY in pod.metadata.labels.keys():
+            return
+
         annotations_string = annotations_for_logging_task_metadata(annotations)
-        """Process status response."""
         if event["type"] == "DELETED" and not pod.metadata.deletion_timestamp:
             # This will happen only when the task pods are adopted by another executor.
             # So, there is no change in the pod state.
@@ -431,7 +438,7 @@ class AirflowKubernetesScheduler(LoggingMixin):
     def delete_pod(self, pod_name: str, namespace: str) -> None:
         """Delete Pod from a namespace; does not raise if it does not exist."""
         try:
-            self.log.debug("Deleting pod %s in namespace %s", pod_name, namespace)
+            self.log.info("Deleting pod %s in namespace %s", pod_name, namespace)
             self.kube_client.delete_namespaced_pod(
                 pod_name,
                 namespace,
@@ -443,6 +450,26 @@ class AirflowKubernetesScheduler(LoggingMixin):
             if str(e.status) != "404":
                 raise
 
+    def patch_pod_revoked(self, *, pod_name: str, namespace: str):
+        """
+        Patch the pod with a label that ensures it's ignored by the kubernetes watcher.
+
+        :meta private:
+        """
+        self.log.info(
+            "Patching pod %s in namespace %s to note that we are revoking the task.",
+            pod_name,
+            namespace,
+        )
+        try:
+            self.kube_client.patch_namespaced_pod(
+                name=pod_name,
+                namespace=namespace,
+                body={"metadata": {"labels": {POD_REVOKED_KEY: "True"}}},
+            )
+        except ApiException:
+            self.log.warning("Failed to patch pod %s with pod revoked key.", pod_name, exc_info=True)
+
     def patch_pod_executor_done(self, *, pod_name: str, namespace: str):
         """Add a "done" annotation to ensure we don't continually adopt pods."""
         self.log.debug("Patching pod %s in namespace %s to mark it as done", pod_name, namespace)

airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py

@@ -17,9 +17,13 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Sequence
+from collections.abc import Sequence
+from typing import TYPE_CHECKING
+
+from deprecated import deprecated
 
 from airflow.configuration import conf
+from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.executors.base_executor import BaseExecutor
 from airflow.providers.cncf.kubernetes.executors.kubernetes_executor import KubernetesExecutor
 
@@ -230,12 +234,21 @@ class LocalKubernetesExecutor(BaseExecutor):
             *self.kubernetes_executor.try_adopt_task_instances(kubernetes_tis),
         ]
 
+    @deprecated(
+        reason="Replaced by function `revoke_task`. Upgrade airflow core to make this go away.",
+        category=AirflowProviderDeprecationWarning,
+        action="ignore",  # ignoring since will get warning from the nested executors
+    )
     def cleanup_stuck_queued_tasks(self, tis: list[TaskInstance]) -> list[str]:
         # LocalExecutor doesn't have a cleanup_stuck_queued_tasks method, so we
         # will only run KubernetesExecutor's
         kubernetes_tis = [ti for ti in tis if ti.queue == self.KUBERNETES_QUEUE]
         return self.kubernetes_executor.cleanup_stuck_queued_tasks(kubernetes_tis)
 
+    def revoke_task(self, *, ti: TaskInstance):
+        if ti.queue == self.KUBERNETES_QUEUE:
+            self.kubernetes_executor.revoke_task(ti=ti)
+
     def end(self) -> None:
         """End local and kubernetes executor."""
         self.local_executor.end()

airflow/providers/cncf/kubernetes/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "Kubernetes",
         "description": "`Kubernetes <https://kubernetes.io/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1730012271,
+        "source-date-epoch": 1732429220,
         "versions": [
+            "10.0.1",
             "10.0.0",
             "9.0.1",
             "9.0.0",

airflow/providers/cncf/kubernetes/hooks/kubernetes.py

@@ -20,9 +20,10 @@ import asyncio
 import contextlib
 import json
 import tempfile
+from collections.abc import Generator
 from functools import cached_property
 from time import sleep
-from typing import TYPE_CHECKING, Any, Generator
+from typing import TYPE_CHECKING, Any
 
 import aiofiles
 import tenacity

airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py

@@ -27,6 +27,7 @@ from kubernetes.client.rest import ApiException
 from slugify import slugify
 
 from airflow.configuration import conf
+from airflow.providers.cncf.kubernetes.backcompat import get_logical_date_key
 
 if TYPE_CHECKING:
     from airflow.models.taskinstancekey import TaskInstanceKey
@@ -106,8 +107,10 @@ def annotations_to_key(annotations: dict[str, str]) -> TaskInstanceKey:
     from airflow.models.taskinstance import TaskInstance, TaskInstanceKey
     from airflow.settings import Session
 
-    if not annotation_run_id and "execution_date" in annotations:
-        execution_date = pendulum.parse(annotations["execution_date"])
+    logical_date_key = get_logical_date_key()
+
+    if not annotation_run_id and logical_date_key in annotations:
+        logical_date = pendulum.parse(annotations[logical_date_key])
         # Do _not_ use create-session, we don't want to expunge
         session = Session()
 
@@ -117,7 +120,7 @@ def annotations_to_key(annotations: dict[str, str]) -> TaskInstanceKey:
             .filter(
                 TaskInstance.dag_id == dag_id,
                 TaskInstance.task_id == task_id,
-                DagRun.execution_date == execution_date,
+                getattr(DagRun, logical_date_key) == logical_date,
             )
             .scalar()
         )

airflow/providers/cncf/kubernetes/operators/job.py

@@ -22,8 +22,9 @@ import copy
 import json
 import logging
 import os
+from collections.abc import Sequence
 from functools import cached_property
-from typing import TYPE_CHECKING, Sequence
+from typing import TYPE_CHECKING
 
 from kubernetes.client import BatchV1Api, models as k8s
 from kubernetes.client.api_client import ApiClient
@@ -273,7 +274,6 @@ class KubernetesJobOperator(KubernetesPodOperator):
             kwargs = {
                 "name": job.metadata.name,
                 "namespace": job.metadata.namespace,
-                "job": self.hook.batch_v1_client.api_client.sanitize_for_serialization(self.job),
             }
             if self.termination_grace_period is not None:
                 kwargs.update(grace_period_seconds=self.termination_grace_period)

airflow/providers/cncf/kubernetes/operators/pod.py

@@ -26,11 +26,11 @@ import os
 import re
 import shlex
 import string
-from collections.abc import Container, Mapping
+from collections.abc import Container, Iterable, Mapping, Sequence
 from contextlib import AbstractContextManager
 from enum import Enum
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, Callable, Iterable, Literal, Sequence
+from typing import TYPE_CHECKING, Any, Callable, Literal
 
 import kubernetes
 import tenacity

airflow/providers/cncf/kubernetes/operators/resource.py

@@ -19,8 +19,9 @@
 from __future__ import annotations
 
 import os
+from collections.abc import Sequence
 from functools import cached_property
-from typing import TYPE_CHECKING, Sequence
+from typing import TYPE_CHECKING
 
 import tenacity
 import yaml

airflow/providers/cncf/kubernetes/operators/spark_kubernetes.py

@@ -198,8 +198,6 @@ class SparkKubernetesOperator(KubernetesPodOperator):
             "task_id": ti.task_id,
             "run_id": run_id,
             "spark_kubernetes_operator": "True",
-            # 'execution_date': context['ts'],
-            # 'try_number': context['ti'].try_number,
         }
 
         # If running on Airflow 2.3+:

airflow/providers/cncf/kubernetes/pod_generator.py

@@ -41,6 +41,7 @@ from airflow.exceptions import (
     AirflowConfigException,
     AirflowException,
 )
+from airflow.providers.cncf.kubernetes.backcompat import get_logical_date_key
 from airflow.providers.cncf.kubernetes.kubernetes_helper_functions import (
     POD_NAME_MAX_LENGTH,
     add_unique_suffix,
@@ -329,7 +330,7 @@ class PodGenerator:
         if map_index >= 0:
             annotations["map_index"] = str(map_index)
         if date:
-            annotations["execution_date"] = date.isoformat()
+            annotations[get_logical_date_key()] = date.isoformat()
         if run_id:
             annotations["run_id"] = run_id
 
@@ -344,7 +345,7 @@ class PodGenerator:
                     try_number=try_number,
                     airflow_worker=scheduler_job_id,
                     map_index=map_index,
-                    execution_date=date,
+                    logical_date=date,
                     run_id=run_id,
                 ),
             ),
@@ -387,7 +388,7 @@ class PodGenerator:
         task_id,
         try_number,
         map_index=None,
-        execution_date=None,
+        logical_date=None,
         run_id=None,
         airflow_worker=None,
         include_version=False,
@@ -402,7 +403,7 @@ class PodGenerator:
             task_id=task_id,
             try_number=try_number,
             map_index=map_index,
-            execution_date=execution_date,
+            logical_date=logical_date,
             run_id=run_id,
             airflow_worker=airflow_worker,
             include_version=include_version,
@@ -422,7 +423,7 @@ class PodGenerator:
         try_number,
         airflow_worker=None,
         map_index=None,
-        execution_date=None,
+        logical_date=None,
         run_id=None,
         include_version=True,
     ):
@@ -443,8 +444,8 @@ class PodGenerator:
             labels["airflow-worker"] = make_safe_label_value(str(airflow_worker))
         if map_index is not None and map_index >= 0:
             labels["map_index"] = str(map_index)
-        if execution_date:
-            labels["execution_date"] = datetime_to_label_safe_datestring(execution_date)
+        if logical_date:
+            labels[get_logical_date_key()] = datetime_to_label_safe_datestring(logical_date)
         if run_id:
             labels["run_id"] = make_safe_label_value(run_id)
         return labels

airflow/providers/cncf/kubernetes/sensors/spark_kubernetes.py

@@ -17,8 +17,9 @@
 # under the License.
 from __future__ import annotations
 
+from collections.abc import Sequence
 from functools import cached_property
-from typing import TYPE_CHECKING, Sequence
+from typing import TYPE_CHECKING
 
 from kubernetes import client
 

airflow/providers/cncf/kubernetes/triggers/job.py

@@ -17,8 +17,9 @@
 from __future__ import annotations
 
 import asyncio
+from collections.abc import AsyncIterator
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, AsyncIterator
+from typing import TYPE_CHECKING, Any
 
 from airflow.providers.cncf.kubernetes.hooks.kubernetes import AsyncKubernetesHook, KubernetesHook
 from airflow.providers.cncf.kubernetes.utils.pod_manager import PodManager

airflow/providers/cncf/kubernetes/triggers/pod.py

@@ -19,9 +19,10 @@ from __future__ import annotations
 import asyncio
 import datetime
 import traceback
+from collections.abc import AsyncIterator
 from enum import Enum
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, AsyncIterator
+from typing import TYPE_CHECKING, Any
 
 from airflow.providers.cncf.kubernetes.hooks.kubernetes import AsyncKubernetesHook
 from airflow.providers.cncf.kubernetes.utils.pod_manager import (

airflow/providers/cncf/kubernetes/utils/k8s_resource_iterator.py

@@ -16,7 +16,8 @@
 # under the License.
 from __future__ import annotations
 
-from typing import Callable, Iterator
+from collections.abc import Iterator
+from typing import Callable
 
 from kubernetes.utils import FailToCreateError
 

airflow/providers/cncf/kubernetes/utils/pod_manager.py

@@ -22,11 +22,11 @@ import enum
 import json
 import math
 import time
-from collections.abc import Iterable
+from collections.abc import Generator, Iterable
 from contextlib import closing, suppress
 from dataclasses import dataclass
 from datetime import timedelta
-from typing import TYPE_CHECKING, Generator, Protocol, cast
+from typing import TYPE_CHECKING, Protocol, cast
 
 import pendulum
 import tenacity

{apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info → apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: apache-airflow-providers-cncf-kubernetes
-Version: 10.0.0
+Version: 10.0.1
 Summary: Provider package apache-airflow-providers-cncf-kubernetes for Apache Airflow
 Keywords: airflow-provider,cncf.kubernetes,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -28,8 +28,8 @@ Requires-Dist: google-re2>=1.0
 Requires-Dist: kubernetes>=29.0.0,<=31.0.0
 Requires-Dist: kubernetes_asyncio>=29.0.0,<=31.0.0
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.0/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.1/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.1
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://twitter.com/ApacheAirflow
@@ -79,7 +79,7 @@ Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 
 Package ``apache-airflow-providers-cncf-kubernetes``
 
-Release: ``10.0.0``
+Release: ``10.0.1``
 
 
 `Kubernetes <https://kubernetes.io/>`__
@@ -92,7 +92,7 @@ This is a provider package for ``cncf.kubernetes`` provider. All classes for thi
 are in ``airflow.providers.cncf.kubernetes`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.1/>`_.
 
 Installation
 ------------
@@ -119,4 +119,4 @@ PIP package             Version required
 ======================  =====================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-cncf-kubernetes/10.0.1/changelog.html>`_.

{apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info → apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info}/RECORD

@@ -1,38 +1,38 @@
 airflow/providers/cncf/kubernetes/LICENSE,sha256=FFb4jd2AXnOOf7XLP04pQW6jbdhG49TxlGY6fFpCV1Y,13609
-airflow/providers/cncf/kubernetes/__init__.py,sha256=nxLYWteyTzCKa1NCHauBZsE1vOl0TSVqpqd4IR6taqg,1503
+airflow/providers/cncf/kubernetes/__init__.py,sha256=WSfPJeVCftXHYzTgnGbSLoI5aWLsAVG5tXsBoOY9wSg,1503
 airflow/providers/cncf/kubernetes/callbacks.py,sha256=SK_gKvGWuU-nxHfsqsYMlNQ8HZbHfpvyItOqieel2lc,4162
-airflow/providers/cncf/kubernetes/get_provider_info.py,sha256=UdYm7n5XGH9N6hcCKJr5hMC5ekk-2lNP5wgj0plC5JY,18008
+airflow/providers/cncf/kubernetes/get_provider_info.py,sha256=EtOtYPHX8N8LfiWkdJpTO2P4QdMSZDO2luESnyiWMUk,18030
 airflow/providers/cncf/kubernetes/k8s_model.py,sha256=xmdFhX29DjegoZ-cq8-KDL9soVYXf4OpU6fAGr3cPTU,2101
 airflow/providers/cncf/kubernetes/kube_client.py,sha256=yflZxLousXA9d7t67KrEy55qzb1cUhEyy6yCPkEem28,5329
 airflow/providers/cncf/kubernetes/kube_config.py,sha256=FAmhZZ_Z2JtoVzL6wENSjcwrlwAenHttTX_Ild9aEms,5225
-airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py,sha256=tb6HykGOUKZIXXZssJyQLKBlWv_KoE5mjjxwOkeJHW0,5391
-airflow/providers/cncf/kubernetes/pod_generator.py,sha256=9Yd4FY0MbXc61JuOZwQu5WlFV6MlLBy72YMHF3lD8ao,19765
+airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py,sha256=Zf7RyNt9BkUHY7Sjm_6CimTmb2H1bv5lrO3T0pni_r0,5524
+airflow/providers/cncf/kubernetes/pod_generator.py,sha256=KGFqR8IIo41-Z5sFtxBFOrc2cvL7s-ER8OVBhp2dFs0,19841
 airflow/providers/cncf/kubernetes/pod_generator_deprecated.py,sha256=aukRdia0Wtv0FVZXgvaIWQxdWfnT9cxzH0UwOVph8dE,11998
 airflow/providers/cncf/kubernetes/python_kubernetes_script.jinja2,sha256=I0EHRGwLHjSiX85e51HBIoddRDnC8TJPFrDBqQq_NJg,1776
 airflow/providers/cncf/kubernetes/python_kubernetes_script.py,sha256=KnTlZSWCZhwvj89fSc2kgIRTaI4iLNKPquHc2wXnluo,3460
 airflow/providers/cncf/kubernetes/secret.py,sha256=wj-T9gouqau_X14slAstGmnSxqXJQzdLwUdURzHna0I,5209
 airflow/providers/cncf/kubernetes/template_rendering.py,sha256=AZesc6MDfpFHoN1PvQpVwM-PzwbKYC1IsdChYvTubZ8,2965
-airflow/providers/cncf/kubernetes/backcompat/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/cncf/kubernetes/backcompat/__init__.py,sha256=KXF76f3v1jIFUBNz8kwxVMvm7i4mNo35LbIG9IijBNc,1299
 airflow/providers/cncf/kubernetes/backcompat/backwards_compat_converters.py,sha256=2jIehZsixt4ZGwTkj7kC8Uq3w8XwFAEzZV4g8SyIUZI,4340
 airflow/providers/cncf/kubernetes/cli/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/cncf/kubernetes/cli/kubernetes_command.py,sha256=W49hsWZ3XWUJgFLte56tMNGqmOvcFitl7iFjmFR8Ezo,6990
+airflow/providers/cncf/kubernetes/cli/kubernetes_command.py,sha256=wkI2gW2VfS_Ru1zX-Tr5ymLRP20ghAjdkD8RBtbUvt0,7384
 airflow/providers/cncf/kubernetes/decorators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/cncf/kubernetes/decorators/kubernetes.py,sha256=zVDHw_zwgRbobTOouJtCEqDrjrVwINGcB1dA-wG-ME0,5787
+airflow/providers/cncf/kubernetes/decorators/kubernetes.py,sha256=-YgRnFGyS-tdoAqwKP_Jj5mh_YJlKpk5TvwYnphvp4o,5814
 airflow/providers/cncf/kubernetes/executors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py,sha256=mVNIA5SSJj3hRWJy05Rp6DVhzNgrxW-pyENeIAv9Jdg,34271
-airflow/providers/cncf/kubernetes/executors/kubernetes_executor_types.py,sha256=9rRhfRuujGtSE73Ax0kC12whZCgWF2m6j5w9G9e0F_I,1673
-airflow/providers/cncf/kubernetes/executors/kubernetes_executor_utils.py,sha256=f2FHTyQyfFR1jjQmZV9bfX7U4Vxk5h61KPJO0yMcGbM,23356
-airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py,sha256=6cEGS83tGVUm6tSsjVenaJXSstViBnwgrvu-XIW4XQg,10976
+airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py,sha256=2WPy9JCKVv9KinR0PZc1Sa8EXhjQMzArIxCms-H_EXE,35065
+airflow/providers/cncf/kubernetes/executors/kubernetes_executor_types.py,sha256=L8_8HOHd_4O8WW6xT2tp49-yOj0EMKCYK5YqMOOx_bI,1973
+airflow/providers/cncf/kubernetes/executors/kubernetes_executor_utils.py,sha256=lNdHp7Q5aTOxHBew-_H3cl9F4NLgmElNidhWXO3UKGw,24302
+airflow/providers/cncf/kubernetes/executors/local_kubernetes_executor.py,sha256=kx6pHAfSnKiDgT6iDOS4fJRq49DodFnyxUyd-iJsuI8,11512
 airflow/providers/cncf/kubernetes/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/cncf/kubernetes/hooks/kubernetes.py,sha256=WSWQZrAzyrjN9y5TCS796X1L7oB-Sq_2BUzhT9xAE2k,33632
+airflow/providers/cncf/kubernetes/hooks/kubernetes.py,sha256=SIdFhIA1Et4asGMNbX96ZaRLPlAxDhM3l5liD_eJuH4,33659
 airflow/providers/cncf/kubernetes/kubernetes_executor_templates/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/cncf/kubernetes/kubernetes_executor_templates/basic_template.yaml,sha256=yzJmXN4ZyB4aDwI_GIugpL9-f1YMVy__X-LQSbeU95A,2567
 airflow/providers/cncf/kubernetes/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py,sha256=ZEXw_PqGItO47AEgGKqAqwFHFo-gb9-7jgEMRJgOfNU,15311
-airflow/providers/cncf/kubernetes/operators/job.py,sha256=RyPuT0yP7-tRrrS7uqDUBGI_xA9_YEg-JLUDJ5QcTA8,23809
-airflow/providers/cncf/kubernetes/operators/pod.py,sha256=f8JoEsgZbjSNlClTfw8XeCm-_vSkwpUglXzOiDyhBUg,53495
-airflow/providers/cncf/kubernetes/operators/resource.py,sha256=ccbZQKB1B5N4Y-ruItacB5Q105Tc0uNSoEmUAaSCrGQ,7570
-airflow/providers/cncf/kubernetes/operators/spark_kubernetes.py,sha256=rsUfqJ_TqjqZSIeUvc6J7pTJiaiX1Ba2WI5Tn9qtfbw,13923
+airflow/providers/cncf/kubernetes/operators/job.py,sha256=mWstFYBzET1Pm7RYJyBOSZgjG_tb1s__DmWjb2An9jY,23738
+airflow/providers/cncf/kubernetes/operators/pod.py,sha256=X_6OLr0BkaffKSTUGhGpC7yvuxvFIpaXZLGW7A2-I7k,53495
+airflow/providers/cncf/kubernetes/operators/resource.py,sha256=4nS-eiVEGlotp-gCkHlwRuenj3pnKhZ4khh9s2cjZms,7597
+airflow/providers/cncf/kubernetes/operators/spark_kubernetes.py,sha256=QQbJ9MwIHkTCnHwu2BLkJdTeYjhmla-7zY0zGwXhTpk,13822
 airflow/providers/cncf/kubernetes/pod_template_file_examples/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/cncf/kubernetes/pod_template_file_examples/dags_in_image_template.yaml,sha256=7JdppZ-XDBpv2Bnde2SthhcME8w3b8xQdPAK1fJGW60,2256
 airflow/providers/cncf/kubernetes/pod_template_file_examples/dags_in_volume_template.yaml,sha256=-Pk_EwKpyWRYZKOnumUxVrDeAfFJ0nr3WZ7JNnvppzg,2442
@@ -42,16 +42,16 @@ airflow/providers/cncf/kubernetes/resource_convert/configmap.py,sha256=gf7DdVeD0
 airflow/providers/cncf/kubernetes/resource_convert/env_variable.py,sha256=CsVgLPXjI5pSDCBwcN3WCyyWPbiNRZc_OJCPK_C_298,1464
 airflow/providers/cncf/kubernetes/resource_convert/secret.py,sha256=ElZCMbTWeTKoPeIJ1fTvlqRXM8nGkWj2MrIlVckX6Ag,1494
 airflow/providers/cncf/kubernetes/sensors/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/cncf/kubernetes/sensors/spark_kubernetes.py,sha256=ogD8m8baqPN_hGEXjN1NGp1dvslG8ZPBjy_LPr6Zo7I,5353
+airflow/providers/cncf/kubernetes/sensors/spark_kubernetes.py,sha256=q5VZPf037pbsunFUjUrtz7M5oJW7waZ1q8snNAugCIk,5380
 airflow/providers/cncf/kubernetes/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/cncf/kubernetes/triggers/job.py,sha256=fe0kaQos_m_1rVE-1YJBZKKFVLxrlHOB8oI5ntVzSN8,6727
-airflow/providers/cncf/kubernetes/triggers/pod.py,sha256=Q-1GPILDgGJHoxtkaH78edhNeR6BekRG8f0qpFLGcSI,12626
+airflow/providers/cncf/kubernetes/triggers/job.py,sha256=DGbC1FZktBF-00Lb0pU9iIKQnmdW8HWklp5Wwq54OEY,6754
+airflow/providers/cncf/kubernetes/triggers/pod.py,sha256=7vLj9SvOOwh9p5XSZIr7mj47YWe_aPQgtN7uEZhma68,12653
 airflow/providers/cncf/kubernetes/utils/__init__.py,sha256=ClZN0VPjWySdVwS_ktH7rrgL9VLAcs3OSJSB9s3zaYw,863
 airflow/providers/cncf/kubernetes/utils/delete_from.py,sha256=poObZSoEJwQyaYWilEURs8f4CDY2sn_pfwS31Lf579A,5195
-airflow/providers/cncf/kubernetes/utils/k8s_resource_iterator.py,sha256=-Pgc5i2WEDl7ZBvtJZ4eWDqqlSj8WdULqwUyOWmsRp8,1928
-airflow/providers/cncf/kubernetes/utils/pod_manager.py,sha256=Xp_K8nay4WYKbG_cJ9mrhc_sFkdmu40Klq859sq9vNs,33286
+airflow/providers/cncf/kubernetes/utils/k8s_resource_iterator.py,sha256=DLypjkD_3YDixRTcsxEjgvHZNbbG9qamlz05eBqaWzU,1955
+airflow/providers/cncf/kubernetes/utils/pod_manager.py,sha256=LmXsFO7CYP8YTZla4poN30UtayLyGdVkqicSPZh6hIA,33286
 airflow/providers/cncf/kubernetes/utils/xcom_sidecar.py,sha256=k6bdmVJ21OrAwGmWwledRrAmaty9ZrmbuM-IbaI4mqo,2519
-apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info/entry_points.txt,sha256=ByD3QJJyP9CfmTYtpNI1953akD38RUDgpGXLaq9vpOw,111
-apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info/METADATA,sha256=GQI7La3mlDH3aGgNxvWhkZSNNkJs4jEBkGLozAKa1WM,5212
-apache_airflow_providers_cncf_kubernetes-10.0.0.dist-info/RECORD,,
+apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info/entry_points.txt,sha256=ByD3QJJyP9CfmTYtpNI1953akD38RUDgpGXLaq9vpOw,111
+apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info/METADATA,sha256=6S8NTSFwzZJyiT7KbJ7i19Km66J9uFo464oGGC7Id18,5212
+apache_airflow_providers_cncf_kubernetes-10.0.1.dist-info/RECORD,,