77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
|
#!/usr/bin/python
|
||
|
|
import requests
|
||
|
|
|
||
|
|
class SensuApi:
|
||
|
|
def __init__(self, url, user, password):
|
||
|
|
self.url = url
|
||
|
|
self.user = user
|
||
|
|
self.password = password
|
||
|
|
self.headers = {}
|
||
|
|
|
||
|
|
def auth(self, user = None, password = None):
|
||
|
|
user = user or self.user
|
||
|
|
password = password or self.password
|
||
|
|
r = requests.get(
|
||
|
|
'{}/auth'.format(self.url),
|
||
|
|
auth=requests.auth.HTTPBasicAuth(user, password)
|
||
|
|
)
|
||
|
|
|
||
|
|
if r.status_code == 401:
|
||
|
|
raise Exception('Authentification has failed')
|
||
|
|
|
||
|
|
self.headers = { 'Authorization': r.json()['access_token'] }
|
||
|
|
|
||
|
|
def get(self, path):
|
||
|
|
r = requests.get(
|
||
|
|
'{}/api/core/v2/{}'.format(self.url, path),
|
||
|
|
headers=self.headers
|
||
|
|
)
|
||
|
|
|
||
|
|
if r.status_code == 500:
|
||
|
|
raise Exception('Server return 500 error: {}'.format(r.text))
|
||
|
|
elif r.status_code == 401:
|
||
|
|
raise Exception('Authentification has failed')
|
||
|
|
|
||
|
|
return r.status_code, r.json()
|
||
|
|
|
||
|
|
def post(self, path, data):
|
||
|
|
r = requests.post(
|
||
|
|
'{}/api/core/v2/{}'.format(self.url, path),
|
||
|
|
headers=self.headers,
|
||
|
|
json=data
|
||
|
|
)
|
||
|
|
|
||
|
|
if r.status_code == 500:
|
||
|
|
raise Exception('Server return 500 error: {}'.format(r.text))
|
||
|
|
elif r.status_code == 401:
|
||
|
|
raise Exception('Authentification has failed')
|
||
|
|
elif r.status_code != 200:
|
||
|
|
raise Exception('Server return an unknown error: {}'.format(r.text))
|
||
|
|
|
||
|
|
def put(self, path, data):
|
||
|
|
r = requests.put(
|
||
|
|
'{}/api/core/v2/{}'.format(self.url, path),
|
||
|
|
headers=self.headers,
|
||
|
|
json=data
|
||
|
|
)
|
||
|
|
|
||
|
|
if r.status_code == 500:
|
||
|
|
raise Exception('Server return 500 error: {}'.format(r.text))
|
||
|
|
elif r.status_code == 401:
|
||
|
|
raise Exception('Authentification has failed')
|
||
|
|
elif r.status_code != 200:
|
||
|
|
raise Exception('Server return an unknown error: {}'.format(r.text))
|
||
|
|
|
||
|
|
def delete(self, path):
|
||
|
|
r = requests.delete(
|
||
|
|
'{}/api/core/v2/{}'.format(self.url, path),
|
||
|
|
headers=self.headers
|
||
|
|
)
|
||
|
|
|
||
|
|
if r.status_code == 500:
|
||
|
|
raise Exception('Server return 500 error: {}'.format(r.text))
|
||
|
|
elif r.status_code == 401:
|
||
|
|
raise Exception('Authentification has failed')
|
||
|
|
elif r.status_code != 204:
|
||
|
|
raise Exception('Server return an unknown error: {}'.format(r.text))
|