diff --git a/server.py b/server.py index 6c8782f14879c06e36ef2876de9d1dc8c34529cd..2c5160c2c46c9b7244b1bdaa56b2bfd6e49efddf 100644 --- a/server.py +++ b/server.py @@ -15,16 +15,16 @@ class UffdAPI: def __init__(self, baseurl, key): self.baseurl = baseurl self.key = key + self.session = requests.Session() + self.session.headers['Authorization'] = 'Bearer '+self.key def get(self, endpoint, **kwargs): - resp = requests.get(self.baseurl + '/' + endpoint, params=kwargs, - headers={'Authorization': 'Bearer '+self.key}) + resp = self.session.get(self.baseurl + '/' + endpoint, params=kwargs) assert(resp.ok) return resp.json() def post(self, endpoint, **kwargs): - resp = requests.post(self.baseurl + '/' + endpoint, data=kwargs, - headers={'Authorization': 'Bearer '+self.key}) + resp = self.session.post(self.baseurl + '/' + endpoint, data=kwargs) assert(resp.ok) return resp.json()