# 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.delete_bulk_response import DeleteBulkResponse
from .types.post_bulk_response import PostBulkResponse
# this is used as the default value for optional parameters
OMIT = typing.cast(typing.Any, ...)
class RawBulkClient:
def __init__(self, *, client_wrapper: SyncClientWrapper):
self._client_wrapper = client_wrapper
def post(
self,
id: int,
*,
all_: bool,
excluded: typing.Optional[typing.Sequence[int]] = OMIT,
included: typing.Optional[typing.Sequence[int]] = OMIT,
request_options: typing.Optional[RequestOptions] = None,
) -> HttpResponse[PostBulkResponse]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Assign workspace members in bulk.
Parameters
----------
id : int
all_ : bool
Apply to all workspace members
excluded : typing.Optional[typing.Sequence[int]]
Excluded user IDs
included : typing.Optional[typing.Sequence[int]]
Included user IDs
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
HttpResponse[PostBulkResponse]
"""
_response = self._client_wrapper.httpx_client.request(
f"api/workspaces/{jsonable_encoder(id)}/memberships/bulk/",
method="POST",
json={
"all": all_,
"excluded": excluded,
"included": included,
},
headers={
"content-type": "application/json",
},
request_options=request_options,
omit=OMIT,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
PostBulkResponse,
construct_type(
type_=PostBulkResponse, # 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[DeleteBulkResponse]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Unassign workspace members in bulk. Allows the same request body as bulk assign.
Parameters
----------
id : int
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
HttpResponse[DeleteBulkResponse]
"""
_response = self._client_wrapper.httpx_client.request(
f"api/workspaces/{jsonable_encoder(id)}/memberships/bulk/",
method="DELETE",
request_options=request_options,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
DeleteBulkResponse,
construct_type(
type_=DeleteBulkResponse, # 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 AsyncRawBulkClient:
def __init__(self, *, client_wrapper: AsyncClientWrapper):
self._client_wrapper = client_wrapper
async def post(
self,
id: int,
*,
all_: bool,
excluded: typing.Optional[typing.Sequence[int]] = OMIT,
included: typing.Optional[typing.Sequence[int]] = OMIT,
request_options: typing.Optional[RequestOptions] = None,
) -> AsyncHttpResponse[PostBulkResponse]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Assign workspace members in bulk.
Parameters
----------
id : int
all_ : bool
Apply to all workspace members
excluded : typing.Optional[typing.Sequence[int]]
Excluded user IDs
included : typing.Optional[typing.Sequence[int]]
Included user IDs
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
AsyncHttpResponse[PostBulkResponse]
"""
_response = await self._client_wrapper.httpx_client.request(
f"api/workspaces/{jsonable_encoder(id)}/memberships/bulk/",
method="POST",
json={
"all": all_,
"excluded": excluded,
"included": included,
},
headers={
"content-type": "application/json",
},
request_options=request_options,
omit=OMIT,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
PostBulkResponse,
construct_type(
type_=PostBulkResponse, # 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[DeleteBulkResponse]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Unassign workspace members in bulk. Allows the same request body as bulk assign.
Parameters
----------
id : int
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
AsyncHttpResponse[DeleteBulkResponse]
"""
_response = await self._client_wrapper.httpx_client.request(
f"api/workspaces/{jsonable_encoder(id)}/memberships/bulk/",
method="DELETE",
request_options=request_options,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
DeleteBulkResponse,
construct_type(
type_=DeleteBulkResponse, # 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)