diff --git a/airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py b/airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py index 62dc351bbd1a5..9129bdf77920f 100644 --- a/airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py +++ b/airflow/providers/cncf/kubernetes/kubernetes_helper_functions.py @@ -104,7 +104,7 @@ def create_unique_id( name += task_id base_name = slugify(name, lowercase=True)[:max_length].strip(".-") if unique: - return add_pod_suffix(pod_name=base_name, rand_len=8, max_len=max_length) + return add_unique_suffix(name=base_name, rand_len=8, max_len=max_length) else: return base_name diff --git a/airflow/providers/cncf/kubernetes/operators/pod.py b/airflow/providers/cncf/kubernetes/operators/pod.py index 7f712dba2fc1f..32b361f0ccd0a 100644 --- a/airflow/providers/cncf/kubernetes/operators/pod.py +++ b/airflow/providers/cncf/kubernetes/operators/pod.py @@ -62,8 +62,8 @@ from airflow.providers.cncf.kubernetes.hooks.kubernetes import KubernetesHook from airflow.providers.cncf.kubernetes.kubernetes_helper_functions import ( POD_NAME_MAX_LENGTH, - add_pod_suffix, - create_pod_id, + add_unique_suffix, + create_unique_id, ) from airflow.providers.cncf.kubernetes.pod_generator import PodGenerator from airflow.providers.cncf.kubernetes.triggers.pod import KubernetesPodTrigger @@ -1049,12 +1049,12 @@ def build_pod_request_obj(self, context: Context | None = None) -> k8s.V1Pod: pod = PodGenerator.reconcile_pods(pod_template, pod) if not pod.metadata.name: - pod.metadata.name = create_pod_id( + pod.metadata.name = create_unique_id( task_id=self.task_id, unique=self.random_name_suffix, max_length=POD_NAME_MAX_LENGTH ) elif self.random_name_suffix: # user has supplied pod name, we're just adding suffix - pod.metadata.name = add_pod_suffix(pod_name=pod.metadata.name) + pod.metadata.name = add_unique_suffix(name=pod.metadata.name) if not pod.metadata.namespace: hook_namespace = self.hook.get_namespace() diff --git a/airflow/providers/cncf/kubernetes/pod_generator.py b/airflow/providers/cncf/kubernetes/pod_generator.py index e9f413985fd0c..8d78145893893 100644 --- a/airflow/providers/cncf/kubernetes/pod_generator.py +++ b/airflow/providers/cncf/kubernetes/pod_generator.py @@ -46,6 +46,7 @@ from airflow.providers.cncf.kubernetes.kubernetes_helper_functions import ( POD_NAME_MAX_LENGTH, add_pod_suffix, + add_unique_suffix, rand_str, ) from airflow.providers.cncf.kubernetes.pod_generator_deprecated import ( @@ -397,7 +398,7 @@ def construct_pod( UserWarning, stacklevel=2, ) - pod_id = add_pod_suffix(pod_name=pod_id, max_len=POD_NAME_MAX_LENGTH) + pod_id = add_unique_suffix(name=pod_id, max_len=POD_NAME_MAX_LENGTH) try: image = pod_override_object.spec.containers[0].image # type: ignore if not image: