11"""Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT."""
22
3- import requests as requests_http
4- from . import utils
3+ from . sdkconfiguration import SDKConfiguration
4+ from airbyte import utils
55from airbyte .models import operations , shared
66from typing import Optional
77
88class Jobs :
9- _client : requests_http .Session
10- _security_client : requests_http .Session
11- _server_url : str
12- _language : str
13- _sdk_version : str
14- _gen_version : str
9+ sdk_configuration : SDKConfiguration
1510
16- def __init__ (self , client : requests_http .Session , security_client : requests_http .Session , server_url : str , language : str , sdk_version : str , gen_version : str ) -> None :
17- self ._client = client
18- self ._security_client = security_client
19- self ._server_url = server_url
20- self ._language = language
21- self ._sdk_version = sdk_version
22- self ._gen_version = gen_version
11+ def __init__ (self , sdk_config : SDKConfiguration ) -> None :
12+ self .sdk_configuration = sdk_config
2313
2414
2515 def cancel_job (self , request : operations .CancelJobRequest ) -> operations .CancelJobResponse :
2616 r"""Cancel a running Job"""
27- base_url = self ._server_url
17+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
2818
2919 url = utils .generate_url (operations .CancelJobRequest , base_url , '/jobs/{jobId}' , request )
3020 headers = {}
3121 headers ['Accept' ] = 'application/json'
32- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
22+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
3323
34- client = self ._security_client
24+ client = self .sdk_configuration . security_client
3525
3626 http_res = client .request ('DELETE' , url , headers = headers )
3727 content_type = http_res .headers .get ('Content-Type' )
@@ -50,19 +40,19 @@ def cancel_job(self, request: operations.CancelJobRequest) -> operations.CancelJ
5040
5141 def create_job (self , request : shared .JobCreateRequest ) -> operations .CreateJobResponse :
5242 r"""Trigger a sync or reset job of a connection"""
53- base_url = self ._server_url
43+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
5444
55- url = base_url . removesuffix ( '/' ) + '/jobs'
45+ url = base_url + '/jobs'
5646 headers = {}
5747 req_content_type , data , form = utils .serialize_request_body (request , "request" , 'json' )
5848 if req_content_type not in ('multipart/form-data' , 'multipart/mixed' ):
5949 headers ['content-type' ] = req_content_type
6050 if data is None and form is None :
6151 raise Exception ('request body is required' )
6252 headers ['Accept' ] = 'application/json'
63- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
53+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
6454
65- client = self ._security_client
55+ client = self .sdk_configuration . security_client
6656
6757 http_res = client .request ('POST' , url , data = data , files = form , headers = headers )
6858 content_type = http_res .headers .get ('Content-Type' )
@@ -81,14 +71,14 @@ def create_job(self, request: shared.JobCreateRequest) -> operations.CreateJobRe
8171
8272 def get_job (self , request : operations .GetJobRequest ) -> operations .GetJobResponse :
8373 r"""Get Job status and details"""
84- base_url = self ._server_url
74+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
8575
8676 url = utils .generate_url (operations .GetJobRequest , base_url , '/jobs/{jobId}' , request )
8777 headers = {}
8878 headers ['Accept' ] = 'application/json'
89- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
79+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
9080
91- client = self ._security_client
81+ client = self .sdk_configuration . security_client
9282
9383 http_res = client .request ('GET' , url , headers = headers )
9484 content_type = http_res .headers .get ('Content-Type' )
@@ -107,15 +97,15 @@ def get_job(self, request: operations.GetJobRequest) -> operations.GetJobRespons
10797
10898 def list_jobs (self , request : operations .ListJobsRequest ) -> operations .ListJobsResponse :
10999 r"""List Jobs by sync type"""
110- base_url = self ._server_url
100+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
111101
112- url = base_url . removesuffix ( '/' ) + '/jobs'
102+ url = base_url + '/jobs'
113103 headers = {}
114104 query_params = utils .get_query_params (operations .ListJobsRequest , request )
115105 headers ['Accept' ] = 'application/json'
116- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
106+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
117107
118- client = self ._security_client
108+ client = self .sdk_configuration . security_client
119109
120110 http_res = client .request ('GET' , url , params = query_params , headers = headers )
121111 content_type = http_res .headers .get ('Content-Type' )
0 commit comments