distil/client/client.py
2014-02-21 14:15:37 +13:00

42 lines
1.5 KiB
Python

import requests
from requests.exceptions import ConnectionError
class Client(object):
def __init__(self, endpoint, **kwargs):
self.endpoint = endpoint
self.auth_token = kwargs.get('token')
def usage(self, tenants):
url = self.endpoint + "usage"
data = {"tenants": tenants}
try:
response = requests.post(url,
headers={"Content-Type":
"application/json",
"token": self.auth_token},
data=data)
if response.status_code != 200:
raise AttributeError("Usage cycle failed: " + response.text +
" code: " + str(response.status_code))
except ConnectionError:
pass
def sales_order(self, tenants):
url = self.endpoint + "sales_order"
data = {"tenants": tenants}
try:
response = requests.post(url,
headers={"Content-Type":
"application/json",
"token": self.auth_token},
data=data)
if response.status_code != 200:
raise AttributeError("Sales order cycle failed: " +
response.text + " code: " +
str(response.status_code))
except ConnectionError:
pass