@@ -22,7 +22,7 @@ def __init__(self, client: requests_http.Session, security_client: requests_http
2222 self ._sdk_version = sdk_version
2323 self ._gen_version = gen_version
2424
25- def create_data_connection (self , request : operations .CreateDataConnectionRequest ) -> operations .CreateDataConnectionResponse :
25+ def create_data_connection (self , request : operations .CreateDataConnectionRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .CreateDataConnectionResponse :
2626 r"""Create a data connection
2727 Create a data connection for a company
2828 """
@@ -37,7 +37,20 @@ def create_data_connection(self, request: operations.CreateDataConnectionRequest
3737
3838 client = self ._security_client
3939
40- http_res = client .request ('POST' , url , data = data , files = form , headers = headers )
40+ retry_config = retries
41+ if retry_config is None :
42+ retry_config = utils .RetryConfig ('backoff' , True )
43+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
44+
45+
46+ def do_request ():
47+ return client .request ('POST' , url , data = data , files = form , headers = headers )
48+
49+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
50+ '408' ,
51+ '429' ,
52+ '5XX'
53+ ]))
4154 content_type = http_res .headers .get ('Content-Type' )
4255
4356 res = operations .CreateDataConnectionResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
@@ -53,7 +66,7 @@ def create_data_connection(self, request: operations.CreateDataConnectionRequest
5366
5467 return res
5568
56- def delete_company_connection (self , request : operations .DeleteCompanyConnectionRequest ) -> operations .DeleteCompanyConnectionResponse :
69+ def delete_company_connection (self , request : operations .DeleteCompanyConnectionRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .DeleteCompanyConnectionResponse :
5770 r"""Delete connection
5871 Revoke and remove a connection from a company.
5972 This operation is not reversible - the end user would need to reauthorize a new data connection if you wish to view new data for this company.
@@ -65,7 +78,20 @@ def delete_company_connection(self, request: operations.DeleteCompanyConnectionR
6578
6679 client = self ._security_client
6780
68- http_res = client .request ('DELETE' , url )
81+ retry_config = retries
82+ if retry_config is None :
83+ retry_config = utils .RetryConfig ('backoff' , True )
84+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
85+
86+
87+ def do_request ():
88+ return client .request ('DELETE' , url )
89+
90+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
91+ '408' ,
92+ '429' ,
93+ '5XX'
94+ ]))
6995 content_type = http_res .headers .get ('Content-Type' )
7096
7197 res = operations .DeleteCompanyConnectionResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
@@ -79,7 +105,7 @@ def delete_company_connection(self, request: operations.DeleteCompanyConnectionR
79105
80106 return res
81107
82- def get_company_connection (self , request : operations .GetCompanyConnectionRequest ) -> operations .GetCompanyConnectionResponse :
108+ def get_company_connection (self , request : operations .GetCompanyConnectionRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .GetCompanyConnectionResponse :
83109 r"""Get connection
84110 Get a single connection for a company
85111 """
@@ -90,7 +116,20 @@ def get_company_connection(self, request: operations.GetCompanyConnectionRequest
90116
91117 client = self ._security_client
92118
93- http_res = client .request ('GET' , url )
119+ retry_config = retries
120+ if retry_config is None :
121+ retry_config = utils .RetryConfig ('backoff' , True )
122+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
123+
124+
125+ def do_request ():
126+ return client .request ('GET' , url )
127+
128+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
129+ '408' ,
130+ '429' ,
131+ '5XX'
132+ ]))
94133 content_type = http_res .headers .get ('Content-Type' )
95134
96135 res = operations .GetCompanyConnectionResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
@@ -106,7 +145,7 @@ def get_company_connection(self, request: operations.GetCompanyConnectionRequest
106145
107146 return res
108147
109- def list_company_connections (self , request : operations .ListCompanyConnectionsRequest ) -> operations .ListCompanyConnectionsResponse :
148+ def list_company_connections (self , request : operations .ListCompanyConnectionsRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .ListCompanyConnectionsResponse :
110149 r"""List connections
111150 List the connections for a company
112151 """
@@ -118,7 +157,20 @@ def list_company_connections(self, request: operations.ListCompanyConnectionsReq
118157
119158 client = self ._security_client
120159
121- http_res = client .request ('GET' , url , params = query_params )
160+ retry_config = retries
161+ if retry_config is None :
162+ retry_config = utils .RetryConfig ('backoff' , True )
163+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
164+
165+
166+ def do_request ():
167+ return client .request ('GET' , url , params = query_params )
168+
169+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
170+ '408' ,
171+ '429' ,
172+ '5XX'
173+ ]))
122174 content_type = http_res .headers .get ('Content-Type' )
123175
124176 res = operations .ListCompanyConnectionsResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
@@ -134,7 +186,7 @@ def list_company_connections(self, request: operations.ListCompanyConnectionsReq
134186
135187 return res
136188
137- def unlink_company_connection (self , request : operations .UnlinkCompanyConnectionRequest ) -> operations .UnlinkCompanyConnectionResponse :
189+ def unlink_company_connection (self , request : operations .UnlinkCompanyConnectionRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .UnlinkCompanyConnectionResponse :
138190 r"""Unlink connection
139191 This allows you to deauthorize a connection, without deleting it from Codat. This means you can still view any data that has previously been pulled into Codat, and also lets you re-authorize in future if your customer wishes to resume sharing their data.
140192 """
@@ -149,7 +201,20 @@ def unlink_company_connection(self, request: operations.UnlinkCompanyConnectionR
149201
150202 client = self ._security_client
151203
152- http_res = client .request ('PATCH' , url , data = data , files = form , headers = headers )
204+ retry_config = retries
205+ if retry_config is None :
206+ retry_config = utils .RetryConfig ('backoff' , True )
207+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
208+
209+
210+ def do_request ():
211+ return client .request ('PATCH' , url , data = data , files = form , headers = headers )
212+
213+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
214+ '408' ,
215+ '429' ,
216+ '5XX'
217+ ]))
153218 content_type = http_res .headers .get ('Content-Type' )
154219
155220 res = operations .UnlinkCompanyConnectionResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
@@ -165,7 +230,7 @@ def unlink_company_connection(self, request: operations.UnlinkCompanyConnectionR
165230
166231 return res
167232
168- def update_connection_authorization (self , request : operations .UpdateConnectionAuthorizationRequest ) -> operations .UpdateConnectionAuthorizationResponse :
233+ def update_connection_authorization (self , request : operations .UpdateConnectionAuthorizationRequest , retries : Optional [ utils . RetryConfig ] = None ) -> operations .UpdateConnectionAuthorizationResponse :
169234 r"""Update authorization
170235 Update data connection's authorization.
171236 """
@@ -180,7 +245,20 @@ def update_connection_authorization(self, request: operations.UpdateConnectionAu
180245
181246 client = self ._security_client
182247
183- http_res = client .request ('PUT' , url , data = data , files = form , headers = headers )
248+ retry_config = retries
249+ if retry_config is None :
250+ retry_config = utils .RetryConfig ('backoff' , True )
251+ retry_config .backoff = utils .BackoffStrategy (500 , 60000 , 1.5 , 3600000 )
252+
253+
254+ def do_request ():
255+ return client .request ('PUT' , url , data = data , files = form , headers = headers )
256+
257+ http_res = utils .retry (do_request , utils .Retries (retry_config , [
258+ '408' ,
259+ '429' ,
260+ '5XX'
261+ ]))
184262 content_type = http_res .headers .get ('Content-Type' )
185263
186264 res = operations .UpdateConnectionAuthorizationResponse (status_code = http_res .status_code , content_type = content_type , raw_response = http_res )
0 commit comments