feat(cloud): add polling_timeout_seconds parameter to polling methods

Author: pedrodeoliveira

src/gcore/resources/cloud/baremetal/servers.py

@@ -477,6 +477,7 @@ def rebuild_and_poll(
         image_id: str | Omit = omit,
         user_data: str | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -502,6 +503,7 @@ def rebuild_and_poll(
             response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         servers = self.list(
             project_id=project_id,
@@ -904,6 +906,7 @@ async def create_and_poll(
         user_data: str | Omit = omit,
         username: str | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -939,6 +942,7 @@ async def create_and_poll(
             response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if not task.created_resources or not task.created_resources.instances:
             raise ValueError("No server was created")
@@ -964,6 +968,7 @@ async def rebuild_and_poll(
         image_id: str | Omit = omit,
         user_data: str | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -989,6 +994,7 @@ async def rebuild_and_poll(
             response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         servers = await self.list(
             project_id=project_id,

src/gcore/resources/cloud/file_shares/file_shares.py

@@ -319,6 +319,7 @@ def update_and_poll(
         share_settings: file_share_update_params.ShareSettings | Omit = omit,
         tags: Optional[TagUpdateMapParam] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -344,6 +345,7 @@ def update_and_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(
             file_share_id,
@@ -840,6 +842,7 @@ async def update_and_poll(
         share_settings: file_share_update_params.ShareSettings | Omit = omit,
         tags: Optional[TagUpdateMapParam] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -865,6 +868,7 @@ async def update_and_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(
             file_share_id,

src/gcore/resources/cloud/floating_ips.py

@@ -445,6 +445,7 @@ def create_and_poll(
         port_id: Optional[str] | Omit = omit,
         tags: Dict[str, str] | 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.
@@ -472,6 +473,7 @@ def create_and_poll(
             task_id=response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if task.created_resources is None or task.created_resources.floatingips is None:
             raise ValueError("Task completed but created_resources or floatingips is missing")
@@ -493,6 +495,7 @@ def delete_and_poll(
         project_id: int | None = None,
         region_id: int | None = None,
         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.
@@ -518,6 +521,7 @@ def delete_and_poll(
             task_id=response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
 
@@ -934,6 +938,7 @@ async def create_and_poll(
         port_id: Optional[str] | Omit = omit,
         tags: Dict[str, str] | 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.
@@ -961,6 +966,7 @@ async def create_and_poll(
             task_id=response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if task.created_resources is None or task.created_resources.floatingips is None:
             raise ValueError("Task completed but created_resources or floatingips is missing")
@@ -982,6 +988,7 @@ async def delete_and_poll(
         project_id: int | None = None,
         region_id: int | None = None,
         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.
@@ -1007,6 +1014,7 @@ async def delete_and_poll(
             task_id=response.tasks[0],
             extra_headers=extra_headers,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
 

src/gcore/resources/cloud/gpu_baremetal_clusters/gpu_baremetal_clusters.py

@@ -635,6 +635,7 @@ def create_and_poll(
         servers_settings: gpu_baremetal_cluster_create_params.ServersSettings,
         tags: Dict[str, str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -667,6 +668,7 @@ def create_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if not task.created_resources or not task.created_resources.ai_clusters:
             raise ValueError("No cluster was created")
@@ -691,6 +693,7 @@ def rebuild_and_poll(
         image_id: Optional[str] | Omit = omit,
         user_data: Optional[str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -721,6 +724,7 @@ def rebuild_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         return self.get(  # pyright: ignore[reportDeprecated]
             cluster_id=cluster_id,
@@ -740,6 +744,7 @@ def resize_and_poll(
         region_id: int | None = None,
         instances_count: int,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -768,6 +773,7 @@ def resize_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         return self.get(  # pyright: ignore[reportDeprecated]
             cluster_id=cluster_id,
@@ -1348,6 +1354,7 @@ async def create_and_poll(
         servers_settings: gpu_baremetal_cluster_create_params.ServersSettings,
         tags: Dict[str, str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -1380,6 +1387,7 @@ async def create_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if not task.created_resources or not task.created_resources.ai_clusters:
             raise ValueError("No cluster was created")
@@ -1404,6 +1412,7 @@ async def rebuild_and_poll(
         image_id: Optional[str] | Omit = omit,
         user_data: Optional[str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -1434,6 +1443,7 @@ async def rebuild_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         return await self.get(  # pyright: ignore[reportDeprecated]
             cluster_id=cluster_id,
@@ -1453,6 +1463,7 @@ async def resize_and_poll(
         region_id: int | None = None,
         instances_count: int,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -1481,6 +1492,7 @@ async def resize_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         return await self.get(  # pyright: ignore[reportDeprecated]
             cluster_id=cluster_id,

src/gcore/resources/cloud/gpu_baremetal_clusters/images.py

@@ -138,6 +138,7 @@ def delete_and_poll(
         project_id: int | None = None,
         region_id: int | None = None,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -165,6 +166,7 @@ def delete_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
     def get(
@@ -318,6 +320,7 @@ def upload_and_poll(
         ssh_key: Literal["allow", "deny", "required"] | Omit = omit,
         tags: Dict[str, str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -354,6 +357,7 @@ def upload_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if not task.created_resources or not task.created_resources.images:
             raise ValueError("No image was created")
@@ -481,6 +485,7 @@ async def delete_and_poll(
         project_id: int | None = None,
         region_id: int | None = None,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -508,6 +513,7 @@ async def delete_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
     async def get(
@@ -661,6 +667,7 @@ async def upload_and_poll(
         ssh_key: Literal["allow", "deny", "required"] | Omit = omit,
         tags: Dict[str, str] | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -697,6 +704,7 @@ async def upload_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
         if not task.created_resources or not task.created_resources.images:
             raise ValueError("No image was created")

src/gcore/resources/cloud/gpu_baremetal_clusters/servers.py

@@ -229,6 +229,7 @@ def delete_and_poll(
         cluster_id: str,
         delete_floatings: bool | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -258,6 +259,7 @@ def delete_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
     @overload
@@ -876,6 +878,7 @@ async def delete_and_poll(
         cluster_id: str,
         delete_floatings: bool | Omit = omit,
         polling_interval_seconds: int | Omit = omit,
+        polling_timeout_seconds: int | Omit = omit,
         # 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,
@@ -905,6 +908,7 @@ async def delete_and_poll(
             extra_query=extra_query,
             extra_body=extra_body,
             polling_interval_seconds=polling_interval_seconds,
+            polling_timeout_seconds=polling_timeout_seconds,
         )
 
     @overload