/
api_interact_new_group_destination.py
77 lines (69 loc) · 2.21 KB
/
api_interact_new_group_destination.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import requests
from requests.auth import HTTPBasicAuth
import json
import colorama
from colorama import Fore, Back, Style
#configuration file for key,secret,params,etc.
#r = 'config.json'
#with open(r, "r") as i:
# l = i.read()
# y = json.loads(l)
#api_key = y['API_KEY']
#api_secret = y['API_SECRET']
#a = HTTPBasicAuth(api_key, api_secret)
api_key = ''
api_secret = ''
a = HTTPBasicAuth(api_key, api_secret)
#new BQ destination + group
def atlas(method, endpoint, payload):
base_url = 'https://api.fivetran.com/v1'
h = {
'Authorization': f'Bearer {api_key}:{api_secret}'
}
url = f'{base_url}/{endpoint}'
try:
if method == 'GET':
response = requests.get(url, headers=h, auth=a)
elif method == 'POST':
response = requests.post(url, headers=h, json=payload, auth=a)
elif method == 'PATCH':
response = requests.patch(url, headers=h, json=payload, auth=a)
elif method == 'DELETE':
response = requests.delete(url, headers=h, auth=a)
else:
raise ValueError('Invalid request method.')
response.raise_for_status() # Raise exception for 4xx or 5xx responses
return response.json()
except requests.exceptions.RequestException as e:
print(f'Request failed: {e}')
return None
#Request
method = 'POST' #'PATCH' 'GET' 'DELETE'
endpoint = 'destinations/'
gendpoint = 'groups/'
gpayload = {
"name": ""
}
#Submit group
gresp = atlas(method, gendpoint, gpayload)
if gresp is not None:
print(Fore.CYAN + 'Call: ' + method + ' ' + gendpoint + ' ' + str(gpayload))
print(Fore.GREEN + 'Response: ' + gresp['code'])
print(Fore.MAGENTA + str(gresp))
payload = {
"group_id": gresp['data']['id'],
"service": "big_query",
"region": "US",
"time_zone_offset": "-5",
"config" :
{
"project_id": "",
"data_set_location": "US"
}
}
# #Submit destination
response = atlas(method, endpoint, payload)
if response is not None:
print(Fore.CYAN + 'Call: ' + method + ' ' + endpoint + ' ' + str(payload))
print(Fore.GREEN + 'Response: ' + response['code'])
print(Fore.MAGENTA + str(response))