# This file was auto-generated by Fern from our API Definition. import typing from json.decoder import JSONDecodeError from ...core.api_error import ApiError from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ...core.http_response import AsyncHttpResponse, HttpResponse from ...core.jsonable_encoder import jsonable_encoder from ...core.request_options import RequestOptions from ...core.unchecked_base_model import construct_type from ...types.azure_blob_import_storage import AzureBlobImportStorage # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawAzureClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def list( self, *, project: int, ordering: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.List[AzureBlobImportStorage]]: """ Get list of all Azure import storage connections. Parameters ---------- project : int Project ID ordering : typing.Optional[str] Which field to use when ordering the results. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.List[AzureBlobImportStorage]] """ _response = self._client_wrapper.httpx_client.request( "api/storages/azure/", method="GET", params={ "ordering": ordering, "project": project, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.List[AzureBlobImportStorage], construct_type( type_=typing.List[AzureBlobImportStorage], # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def create( self, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[AzureBlobImportStorage]: """ Create new Azure import storage Parameters ---------- account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[AzureBlobImportStorage] """ _response = self._client_wrapper.httpx_client.request( "api/storages/azure/", method="POST", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def validate( self, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, id: typing.Optional[int] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[None]: """ Validate a specific Azure import storage connection. Parameters ---------- account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description id : typing.Optional[int] Storage ID. If set, storage with specified ID will be updated prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[None] """ _response = self._client_wrapper.httpx_client.request( "api/storages/azure/validate", method="POST", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "id": id, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: return HttpResponse(response=_response, data=None) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get( self, id: int, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[AzureBlobImportStorage]: """ Get a specific Azure import storage connection. Parameters ---------- id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[AzureBlobImportStorage] """ _response = self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[None]: """ Delete a specific Azure import storage connection. Parameters ---------- id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[None] """ _response = self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="DELETE", request_options=request_options, ) try: if 200 <= _response.status_code < 300: return HttpResponse(response=_response, data=None) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def update( self, id: int, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[AzureBlobImportStorage]: """ Update a specific Azure import storage connection. Parameters ---------- id : int account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[AzureBlobImportStorage] """ _response = self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="PATCH", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def sync( self, id: int, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[AzureBlobImportStorage]: """ Sync tasks from an Azure import storage connection. Parameters ---------- id : int Storage ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[AzureBlobImportStorage] """ _response = self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}/sync", method="POST", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) class AsyncRawAzureClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def list( self, *, project: int, ordering: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.List[AzureBlobImportStorage]]: """ Get list of all Azure import storage connections. Parameters ---------- project : int Project ID ordering : typing.Optional[str] Which field to use when ordering the results. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.List[AzureBlobImportStorage]] """ _response = await self._client_wrapper.httpx_client.request( "api/storages/azure/", method="GET", params={ "ordering": ordering, "project": project, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.List[AzureBlobImportStorage], construct_type( type_=typing.List[AzureBlobImportStorage], # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def create( self, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[AzureBlobImportStorage]: """ Create new Azure import storage Parameters ---------- account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[AzureBlobImportStorage] """ _response = await self._client_wrapper.httpx_client.request( "api/storages/azure/", method="POST", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def validate( self, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, id: typing.Optional[int] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[None]: """ Validate a specific Azure import storage connection. Parameters ---------- account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description id : typing.Optional[int] Storage ID. If set, storage with specified ID will be updated prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[None] """ _response = await self._client_wrapper.httpx_client.request( "api/storages/azure/validate", method="POST", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "id": id, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: return AsyncHttpResponse(response=_response, data=None) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get( self, id: int, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[AzureBlobImportStorage]: """ Get a specific Azure import storage connection. Parameters ---------- id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[AzureBlobImportStorage] """ _response = await self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def delete( self, id: int, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[None]: """ Delete a specific Azure import storage connection. Parameters ---------- id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[None] """ _response = await self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="DELETE", request_options=request_options, ) try: if 200 <= _response.status_code < 300: return AsyncHttpResponse(response=_response, data=None) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def update( self, id: int, *, account_key: typing.Optional[str] = OMIT, account_name: typing.Optional[str] = OMIT, container: typing.Optional[str] = OMIT, description: typing.Optional[str] = OMIT, prefix: typing.Optional[str] = OMIT, presign: typing.Optional[bool] = OMIT, presign_ttl: typing.Optional[int] = OMIT, project: typing.Optional[int] = OMIT, regex_filter: typing.Optional[str] = OMIT, title: typing.Optional[str] = OMIT, use_blob_urls: typing.Optional[bool] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[AzureBlobImportStorage]: """ Update a specific Azure import storage connection. Parameters ---------- id : int account_key : typing.Optional[str] Azure Blob account key account_name : typing.Optional[str] Azure Blob account name container : typing.Optional[str] Azure blob container description : typing.Optional[str] Storage description prefix : typing.Optional[str] Azure blob prefix name presign : typing.Optional[bool] Presign URLs for direct download presign_ttl : typing.Optional[int] Presign TTL in minutes project : typing.Optional[int] Project ID regex_filter : typing.Optional[str] Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported. title : typing.Optional[str] Storage title use_blob_urls : typing.Optional[bool] Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[AzureBlobImportStorage] """ _response = await self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}", method="PATCH", json={ "account_key": account_key, "account_name": account_name, "container": container, "description": description, "prefix": prefix, "presign": presign, "presign_ttl": presign_ttl, "project": project, "regex_filter": regex_filter, "title": title, "use_blob_urls": use_blob_urls, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def sync( self, id: int, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[AzureBlobImportStorage]: """ Sync tasks from an Azure import storage connection. Parameters ---------- id : int Storage ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[AzureBlobImportStorage] """ _response = await self._client_wrapper.httpx_client.request( f"api/storages/azure/{jsonable_encoder(id)}/sync", method="POST", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( AzureBlobImportStorage, construct_type( type_=AzureBlobImportStorage, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)