@@ -24,18 +24,18 @@ def partition(self, request: Optional[shared.PartitionParameters], retries: Opti
2424 base_url = utils .template_url (* self .sdk_configuration .get_server_details ())
2525
2626 url = base_url + '/general/v0/general'
27- headers = {}
27+
28+ if callable (self .sdk_configuration .security ):
29+ headers , query_params = utils .get_security (self .sdk_configuration .security ())
30+ else :
31+ headers , query_params = utils .get_security (self .sdk_configuration .security )
32+
2833 req_content_type , data , form = utils .serialize_request_body (request , Optional [shared .PartitionParameters ], "request" , False , True , 'multipart' )
2934 if req_content_type is not None and req_content_type not in ('multipart/form-data' , 'multipart/mixed' ):
3035 headers ['content-type' ] = req_content_type
3136 headers ['Accept' ] = 'application/json'
3237 headers ['user-agent' ] = self .sdk_configuration .user_agent
33-
34- if callable (self .sdk_configuration .security ):
35- client = utils .configure_security_client (self .sdk_configuration .client , self .sdk_configuration .security ())
36- else :
37- client = utils .configure_security_client (self .sdk_configuration .client , self .sdk_configuration .security )
38-
38+ client = self .sdk_configuration .client
3939
4040 global_retry_config = self .sdk_configuration .retry_config
4141 retry_config = retries
@@ -45,11 +45,13 @@ def partition(self, request: Optional[shared.PartitionParameters], retries: Opti
4545 else :
4646 retry_config = utils .RetryConfig ('backoff' , utils .BackoffStrategy (500 , 60000 , 1.5 , 900000 ), True )
4747
48+ req = None
4849 def do_request ():
50+ nonlocal req
4951 try :
5052 req = self .sdk_configuration .get_hooks ().before_request (
5153 hook_ctx ,
52- requests_http .Request ('POST' , url , data = data , files = form , headers = headers ).prepare (),
54+ requests_http .Request ('POST' , url , params = query_params , data = data , files = form , headers = headers ).prepare (),
5355 )
5456 http_res = client .send (req )
5557 except Exception as e :
@@ -72,26 +74,28 @@ def do_request():
7274 '5xx'
7375 ]))
7476
75- content_type = http_res .headers .get ('Content-Type' )
7677
77- res = operations .PartitionResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
78+ res = operations .PartitionResponse (status_code = http_res .status_code , content_type = http_res . headers . get ( 'Content-Type' ) , raw_response = http_res )
7879
7980 if http_res .status_code == 200 :
80- if utils .match_content_type (content_type , 'application/json' ):
81+ if utils .match_content_type (http_res . headers . get ( 'Content-Type' ) , 'application/json' ):
8182 out = utils .unmarshal_json (http_res .text , Optional [List [Any ]])
8283 res .elements = out
8384 else :
85+ content_type = http_res .headers .get ('Content-Type' )
8486 raise errors .SDKError (f'unknown content-type received: { content_type } ' , http_res .status_code , http_res .text , http_res )
8587 elif http_res .status_code == 422 :
86- if utils .match_content_type (content_type , 'application/json' ):
88+ if utils .match_content_type (http_res . headers . get ( 'Content-Type' ) , 'application/json' ):
8789 out = utils .unmarshal_json (http_res .text , errors .HTTPValidationError )
88- out .raw_response = http_res
8990 raise out
9091 else :
92+ content_type = http_res .headers .get ('Content-Type' )
9193 raise errors .SDKError (f'unknown content-type received: { content_type } ' , http_res .status_code , http_res .text , http_res )
9294 elif http_res .status_code >= 400 and http_res .status_code < 500 or http_res .status_code >= 500 and http_res .status_code < 600 :
9395 raise errors .SDKError ('API error occurred' , http_res .status_code , http_res .text , http_res )
96+ else :
97+ raise errors .SDKError ('unknown status code received' , http_res .status_code , http_res .text , http_res )
9498
9599 return res
96100
97-
101+
0 commit comments