magic_hour 0.9.5__py3-none-any.whl → 0.44.0__py3-none-any.whl

magic_hour/resources/v1/video_projects/client.py

@@ -1,18 +1,96 @@
+import os
+import pydantic
+import time
 import typing
 
-from magic_hour.core import (
+from magic_hour.helpers.download import download_files_async, download_files_sync
+from magic_hour.helpers.logger import get_sdk_logger
+from magic_hour.types import models
+from make_api_request import (
     AsyncBaseClient,
     RequestOptions,
     SyncBaseClient,
     default_request_options,
 )
-from magic_hour.types import models
+
+
+logger = get_sdk_logger(__name__)
+
+
+class V1VideoProjectsGetResponseWithDownloads(models.V1VideoProjectsGetResponse):
+    downloaded_paths: typing.Optional[typing.List[str]] = pydantic.Field(
+        default=None, alias="downloaded_paths"
+    )
+    """
+    The paths to the downloaded files.
+
+    This field is only populated if `download_outputs` is True and the video project is complete.
+    """
 
 
 class VideoProjectsClient:
     def __init__(self, *, base_client: SyncBaseClient):
         self._base_client = base_client
 
+    def check_result(
+        self,
+        id: str,
+        wait_for_completion: bool,
+        download_outputs: bool,
+        download_directory: typing.Optional[str] = None,
+    ) -> V1VideoProjectsGetResponseWithDownloads:
+        """
+        Check the result of a video project with optional waiting and downloading.
+
+        This method retrieves the status of a video project and optionally waits for completion
+        and downloads the output files.
+
+        Args:
+            id: Unique ID of the video project
+            wait_for_completion: Whether to wait for the video project to complete
+            download_outputs: Whether to download the outputs
+            download_directory: The directory to download the outputs to. If not provided,
+                the outputs will be downloaded to the current working directory
+
+        Returns:
+            V1VideoProjectsGetResponseWithDownloads: The video project response with optional
+                downloaded file paths included
+        """
+        api_response = self.get(id=id)
+        if not wait_for_completion:
+            response = V1VideoProjectsGetResponseWithDownloads(
+                **api_response.model_dump()
+            )
+            return response
+
+        poll_interval = float(os.getenv("MAGIC_HOUR_POLL_INTERVAL", "0.5"))
+
+        status = api_response.status
+
+        while status not in ["complete", "error", "canceled"]:
+            api_response = self.get(id=id)
+            status = api_response.status
+            time.sleep(poll_interval)
+
+        if api_response.status != "complete":
+            log = logger.error if api_response.status == "error" else logger.info
+            log(
+                f"Video project {id} has status {api_response.status}: {api_response.error}"
+            )
+            return V1VideoProjectsGetResponseWithDownloads(**api_response.model_dump())
+
+        if not download_outputs:
+            return V1VideoProjectsGetResponseWithDownloads(**api_response.model_dump())
+
+        downloaded_paths = download_files_sync(
+            downloads=api_response.downloads,
+            download_directory=download_directory,
+        )
+
+        return V1VideoProjectsGetResponseWithDownloads(
+            **api_response.model_dump(), downloaded_paths=downloaded_paths
+        )
+
     def delete(
         self, *, id: str, request_options: typing.Optional[RequestOptions] = None
     ) -> None:
@@ -24,7 +102,7 @@ class VideoProjectsClient:
         DELETE /v1/video-projects/{id}
 
         Args:
-            id: The id of the video project
+            id: Unique ID of the video project. This value is returned by all of the POST APIs that create a video.
             request_options: Additional options to customize the HTTP request
 
         Returns:
@@ -36,7 +114,7 @@ class VideoProjectsClient:
 
         Examples:
         ```py
-        client.v1.video_projects.delete(id="cm6pvghix03bvyz0zwash6noj")
+        client.v1.video_projects.delete(id="cuid-example")
         ```
         """
         self._base_client.request(
@@ -49,7 +127,7 @@ class VideoProjectsClient:
 
     def get(
         self, *, id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> models.GetV1VideoProjectsIdResponse:
+    ) -> models.V1VideoProjectsGetResponse:
         """
         Get video details
 
@@ -67,7 +145,7 @@ class VideoProjectsClient:
         GET /v1/video-projects/{id}
 
         Args:
-            id: The id of the video
+            id: Unique ID of the video project. This value is returned by all of the POST APIs that create a video.
             request_options: Additional options to customize the HTTP request
 
         Returns:
@@ -79,14 +157,14 @@ class VideoProjectsClient:
 
         Examples:
         ```py
-        client.v1.video_projects.get(id="cm6pvghix03bvyz0zwash6noj")
+        client.v1.video_projects.get(id="cuid-example")
         ```
         """
         return self._base_client.request(
             method="GET",
             path=f"/v1/video-projects/{id}",
             auth_names=["bearerAuth"],
-            cast_to=models.GetV1VideoProjectsIdResponse,
+            cast_to=models.V1VideoProjectsGetResponse,
             request_options=request_options or default_request_options(),
         )
 
@@ -95,6 +173,65 @@ class AsyncVideoProjectsClient:
     def __init__(self, *, base_client: AsyncBaseClient):
         self._base_client = base_client
 
+    async def check_result(
+        self,
+        id: str,
+        wait_for_completion: bool,
+        download_outputs: bool,
+        download_directory: typing.Optional[str] = None,
+    ) -> V1VideoProjectsGetResponseWithDownloads:
+        """
+        Check the result of a video project with optional waiting and downloading.
+
+        This method retrieves the status of a video project and optionally waits for completion
+        and downloads the output files.
+
+        Args:
+            id: Unique ID of the video project
+            wait_for_completion: Whether to wait for the video project to complete
+            download_outputs: Whether to download the outputs
+            download_directory: The directory to download the outputs to. If not provided,
+                the outputs will be downloaded to the current working directory
+
+        Returns:
+            V1VideoProjectsGetResponseWithDownloads: The video project response with optional
+                downloaded file paths included
+        """
+        api_response = await self.get(id=id)
+        if not wait_for_completion:
+            response = V1VideoProjectsGetResponseWithDownloads(
+                **api_response.model_dump()
+            )
+            return response
+
+        poll_interval = float(os.getenv("MAGIC_HOUR_POLL_INTERVAL", "0.5"))
+
+        status = api_response.status
+
+        while status not in ["complete", "error", "canceled"]:
+            api_response = await self.get(id=id)
+            status = api_response.status
+            time.sleep(poll_interval)
+
+        if api_response.status != "complete":
+            log = logger.error if api_response.status == "error" else logger.info
+            log(
+                f"Video project {id} has status {api_response.status}: {api_response.error}"
+            )
+            return V1VideoProjectsGetResponseWithDownloads(**api_response.model_dump())
+
+        if not download_outputs:
+            return V1VideoProjectsGetResponseWithDownloads(**api_response.model_dump())
+
+        downloaded_paths = await download_files_async(
+            downloads=api_response.downloads,
+            download_directory=download_directory,
+        )
+
+        return V1VideoProjectsGetResponseWithDownloads(
+            **api_response.model_dump(), downloaded_paths=downloaded_paths
+        )
+
     async def delete(
         self, *, id: str, request_options: typing.Optional[RequestOptions] = None
     ) -> None:
@@ -106,7 +243,7 @@ class AsyncVideoProjectsClient:
         DELETE /v1/video-projects/{id}
 
         Args:
-            id: The id of the video project
+            id: Unique ID of the video project. This value is returned by all of the POST APIs that create a video.
             request_options: Additional options to customize the HTTP request
 
         Returns:
@@ -118,7 +255,7 @@ class AsyncVideoProjectsClient:
 
         Examples:
         ```py
-        await client.v1.video_projects.delete(id="cm6pvghix03bvyz0zwash6noj")
+        await client.v1.video_projects.delete(id="cuid-example")
         ```
         """
         await self._base_client.request(
@@ -131,7 +268,7 @@ class AsyncVideoProjectsClient:
 
     async def get(
         self, *, id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> models.GetV1VideoProjectsIdResponse:
+    ) -> models.V1VideoProjectsGetResponse:
         """
         Get video details
 
@@ -149,7 +286,7 @@ class AsyncVideoProjectsClient:
         GET /v1/video-projects/{id}
 
         Args:
-            id: The id of the video
+            id: Unique ID of the video project. This value is returned by all of the POST APIs that create a video.
             request_options: Additional options to customize the HTTP request
 
         Returns:
@@ -161,13 +298,13 @@ class AsyncVideoProjectsClient:
 
         Examples:
         ```py
-        await client.v1.video_projects.get(id="cm6pvghix03bvyz0zwash6noj")
+        await client.v1.video_projects.get(id="cuid-example")
         ```
         """
         return await self._base_client.request(
             method="GET",
             path=f"/v1/video-projects/{id}",
             auth_names=["bearerAuth"],
-            cast_to=models.GetV1VideoProjectsIdResponse,
+            cast_to=models.V1VideoProjectsGetResponse,
             request_options=request_options or default_request_options(),
         )