Update dataproc.py (#40643)

Adding DataprocBatchLink.persist to display link for createbatch operator
apache · Jul 8, 2024 · f12714d · f12714d
1 parent 03b1840
commit f12714d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py
@@ -3055,6 +3055,13 @@ def execute(self, context: Context):
                     self.log.info("Batch %s created", self.batch_id)
 
                 else:
+                    DataprocBatchLink.persist(
+                        context=context,
+                        operator=self,
+                        project_id=self.project_id,
+                        region=self.region,
+                        batch_id=self.batch_id,
+                    )
                     return self.operation.operation.name
 
             else: