22
33from __future__ import annotations
44
5- from typing import Any , Optional , cast
5+ from typing import Any , Type , Optional , cast
66
77import httpx
88
2525 AsyncPaginator ,
2626 make_request_options ,
2727)
28- from ...types .stream import (
29- Watermaks ,
30- WatermarkGetResponse ,
31- WatermarkCreateResponse ,
32- WatermarkDeleteResponse ,
33- watermark_create_params ,
34- watermark_delete_params ,
35- )
28+ from ...types .stream import Watermaks , WatermarkDeleteResponse , watermark_create_params , watermark_delete_params
3629
3730__all__ = ["Watermarks" , "AsyncWatermarks" ]
3831
@@ -62,7 +55,7 @@ def create(
6255 extra_query : Query | None = None ,
6356 extra_body : Body | None = None ,
6457 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
65- ) -> Optional [WatermarkCreateResponse ]:
58+ ) -> Optional [Watermaks ]:
6659 """
6760 Creates watermark profiles using a single `HTTP POST multipart/form-data`
6861 request.
@@ -101,32 +94,27 @@ def create(
10194 """
10295 if not account_id :
10396 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
104- return cast (
105- Optional [WatermarkCreateResponse ],
106- self ._post (
107- f"/accounts/{ account_id } /stream/watermarks" ,
108- body = maybe_transform (
109- {
110- "file" : file ,
111- "name" : name ,
112- "opacity" : opacity ,
113- "padding" : padding ,
114- "position" : position ,
115- "scale" : scale ,
116- },
117- watermark_create_params .WatermarkCreateParams ,
118- ),
119- options = make_request_options (
120- extra_headers = extra_headers ,
121- extra_query = extra_query ,
122- extra_body = extra_body ,
123- timeout = timeout ,
124- post_parser = ResultWrapper [Optional [WatermarkCreateResponse ]]._unwrapper ,
125- ),
126- cast_to = cast (
127- Any , ResultWrapper [WatermarkCreateResponse ]
128- ), # Union types cannot be passed in as arguments in the type system
97+ return self ._post (
98+ f"/accounts/{ account_id } /stream/watermarks" ,
99+ body = maybe_transform (
100+ {
101+ "file" : file ,
102+ "name" : name ,
103+ "opacity" : opacity ,
104+ "padding" : padding ,
105+ "position" : position ,
106+ "scale" : scale ,
107+ },
108+ watermark_create_params .WatermarkCreateParams ,
129109 ),
110+ options = make_request_options (
111+ extra_headers = extra_headers ,
112+ extra_query = extra_query ,
113+ extra_body = extra_body ,
114+ timeout = timeout ,
115+ post_parser = ResultWrapper [Optional [Watermaks ]]._unwrapper ,
116+ ),
117+ cast_to = cast (Type [Optional [Watermaks ]], ResultWrapper [Watermaks ]),
130118 )
131119
132120 def list (
@@ -227,7 +215,7 @@ def get(
227215 extra_query : Query | None = None ,
228216 extra_body : Body | None = None ,
229217 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
230- ) -> Optional [WatermarkGetResponse ]:
218+ ) -> Optional [Watermaks ]:
231219 """
232220 Retrieves details for a single watermark profile.
233221
@@ -248,21 +236,16 @@ def get(
248236 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
249237 if not identifier :
250238 raise ValueError (f"Expected a non-empty value for `identifier` but received { identifier !r} " )
251- return cast (
252- Optional [WatermarkGetResponse ],
253- self ._get (
254- f"/accounts/{ account_id } /stream/watermarks/{ identifier } " ,
255- options = make_request_options (
256- extra_headers = extra_headers ,
257- extra_query = extra_query ,
258- extra_body = extra_body ,
259- timeout = timeout ,
260- post_parser = ResultWrapper [Optional [WatermarkGetResponse ]]._unwrapper ,
261- ),
262- cast_to = cast (
263- Any , ResultWrapper [WatermarkGetResponse ]
264- ), # Union types cannot be passed in as arguments in the type system
239+ return self ._get (
240+ f"/accounts/{ account_id } /stream/watermarks/{ identifier } " ,
241+ options = make_request_options (
242+ extra_headers = extra_headers ,
243+ extra_query = extra_query ,
244+ extra_body = extra_body ,
245+ timeout = timeout ,
246+ post_parser = ResultWrapper [Optional [Watermaks ]]._unwrapper ,
265247 ),
248+ cast_to = cast (Type [Optional [Watermaks ]], ResultWrapper [Watermaks ]),
266249 )
267250
268251
@@ -291,7 +274,7 @@ async def create(
291274 extra_query : Query | None = None ,
292275 extra_body : Body | None = None ,
293276 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
294- ) -> Optional [WatermarkCreateResponse ]:
277+ ) -> Optional [Watermaks ]:
295278 """
296279 Creates watermark profiles using a single `HTTP POST multipart/form-data`
297280 request.
@@ -330,32 +313,27 @@ async def create(
330313 """
331314 if not account_id :
332315 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
333- return cast (
334- Optional [WatermarkCreateResponse ],
335- await self ._post (
336- f"/accounts/{ account_id } /stream/watermarks" ,
337- body = await async_maybe_transform (
338- {
339- "file" : file ,
340- "name" : name ,
341- "opacity" : opacity ,
342- "padding" : padding ,
343- "position" : position ,
344- "scale" : scale ,
345- },
346- watermark_create_params .WatermarkCreateParams ,
347- ),
348- options = make_request_options (
349- extra_headers = extra_headers ,
350- extra_query = extra_query ,
351- extra_body = extra_body ,
352- timeout = timeout ,
353- post_parser = ResultWrapper [Optional [WatermarkCreateResponse ]]._unwrapper ,
354- ),
355- cast_to = cast (
356- Any , ResultWrapper [WatermarkCreateResponse ]
357- ), # Union types cannot be passed in as arguments in the type system
316+ return await self ._post (
317+ f"/accounts/{ account_id } /stream/watermarks" ,
318+ body = await async_maybe_transform (
319+ {
320+ "file" : file ,
321+ "name" : name ,
322+ "opacity" : opacity ,
323+ "padding" : padding ,
324+ "position" : position ,
325+ "scale" : scale ,
326+ },
327+ watermark_create_params .WatermarkCreateParams ,
328+ ),
329+ options = make_request_options (
330+ extra_headers = extra_headers ,
331+ extra_query = extra_query ,
332+ extra_body = extra_body ,
333+ timeout = timeout ,
334+ post_parser = ResultWrapper [Optional [Watermaks ]]._unwrapper ,
358335 ),
336+ cast_to = cast (Type [Optional [Watermaks ]], ResultWrapper [Watermaks ]),
359337 )
360338
361339 def list (
@@ -456,7 +434,7 @@ async def get(
456434 extra_query : Query | None = None ,
457435 extra_body : Body | None = None ,
458436 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
459- ) -> Optional [WatermarkGetResponse ]:
437+ ) -> Optional [Watermaks ]:
460438 """
461439 Retrieves details for a single watermark profile.
462440
@@ -477,21 +455,16 @@ async def get(
477455 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
478456 if not identifier :
479457 raise ValueError (f"Expected a non-empty value for `identifier` but received { identifier !r} " )
480- return cast (
481- Optional [WatermarkGetResponse ],
482- await self ._get (
483- f"/accounts/{ account_id } /stream/watermarks/{ identifier } " ,
484- options = make_request_options (
485- extra_headers = extra_headers ,
486- extra_query = extra_query ,
487- extra_body = extra_body ,
488- timeout = timeout ,
489- post_parser = ResultWrapper [Optional [WatermarkGetResponse ]]._unwrapper ,
490- ),
491- cast_to = cast (
492- Any , ResultWrapper [WatermarkGetResponse ]
493- ), # Union types cannot be passed in as arguments in the type system
458+ return await self ._get (
459+ f"/accounts/{ account_id } /stream/watermarks/{ identifier } " ,
460+ options = make_request_options (
461+ extra_headers = extra_headers ,
462+ extra_query = extra_query ,
463+ extra_body = extra_body ,
464+ timeout = timeout ,
465+ post_parser = ResultWrapper [Optional [Watermaks ]]._unwrapper ,
494466 ),
467+ cast_to = cast (Type [Optional [Watermaks ]], ResultWrapper [Watermaks ]),
495468 )
496469
497470
0 commit comments