This is an automated email from the ASF dual-hosted git repository.

dimberman pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/master by this push:
     new 7b7cc3c  backport for add_xcom_sidecar (#11478)
7b7cc3c is described below

commit 7b7cc3c391d756e0ef55df8f0c674da4de3fb4f0
Author: Daniel Imberman <[email protected]>
AuthorDate: Wed Oct 14 12:49:44 2020 -0700

    backport for add_xcom_sidecar (#11478)
    
    * backport for add_xcom_sidecar
    
    * activate function
    
    * add k8s filter
    
    * nit
    
    * fix to work
---
 provider_packages/refactor_provider_packages.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/provider_packages/refactor_provider_packages.py 
b/provider_packages/refactor_provider_packages.py
index 41b706c..9290e48 100755
--- a/provider_packages/refactor_provider_packages.py
+++ b/provider_packages/refactor_provider_packages.py
@@ -650,6 +650,18 @@ class RefactorBackportPackages:
             rename("airflow.providers.odbc.utils.helpers")
         )
 
+    def refactor_kubernetes_pod_operator(self):
+        def kubernetes_package_filter(node: LN, capture: Capture, filename: 
Filename) -> bool:
+            return filename.startswith("./airflow/providers/cncf/kubernetes")
+
+        (
+            self.qry.
+            select_class("KubernetesPodOperator").
+            select_method("add_xcom_sidecar").
+            filter(callback=kubernetes_package_filter).
+            rename("add_sidecar")
+        )
+
     def do_refactor(self, in_process: bool = False) -> None:  # noqa
         self.rename_deprecated_modules()
         self.refactor_amazon_package()
@@ -659,6 +671,7 @@ class RefactorBackportPackages:
         self.remove_super_init_call()
         self.add_provide_context_to_python_operators()
         self.remove_poke_mode_only_decorator()
+        self.refactor_kubernetes_pod_operator()
         # In order to debug Bowler - set in_process to True
         self.qry.execute(write=True, silent=False, interactive=False, 
in_process=in_process)
 

Reply via email to