244 lines
4.2 KiB
Python
244 lines
4.2 KiB
Python
from typing import Any, Dict, Optional, Union
|
|
|
|
import httpx
|
|
|
|
from ...client import Client
|
|
from ...models.error import Error
|
|
from ...models.file_export_format import FileExportFormat
|
|
from ...models.image_type import ImageType
|
|
from ...models.mesh import Mesh
|
|
from ...types import Response
|
|
|
|
|
|
def _get_kwargs(
|
|
|
|
|
|
input_format: ImageType,
|
|
|
|
|
|
|
|
output_format: FileExportFormat,
|
|
|
|
|
|
|
|
body: bytes,
|
|
|
|
|
|
*,
|
|
client: Client,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
) -> Dict[str, Any]:
|
|
url = "{}/ai/image-to-3d/{input_format}/{output_format}".format(client.base_url, input_format=input_format,output_format=output_format,) # noqa: E501
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
headers: Dict[str, Any] = client.get_headers()
|
|
cookies: Dict[str, Any] = client.get_cookies()
|
|
|
|
return {
|
|
"url": url,
|
|
"headers": headers,
|
|
"cookies": cookies,
|
|
"timeout": client.get_timeout(),
|
|
"content": body,
|
|
}
|
|
|
|
|
|
def _parse_response(*, response: httpx.Response) -> Optional[Union[Mesh, Error]] :
|
|
if response.status_code == 200:
|
|
response_200 = Mesh.from_dict(response.json())
|
|
return response_200
|
|
if response.status_code == 400:
|
|
response_4XX = Error.from_dict(response.json())
|
|
return response_4XX
|
|
if response.status_code == 500:
|
|
response_5XX = Error.from_dict(response.json())
|
|
return response_5XX
|
|
return Error.from_dict(response.json())
|
|
|
|
|
|
|
|
def _build_response(
|
|
*, response: httpx.Response
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
return Response(
|
|
status_code=response.status_code,
|
|
content=response.content,
|
|
headers=response.headers,
|
|
parsed=_parse_response(response=response),
|
|
)
|
|
|
|
|
|
def sync_detailed(
|
|
|
|
|
|
input_format: ImageType,
|
|
|
|
|
|
|
|
output_format: FileExportFormat,
|
|
|
|
|
|
|
|
body: bytes,
|
|
|
|
|
|
*,
|
|
client: Client,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
kwargs = _get_kwargs(
|
|
|
|
input_format=input_format,
|
|
|
|
output_format=output_format,
|
|
|
|
body=body,
|
|
|
|
client=client,
|
|
)
|
|
|
|
response = httpx.post(
|
|
verify=client.verify_ssl,
|
|
**kwargs,
|
|
)
|
|
|
|
return _build_response(response=response)
|
|
|
|
|
|
def sync(
|
|
|
|
|
|
input_format: ImageType,
|
|
|
|
|
|
|
|
output_format: FileExportFormat,
|
|
|
|
|
|
|
|
body: bytes,
|
|
|
|
|
|
*,
|
|
client: Client,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
) -> Optional[Union[Mesh, Error]] :
|
|
"""This is an alpha endpoint. It will change in the future. The current output is honestly pretty bad. So if you find this endpoint, you get what you pay for, which currently is nothing. But in the future will be made a lot better.""" # noqa: E501
|
|
|
|
return sync_detailed(
|
|
|
|
input_format=input_format,
|
|
|
|
output_format=output_format,
|
|
|
|
body=body,
|
|
|
|
client=client,
|
|
).parsed
|
|
|
|
|
|
async def asyncio_detailed(
|
|
|
|
|
|
input_format: ImageType,
|
|
|
|
|
|
|
|
output_format: FileExportFormat,
|
|
|
|
|
|
|
|
body: bytes,
|
|
|
|
|
|
*,
|
|
client: Client,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
kwargs = _get_kwargs(
|
|
|
|
input_format=input_format,
|
|
|
|
output_format=output_format,
|
|
|
|
body=body,
|
|
|
|
client=client,
|
|
)
|
|
|
|
async with httpx.AsyncClient(verify=client.verify_ssl) as _client:
|
|
response = await _client.post(**kwargs)
|
|
|
|
return _build_response(response=response)
|
|
|
|
|
|
async def asyncio(
|
|
|
|
|
|
input_format: ImageType,
|
|
|
|
|
|
|
|
output_format: FileExportFormat,
|
|
|
|
|
|
|
|
body: bytes,
|
|
|
|
|
|
*,
|
|
client: Client,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
) -> Optional[Union[Mesh, Error]] :
|
|
"""This is an alpha endpoint. It will change in the future. The current output is honestly pretty bad. So if you find this endpoint, you get what you pay for, which currently is nothing. But in the future will be made a lot better.""" # noqa: E501
|
|
|
|
return (
|
|
await asyncio_detailed(
|
|
|
|
input_format=input_format,
|
|
|
|
output_format=output_format,
|
|
|
|
body=body,
|
|
|
|
client=client,
|
|
)
|
|
).parsed
|