1
1
"""Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT."""
2
2
3
- import requests as requests_http
4
- from . import utils
3
+ from . sdkconfiguration import SDKConfiguration
4
+ from airbyte import utils
5
5
from airbyte .models import operations , shared
6
6
from typing import Optional
7
7
8
8
class Connections :
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
15
10
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
23
13
24
14
25
15
def create_connection (self , request : shared .ConnectionCreateRequest ) -> operations .CreateConnectionResponse :
26
16
r"""Create a connection"""
27
- base_url = self ._server_url
17
+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
28
18
29
- url = base_url . removesuffix ( '/' ) + '/connections'
19
+ url = base_url + '/connections'
30
20
headers = {}
31
21
req_content_type , data , form = utils .serialize_request_body (request , "request" , 'json' )
32
22
if req_content_type not in ('multipart/form-data' , 'multipart/mixed' ):
33
23
headers ['content-type' ] = req_content_type
34
24
if data is None and form is None :
35
25
raise Exception ('request body is required' )
36
26
headers ['Accept' ] = 'application/json'
37
- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
27
+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
38
28
39
- client = self ._security_client
29
+ client = self .sdk_configuration . security_client
40
30
41
31
http_res = client .request ('POST' , url , data = data , files = form , headers = headers )
42
32
content_type = http_res .headers .get ('Content-Type' )
@@ -55,14 +45,14 @@ def create_connection(self, request: shared.ConnectionCreateRequest) -> operatio
55
45
56
46
def delete_connection (self , request : operations .DeleteConnectionRequest ) -> operations .DeleteConnectionResponse :
57
47
r"""Delete a Connection"""
58
- base_url = self ._server_url
48
+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
59
49
60
50
url = utils .generate_url (operations .DeleteConnectionRequest , base_url , '/connections/{connectionId}' , request )
61
51
headers = {}
62
52
headers ['Accept' ] = '*/*'
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 } '
64
54
65
- client = self ._security_client
55
+ client = self .sdk_configuration . security_client
66
56
67
57
http_res = client .request ('DELETE' , url , headers = headers )
68
58
content_type = http_res .headers .get ('Content-Type' )
@@ -75,14 +65,14 @@ def delete_connection(self, request: operations.DeleteConnectionRequest) -> oper
75
65
76
66
def get_connection (self , request : operations .GetConnectionRequest ) -> operations .GetConnectionResponse :
77
67
r"""Get Connection details"""
78
- base_url = self ._server_url
68
+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
79
69
80
70
url = utils .generate_url (operations .GetConnectionRequest , base_url , '/connections/{connectionId}' , request )
81
71
headers = {}
82
72
headers ['Accept' ] = 'application/json'
83
- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
73
+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
84
74
85
- client = self ._security_client
75
+ client = self .sdk_configuration . security_client
86
76
87
77
http_res = client .request ('GET' , url , headers = headers )
88
78
content_type = http_res .headers .get ('Content-Type' )
@@ -101,15 +91,15 @@ def get_connection(self, request: operations.GetConnectionRequest) -> operations
101
91
102
92
def list_connections (self , request : operations .ListConnectionsRequest ) -> operations .ListConnectionsResponse :
103
93
r"""List connections"""
104
- base_url = self ._server_url
94
+ base_url = utils . template_url ( * self .sdk_configuration . get_server_details ())
105
95
106
- url = base_url . removesuffix ( '/' ) + '/connections'
96
+ url = base_url + '/connections'
107
97
headers = {}
108
98
query_params = utils .get_query_params (operations .ListConnectionsRequest , request )
109
99
headers ['Accept' ] = 'application/json'
110
- headers ['user-agent' ] = f'speakeasy-sdk/{ self ._language } { self ._sdk_version } { self ._gen_version } '
100
+ headers ['user-agent' ] = f'speakeasy-sdk/{ self .sdk_configuration . language } { self .sdk_configuration . sdk_version } { self .sdk_configuration . gen_version } '
111
101
112
- client = self ._security_client
102
+ client = self .sdk_configuration . security_client
113
103
114
104
http_res = client .request ('GET' , url , params = query_params , headers = headers )
115
105
content_type = http_res .headers .get ('Content-Type' )
0 commit comments