Refactor: Consolidate import textwrap in providers (#34220)

Author: eumiro

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

@@ -20,10 +20,10 @@
 import inspect
 import os
 import pickle
+import textwrap
 import uuid
 from shlex import quote
 from tempfile import TemporaryDirectory
-from textwrap import dedent
 from typing import TYPE_CHECKING, Callable, Sequence
 
 import dill
@@ -80,7 +80,7 @@ def __init__(self, namespace: str = "default", use_dill: bool = False, **kwargs)
     # TODO: Remove me once this provider min supported Airflow version is 2.6
     def get_python_source(self):
         raw_source = inspect.getsource(self.python_callable)
-        res = dedent(raw_source)
+        res = textwrap.dedent(raw_source)
         res = remove_task_decorator(res, self.custom_operator_name)
         return res
 

airflow/providers/docker/decorators/docker.py

@@ -20,8 +20,8 @@
 import inspect
 import os
 import pickle
+import textwrap
 from tempfile import TemporaryDirectory
-from textwrap import dedent
 from typing import TYPE_CHECKING, Callable, Sequence
 
 import dill
@@ -139,7 +139,7 @@ def execute(self, context: Context):
     # TODO: Remove me once this provider min supported Airflow version is 2.6
     def get_python_source(self):
         raw_source = inspect.getsource(self.python_callable)
-        res = dedent(raw_source)
+        res = textwrap.dedent(raw_source)
         res = remove_task_decorator(res, self.custom_operator_name)
         return res