fix(microsoft/azure): add return statement to yield within a while lo…

…op in triggers (#38393)
apache · Mar 22, 2024 · 51c94e1 · 51c94e1
1 parent cddf1cc
commit 51c94e1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow/providers/microsoft/azure/triggers/wasb.py b/airflow/providers/microsoft/azure/triggers/wasb.py
@@ -78,6 +78,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]:
                     if blob_exists:
                         message = f"Blob {self.blob_name} found in container {self.container_name}."
                         yield TriggerEvent({"status": "success", "message": message})
+                        return
                     else:
                         message = (
                             f"Blob {self.blob_name} not available yet in container {self.container_name}."
@@ -154,6 +155,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]:
                     if prefix_exists:
                         message = f"Prefix {self.prefix} found in container {self.container_name}."
                         yield TriggerEvent({"status": "success", "message": message})
+                        return
                     else:
                         message = (
                             f"Prefix {self.prefix} not available yet in container {self.container_name}."