gopidesupavan commented on code in PR #50001:
URL: https://github.com/apache/airflow/pull/50001#discussion_r2071654632
##########
providers/microsoft/azure/src/airflow/providers/microsoft/azure/triggers/powerbi.py:
##########
@@ -230,3 +230,165 @@ async def fetch_refresh_status_and_error() -> tuple[str,
str]:
"dataset_refresh_id": self.dataset_refresh_id,
}
)
+
+
+class PowerBIWorkspaceListTrigger(BaseTrigger):
+ """
+ Triggers a call to the API to request the available workspace IDs.
+
+ :param conn_id: The connection Id to connect to PowerBI.
+ :param timeout: The HTTP timeout being used by the `KiotaRequestAdapter`.
Default is 1 week (60s * 60m * 24h * 7d).
+ When no timeout is specified or set to None then there is no HTTP
timeout on each request.
+ :param proxies: A dict defining the HTTP proxies to be used (default is
None).
+ :param api_version: The API version of the Microsoft Graph API to be used
(default is v1).
+ You can pass an enum named APIVersion which has 2 possible members v1
and beta,
+ or you can pass a string as `v1.0` or `beta`.
+ """
+
+ def __init__(
+ self,
+ conn_id: str,
+ workspace_ids: list[str] | None = None,
+ timeout: float = 60 * 60 * 24 * 7,
+ proxies: dict | None = None,
+ api_version: APIVersion | str | None = None,
+ ):
+ super().__init__()
+ self.hook = PowerBIHook(conn_id=conn_id, proxies=proxies,
api_version=api_version, timeout=timeout)
+ self.timeout = timeout
+ self.workspace_ids = workspace_ids
+
+ def serialize(self):
+ """Serialize the trigger instance."""
+ return (
+
"airflow.providers.microsoft.azure.triggers.powerbi.PowerBIWorkspaceListTrigger",
+ {
+ "conn_id": self.conn_id,
+ "proxies": self.proxies,
+ "api_version": self.api_version,
+ "timeout": self.timeout,
+ "workspace_ids": self.workspace_ids,
+ },
+ )
+
+ @property
+ def conn_id(self) -> str:
+ return self.hook.conn_id
+
+ @property
+ def proxies(self) -> dict | None:
+ return self.hook.proxies
+
+ @property
+ def api_version(self) -> APIVersion | str:
+ return self.hook.api_version
+
+ async def run(self) -> AsyncIterator[TriggerEvent]:
+ """Make async connection to the PowerBI and polls for the list of
workspace IDs."""
+ if not self.workspace_ids:
+ # Trigger the API to get the workspace list
+ workspace_ids = await self.hook.get_workspace_list()
+
+ if workspace_ids:
+ self.log.info("Triggered request to get workspace list.")
+ yield TriggerEvent(
+ {
+ "status": "success",
+ "message": "The workspace list get request has been
successful.",
+ "workspace_ids": workspace_ids,
+ }
+ )
+ return
+
+ yield TriggerEvent(
+ {
+ "status": "error",
+ "message": "Error grabbing the workspace list.",
+ "workspace_ids": None,
+ }
+ )
+ return
Review Comment:
When condition not met here this will not yield anything, Triggers should
yield with `TriggerEvent`, please add some information and return.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
