-
Notifications
You must be signed in to change notification settings - Fork 0
/
data_manager.py
43 lines (36 loc) · 1.16 KB
/
data_manager.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
import requests
import os
from pprint import pprint
URL = os.environ.get("SH_URL")
USER_URL = os.environ.get("USER_URL")
TOKEN = {
"authorization": os.environ.get("SH_TOKEN")
}
class DataManager:
"""This class is responsible for talking to the Google Sheet."""
def __init__(self):
self.destination_data = {}
self.users_email = {}
def get_destination_data(self):
response = requests.get(url=URL, headers=TOKEN)
result = response.json()
self.destination_data = result["prices"]
return self.destination_data
def update_destination_codes(self):
for city in self.destination_data:
new_data = {
"price": {
"iataCode": city["iataCode"]
}
}
response = requests.put(
url=f"{URL}/{city['id']}",
json=new_data,
headers=TOKEN
)
print(response.text)
def get_email(self):
response = requests.get(url=USER_URL, headers=TOKEN)
result = response.json()
self.users_email = result["users"]
return self.users_email