Skip to content

Commit f971cf9

Browse files
yuming-longyuming-long
committed
all endpoints
1 parent 2ee0abf commit f971cf9

File tree

5 files changed

+85
-314
lines changed

5 files changed

+85
-314
lines changed

‎src/unstructured_client/destinations.py‎

Lines changed: 20 additions & 75 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,16 @@
66
from unstructured_client._hooks import HookContext
77
from unstructured_client.models import errors, operations, shared
88
from unstructured_client.types import BaseModel, OptionalNullable, UNSET
9-
9+
from unstructured_client.sdkconfiguration import SERVER_PLATFORM_API, SERVERS
1010

1111
class Destinations(BaseSDK):
12+
def get_default_server_url(self) -> str:
13+
client_url, *_ = self.sdk_configuration.get_server_details()
14+
if client_url is None:
15+
return utils.remove_suffix(SERVERS[SERVER_PLATFORM_API], "/")
16+
return client_url
17+
18+
1219
def create_destination(
1320
self,
1421
*,
@@ -31,18 +38,12 @@ def create_destination(
3138
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
3239
:param http_headers: Additional headers to set or replace on requests.
3340
"""
34-
base_url = None
41+
base_url = server_url if server_url is not None else self.get_default_server_url()
42+
3543
url_variables = None
3644
if timeout_ms is None:
3745
timeout_ms = self.sdk_configuration.timeout_ms
3846

39-
if server_url is not None:
40-
base_url = server_url
41-
else:
42-
base_url = operations.CREATE_DESTINATION_SERVERS[
43-
operations.CREATE_DESTINATION_SERVER_PLATFORM_API
44-
]
45-
4647
if not isinstance(request, BaseModel):
4748
request = utils.unmarshal(request, operations.CreateDestinationRequest)
4849
request = cast(operations.CreateDestinationRequest, request)
@@ -151,17 +152,11 @@ async def create_destination_async(
151152
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
152153
:param http_headers: Additional headers to set or replace on requests.
153154
"""
154-
base_url = None
155+
base_url = server_url if server_url is not None else self.get_default_server_url()
155156
url_variables = None
156157
if timeout_ms is None:
157158
timeout_ms = self.sdk_configuration.timeout_ms
158159

159-
if server_url is not None:
160-
base_url = server_url
161-
else:
162-
base_url = operations.CREATE_DESTINATION_SERVERS[
163-
operations.CREATE_DESTINATION_SERVER_PLATFORM_API
164-
]
165160

166161
if not isinstance(request, BaseModel):
167162
request = utils.unmarshal(request, operations.CreateDestinationRequest)
@@ -271,17 +266,12 @@ def delete_destination(
271266
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
272267
:param http_headers: Additional headers to set or replace on requests.
273268
"""
274-
base_url = None
269+
base_url = server_url if server_url is not None else self.get_default_server_url()
275270
url_variables = None
276271
if timeout_ms is None:
277272
timeout_ms = self.sdk_configuration.timeout_ms
278273

279-
if server_url is not None:
280-
base_url = server_url
281-
else:
282-
base_url = operations.DELETE_DESTINATION_SERVERS[
283-
operations.DELETE_DESTINATION_SERVER_PLATFORM_API
284-
]
274+
285275

286276
if not isinstance(request, BaseModel):
287277
request = utils.unmarshal(request, operations.DeleteDestinationRequest)
@@ -382,17 +372,11 @@ async def delete_destination_async(
382372
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
383373
:param http_headers: Additional headers to set or replace on requests.
384374
"""
385-
base_url = None
375+
base_url = server_url if server_url is not None else self.get_default_server_url()
386376
url_variables = None
387377
if timeout_ms is None:
388378
timeout_ms = self.sdk_configuration.timeout_ms
389379

390-
if server_url is not None:
391-
base_url = server_url
392-
else:
393-
base_url = operations.DELETE_DESTINATION_SERVERS[
394-
operations.DELETE_DESTINATION_SERVER_PLATFORM_API
395-
]
396380

397381
if not isinstance(request, BaseModel):
398382
request = utils.unmarshal(request, operations.DeleteDestinationRequest)
@@ -492,17 +476,11 @@ def get_destination(
492476
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
493477
:param http_headers: Additional headers to set or replace on requests.
494478
"""
495-
base_url = None
479+
base_url = server_url if server_url is not None else self.get_default_server_url()
496480
url_variables = None
497481
if timeout_ms is None:
498482
timeout_ms = self.sdk_configuration.timeout_ms
499483

500-
if server_url is not None:
501-
base_url = server_url
502-
else:
503-
base_url = operations.GET_DESTINATION_SERVERS[
504-
operations.GET_DESTINATION_SERVER_PLATFORM_API
505-
]
506484

507485
if not isinstance(request, BaseModel):
508486
request = utils.unmarshal(request, operations.GetDestinationRequest)
@@ -604,17 +582,11 @@ async def get_destination_async(
604582
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
605583
:param http_headers: Additional headers to set or replace on requests.
606584
"""
607-
base_url = None
585+
base_url = server_url if server_url is not None else self.get_default_server_url()
608586
url_variables = None
609587
if timeout_ms is None:
610588
timeout_ms = self.sdk_configuration.timeout_ms
611589

612-
if server_url is not None:
613-
base_url = server_url
614-
else:
615-
base_url = operations.GET_DESTINATION_SERVERS[
616-
operations.GET_DESTINATION_SERVER_PLATFORM_API
617-
]
618590

619591
if not isinstance(request, BaseModel):
620592
request = utils.unmarshal(request, operations.GetDestinationRequest)
@@ -717,18 +689,11 @@ def list_destinations(
717689
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
718690
:param http_headers: Additional headers to set or replace on requests.
719691
"""
720-
base_url = None
692+
base_url = server_url if server_url is not None else self.get_default_server_url()
721693
url_variables = None
722694
if timeout_ms is None:
723695
timeout_ms = self.sdk_configuration.timeout_ms
724696

725-
if server_url is not None:
726-
base_url = server_url
727-
else:
728-
base_url = operations.LIST_DESTINATIONS_SERVERS[
729-
operations.LIST_DESTINATIONS_SERVER_PLATFORM_API
730-
]
731-
732697
if not isinstance(request, BaseModel):
733698
request = utils.unmarshal(request, operations.ListDestinationsRequest)
734699
request = cast(operations.ListDestinationsRequest, request)
@@ -831,18 +796,11 @@ async def list_destinations_async(
831796
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
832797
:param http_headers: Additional headers to set or replace on requests.
833798
"""
834-
base_url = None
799+
base_url = server_url if server_url is not None else self.get_default_server_url()
835800
url_variables = None
836801
if timeout_ms is None:
837802
timeout_ms = self.sdk_configuration.timeout_ms
838803

839-
if server_url is not None:
840-
base_url = server_url
841-
else:
842-
base_url = operations.LIST_DESTINATIONS_SERVERS[
843-
operations.LIST_DESTINATIONS_SERVER_PLATFORM_API
844-
]
845-
846804
if not isinstance(request, BaseModel):
847805
request = utils.unmarshal(request, operations.ListDestinationsRequest)
848806
request = cast(operations.ListDestinationsRequest, request)
@@ -945,17 +903,11 @@ def update_destination(
945903
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
946904
:param http_headers: Additional headers to set or replace on requests.
947905
"""
948-
base_url = None
906+
base_url = server_url if server_url is not None else self.get_default_server_url()
949907
url_variables = None
950908
if timeout_ms is None:
951909
timeout_ms = self.sdk_configuration.timeout_ms
952910

953-
if server_url is not None:
954-
base_url = server_url
955-
else:
956-
base_url = operations.UPDATE_DESTINATION_SERVERS[
957-
operations.UPDATE_DESTINATION_SERVER_PLATFORM_API
958-
]
959911

960912
if not isinstance(request, BaseModel):
961913
request = utils.unmarshal(request, operations.UpdateDestinationRequest)
@@ -1065,18 +1017,11 @@ async def update_destination_async(
10651017
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
10661018
:param http_headers: Additional headers to set or replace on requests.
10671019
"""
1068-
base_url = None
1020+
base_url = server_url if server_url is not None else self.get_default_server_url()
10691021
url_variables = None
10701022
if timeout_ms is None:
10711023
timeout_ms = self.sdk_configuration.timeout_ms
10721024

1073-
if server_url is not None:
1074-
base_url = server_url
1075-
else:
1076-
base_url = operations.UPDATE_DESTINATION_SERVERS[
1077-
operations.UPDATE_DESTINATION_SERVER_PLATFORM_API
1078-
]
1079-
10801025
if not isinstance(request, BaseModel):
10811026
request = utils.unmarshal(request, operations.UpdateDestinationRequest)
10821027
request = cast(operations.UpdateDestinationRequest, request)

‎src/unstructured_client/general.py‎

Lines changed: 12 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from typing import Any, Dict, List, Mapping, Optional, Union, cast
66
from unstructured_client import utils
77
from unstructured_client._hooks import HookContext
8-
from unstructured_client._hooks.custom.clean_server_url_hook import choose_server_url
98
from unstructured_client.models import errors, operations, shared
109
from unstructured_client.types import BaseModel, OptionalNullable, UNSET
1110

@@ -16,6 +15,14 @@ class PartitionAcceptEnum(str, Enum):
1615

1716

1817
class General(BaseSDK):
18+
def get_default_server_url(self) -> str:
19+
client_url, *_ = self.sdk_configuration.get_server_details()
20+
if client_url is None:
21+
return operations.PARTITION_SERVERS[
22+
operations.PARTITION_SERVER_SAAS_API
23+
]
24+
return client_url
25+
1926
def partition(
2027
self,
2128
*,
@@ -39,21 +46,12 @@ def partition(
3946
:param accept_header_override: Override the default accept header for this method
4047
:param http_headers: Additional headers to set or replace on requests.
4148
"""
42-
base_url = None
49+
base_url = server_url if server_url is not None else self.get_default_server_url()
50+
4351
url_variables = None
4452
if timeout_ms is None:
4553
timeout_ms = self.sdk_configuration.timeout_ms
4654

47-
client_url, *_ = self.sdk_configuration.get_server_details()
48-
49-
base_url = choose_server_url(
50-
endpoint_url=server_url,
51-
client_url=client_url,
52-
default_endpoint_url=operations.PARTITION_SERVERS[
53-
operations.PARTITION_SERVER_SAAS_API
54-
]
55-
)
56-
5755
if not isinstance(request, BaseModel):
5856
request = utils.unmarshal(request, operations.PartitionRequest)
5957
request = cast(operations.PartitionRequest, request)
@@ -170,22 +168,12 @@ async def partition_async(
170168
:param accept_header_override: Override the default accept header for this method
171169
:param http_headers: Additional headers to set or replace on requests.
172170
"""
173-
base_url = None
171+
base_url = server_url if server_url is not None else self.get_default_server_url()
172+
174173
url_variables = None
175174
if timeout_ms is None:
176175
timeout_ms = self.sdk_configuration.timeout_ms
177176

178-
179-
client_url, *_ = self.sdk_configuration.get_server_details()
180-
181-
base_url = choose_server_url(
182-
endpoint_url=server_url,
183-
client_url=client_url,
184-
default_endpoint_url=operations.PARTITION_SERVERS[
185-
operations.PARTITION_SERVER_SAAS_API
186-
]
187-
)
188-
189177
if not isinstance(request, BaseModel):
190178
request = utils.unmarshal(request, operations.PartitionRequest)
191179
request = cast(operations.PartitionRequest, request)

0 commit comments

Comments
 (0)