|
6 | 6 |
|
7 | 7 | import httpx |
8 | 8 |
|
9 | | -from ....._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes |
10 | | -from ....._utils import maybe_transform, async_maybe_transform |
| 9 | +from ....._files import read_file_content, async_read_file_content |
| 10 | +from ....._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileContent |
11 | 11 | from ....._compat import cached_property |
12 | 12 | from ....._resource import SyncAPIResource, AsyncAPIResource |
13 | 13 | from ....._response import ( |
|
19 | 19 | from ....._wrappers import ResultWrapper |
20 | 20 | from ....._base_client import make_request_options |
21 | 21 | from .....types.zero_trust.dlp.dataset import Dataset |
22 | | -from .....types.zero_trust.dlp.datasets import upload_edit_params |
23 | 22 | from .....types.zero_trust.dlp.datasets.new_version import NewVersion |
24 | 23 |
|
25 | 24 | __all__ = ["UploadResource", "AsyncUploadResource"] |
@@ -88,10 +87,10 @@ def create( |
88 | 87 | def edit( |
89 | 88 | self, |
90 | 89 | version: int, |
| 90 | + body: FileContent, |
91 | 91 | *, |
92 | 92 | account_id: str, |
93 | 93 | dataset_id: str, |
94 | | - body: FileTypes, |
95 | 94 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
96 | 95 | # The extra values given here take precedence over values defined on the client or passed to this method. |
97 | 96 | extra_headers: Headers | None = None, |
@@ -119,9 +118,10 @@ def edit( |
119 | 118 | raise ValueError(f"Expected a non-empty value for `account_id` but received {account_id!r}") |
120 | 119 | if not dataset_id: |
121 | 120 | raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}") |
| 121 | + extra_headers = {"Content-Type": "application/octet-stream", **(extra_headers or {})} |
122 | 122 | return self._post( |
123 | 123 | f"/accounts/{account_id}/dlp/datasets/{dataset_id}/upload/{version}", |
124 | | - body=maybe_transform(body, upload_edit_params.UploadEditParams), |
| 124 | + body=read_file_content(body), |
125 | 125 | options=make_request_options( |
126 | 126 | extra_headers=extra_headers, |
127 | 127 | extra_query=extra_query, |
@@ -196,10 +196,10 @@ async def create( |
196 | 196 | async def edit( |
197 | 197 | self, |
198 | 198 | version: int, |
| 199 | + body: FileContent, |
199 | 200 | *, |
200 | 201 | account_id: str, |
201 | 202 | dataset_id: str, |
202 | | - body: FileTypes, |
203 | 203 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
204 | 204 | # The extra values given here take precedence over values defined on the client or passed to this method. |
205 | 205 | extra_headers: Headers | None = None, |
@@ -227,9 +227,10 @@ async def edit( |
227 | 227 | raise ValueError(f"Expected a non-empty value for `account_id` but received {account_id!r}") |
228 | 228 | if not dataset_id: |
229 | 229 | raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}") |
| 230 | + extra_headers = {"Content-Type": "application/octet-stream", **(extra_headers or {})} |
230 | 231 | return await self._post( |
231 | 232 | f"/accounts/{account_id}/dlp/datasets/{dataset_id}/upload/{version}", |
232 | | - body=await async_maybe_transform(body, upload_edit_params.UploadEditParams), |
| 233 | + body=await async_read_file_content(body), |
233 | 234 | options=make_request_options( |
234 | 235 | extra_headers=extra_headers, |
235 | 236 | extra_query=extra_query, |
|
0 commit comments