gcore 0.1.0a2__py3-none-any.whl → 0.3.0__py3-none-any.whl

gcore/resources/cloud/gpu_baremetal_clusters/images.py

@@ -132,6 +132,42 @@ class ImagesResource(SyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    def delete_and_poll(
+        self,
+        image_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Delete a bare metal GPU image 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.delete(
+            image_id=image_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
     def get(
         self,
         image_id: str,
@@ -211,7 +247,7 @@ class ImagesResource(SyncAPIResource):
 
           url: Image URL
 
-          architecture: Image architecture type: aarch64, x86_64
+          architecture: Image architecture type: aarch64, `x86_64`
 
           cow_format: When True, image cannot be deleted unless all volumes, created from it, are
               deleted.
@@ -267,6 +303,72 @@ class ImagesResource(SyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    def upload_and_poll(
+        self,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        name: str,
+        url: str,
+        architecture: Optional[Literal["aarch64", "x86_64"]] | NotGiven = NOT_GIVEN,
+        cow_format: bool | NotGiven = NOT_GIVEN,
+        hw_firmware_type: Optional[Literal["bios", "uefi"]] | NotGiven = NOT_GIVEN,
+        os_distro: Optional[str] | NotGiven = NOT_GIVEN,
+        os_type: Optional[Literal["linux", "windows"]] | NotGiven = NOT_GIVEN,
+        os_version: Optional[str] | NotGiven = NOT_GIVEN,
+        ssh_key: Literal["allow", "deny", "required"] | NotGiven = NOT_GIVEN,
+        tags: TagUpdateMapParam | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> GPUImage:
+        """
+        Upload a new bare metal GPU image and wait for the upload to complete.
+        """
+        response = self.upload(
+            project_id=project_id,
+            region_id=region_id,
+            name=name,
+            url=url,
+            architecture=architecture,
+            cow_format=cow_format,
+            hw_firmware_type=hw_firmware_type,
+            os_distro=os_distro,
+            os_type=os_type,
+            os_version=os_version,
+            ssh_key=ssh_key,
+            tags=tags,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) != 1:
+            raise ValueError(f"Expected exactly one task to be created")
+        task = self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        if not task.created_resources or not task.created_resources.images:
+            raise ValueError("No image was created")
+        image_id = task.created_resources.images[0]
+        return self.get(
+            image_id=image_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
 
 class AsyncImagesResource(AsyncAPIResource):
     @cached_property
@@ -373,6 +475,42 @@ class AsyncImagesResource(AsyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    async def delete_and_poll(
+        self,
+        image_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Delete a bare metal GPU image 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.delete(
+            image_id=image_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        await self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
     async def get(
         self,
         image_id: str,
@@ -452,7 +590,7 @@ class AsyncImagesResource(AsyncAPIResource):
 
           url: Image URL
 
-          architecture: Image architecture type: aarch64, x86_64
+          architecture: Image architecture type: aarch64, `x86_64`
 
           cow_format: When True, image cannot be deleted unless all volumes, created from it, are
               deleted.
@@ -508,6 +646,72 @@ class AsyncImagesResource(AsyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    async def upload_and_poll(
+        self,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        name: str,
+        url: str,
+        architecture: Optional[Literal["aarch64", "x86_64"]] | NotGiven = NOT_GIVEN,
+        cow_format: bool | NotGiven = NOT_GIVEN,
+        hw_firmware_type: Optional[Literal["bios", "uefi"]] | NotGiven = NOT_GIVEN,
+        os_distro: Optional[str] | NotGiven = NOT_GIVEN,
+        os_type: Optional[Literal["linux", "windows"]] | NotGiven = NOT_GIVEN,
+        os_version: Optional[str] | NotGiven = NOT_GIVEN,
+        ssh_key: Literal["allow", "deny", "required"] | NotGiven = NOT_GIVEN,
+        tags: TagUpdateMapParam | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> GPUImage:
+        """
+        Upload a new bare metal GPU image and wait for the upload to complete.
+        """
+        response = await self.upload(
+            project_id=project_id,
+            region_id=region_id,
+            name=name,
+            url=url,
+            architecture=architecture,
+            cow_format=cow_format,
+            hw_firmware_type=hw_firmware_type,
+            os_distro=os_distro,
+            os_type=os_type,
+            os_version=os_version,
+            ssh_key=ssh_key,
+            tags=tags,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) != 1:
+            raise ValueError(f"Expected exactly one task to be created")
+        task = await self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        if not task.created_resources or not task.created_resources.images:
+            raise ValueError("No image was created")
+        image_id = task.created_resources.images[0]
+        return await self.get(
+            image_id=image_id,
+            project_id=project_id,
+            region_id=region_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
 
 class ImagesResourceWithRawResponse:
     def __init__(self, images: ImagesResource) -> None:

gcore/resources/cloud/gpu_baremetal_clusters/servers.py

@@ -100,6 +100,46 @@ class ServersResource(SyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    def delete_and_poll(
+        self,
+        instance_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        cluster_id: str,
+        delete_floatings: bool | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Delete a bare metal GPU server from cluster 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.delete(
+            instance_id=instance_id,
+            project_id=project_id,
+            region_id=region_id,
+            cluster_id=cluster_id,
+            delete_floatings=delete_floatings,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
     @overload
     def attach_interface(
         self,
@@ -233,7 +273,7 @@ class ServersResource(SyncAPIResource):
 
           security_groups: List of security group IDs
 
-          type: Must be 'any_subnet'
+          type: Must be '`any_subnet`'
 
           extra_headers: Send extra headers
 
@@ -281,7 +321,7 @@ class ServersResource(SyncAPIResource):
 
           security_groups: List of security group IDs
 
-          type: Must be 'reserved_fixed_ip'. Union tag
+          type: Must be '`reserved_fixed_ip`'. Union tag
 
           extra_headers: Send extra headers
 
@@ -587,6 +627,46 @@ class AsyncServersResource(AsyncAPIResource):
             cast_to=TaskIDList,
         )
 
+    async def delete_and_poll(
+        self,
+        instance_id: str,
+        *,
+        project_id: int | None = None,
+        region_id: int | None = None,
+        cluster_id: str,
+        delete_floatings: bool | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | 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,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Delete a bare metal GPU server from cluster 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.delete(
+            instance_id=instance_id,
+            project_id=project_id,
+            region_id=region_id,
+            cluster_id=cluster_id,
+            delete_floatings=delete_floatings,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        await self._client.cloud.tasks.poll(
+            response.tasks[0],
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
     @overload
     async def attach_interface(
         self,
@@ -720,7 +800,7 @@ class AsyncServersResource(AsyncAPIResource):
 
           security_groups: List of security group IDs
 
-          type: Must be 'any_subnet'
+          type: Must be '`any_subnet`'
 
           extra_headers: Send extra headers
 
@@ -768,7 +848,7 @@ class AsyncServersResource(AsyncAPIResource):
 
           security_groups: List of security group IDs
 
-          type: Must be 'reserved_fixed_ip'. Union tag
+          type: Must be '`reserved_fixed_ip`'. Union tag
 
           extra_headers: Send extra headers