feat(cloud): add update_and_poll method for routers

Author: pedrodeoliveira

src/gcore/resources/cloud/networks/routers.py

@@ -188,6 +188,56 @@ def update(
             cast_to=TaskIDList,
         )
 
+    def update_and_poll(
+        self,
+        router_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        external_gateway_info: router_update_params.ExternalGatewayInfo | Omit = omit,
+        name: str | Omit = omit,
+        routes: Iterable[router_update_params.Route] | Omit = omit,
+        polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+    ) -> Router:
+        """
+        Update router and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = self.update(
+            router_id=router_id,
+            project_id=project_id,
+            region_id=region_id,
+            external_gateway_info=external_gateway_info,
+            name=name,
+            routes=routes,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if response.tasks:
+            self._client.cloud.tasks.poll(
+                task_id=response.tasks[0],
+                extra_headers=extra_headers,
+                polling_interval_seconds=polling_interval_seconds,
+                polling_timeout_seconds=polling_timeout_seconds,
+            )
+        return self.get(
+            router_id=router_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
     def list(
         self,
         *,
@@ -607,6 +657,56 @@ async def update(
             cast_to=TaskIDList,
         )
 
+    async def update_and_poll(
+        self,
+        router_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        external_gateway_info: router_update_params.ExternalGatewayInfo | Omit = omit,
+        name: str | Omit = omit,
+        routes: Iterable[router_update_params.Route] | Omit = omit,
+        polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+    ) -> Router:
+        """
+        Update router and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = await self.update(
+            router_id=router_id,
+            project_id=project_id,
+            region_id=region_id,
+            external_gateway_info=external_gateway_info,
+            name=name,
+            routes=routes,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if response.tasks:
+            await self._client.cloud.tasks.poll(
+                task_id=response.tasks[0],
+                extra_headers=extra_headers,
+                polling_interval_seconds=polling_interval_seconds,
+                polling_timeout_seconds=polling_timeout_seconds,
+            )
+        return await self.get(
+            router_id=router_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
     def list(
         self,
         *,
@@ -881,6 +981,9 @@ def __init__(self, routers: RoutersResource) -> None:
         self.update = to_raw_response_wrapper(
             routers.update,
         )
+        self.update_and_poll = to_raw_response_wrapper(
+            routers.update_and_poll,
+        )
         self.list = to_raw_response_wrapper(
             routers.list,
         )
@@ -908,6 +1011,9 @@ def __init__(self, routers: AsyncRoutersResource) -> None:
         self.update = async_to_raw_response_wrapper(
             routers.update,
         )
+        self.update_and_poll = async_to_raw_response_wrapper(
+            routers.update_and_poll,
+        )
         self.list = async_to_raw_response_wrapper(
             routers.list,
         )
@@ -935,6 +1041,9 @@ def __init__(self, routers: RoutersResource) -> None:
         self.update = to_streamed_response_wrapper(
             routers.update,
         )
+        self.update_and_poll = to_streamed_response_wrapper(
+            routers.update_and_poll,
+        )
         self.list = to_streamed_response_wrapper(
             routers.list,
         )
@@ -962,6 +1071,9 @@ def __init__(self, routers: AsyncRoutersResource) -> None:
         self.update = async_to_streamed_response_wrapper(
             routers.update,
         )
+        self.update_and_poll = async_to_streamed_response_wrapper(
+            routers.update_and_poll,
+        )
         self.list = async_to_streamed_response_wrapper(
             routers.list,
         )