mirror of
https://github.com/alerta/alerta.git
synced 2025-01-30 03:33:59 +00:00
256 lines
8.1 KiB
Python
256 lines
8.1 KiB
Python
|
|
import sys
|
|
import json
|
|
import urllib
|
|
import requests
|
|
|
|
from alerta.common import log as logging
|
|
from alerta.common import config
|
|
from alerta.common.utils import DateEncoder
|
|
|
|
Version = '3.0.0'
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
CONF = config.CONF
|
|
|
|
|
|
class ApiClient(object):
|
|
|
|
api_opts = {
|
|
'api_host': 'localhost',
|
|
'api_port': 8080,
|
|
'api_root': '/api',
|
|
}
|
|
|
|
def __init__(self, host=None, port=None, root=None):
|
|
|
|
config.register_opts(ApiClient.api_opts)
|
|
|
|
self.host = host or CONF.api_host
|
|
self.port = port or CONF.api_port
|
|
self.root = root or CONF.api_root
|
|
|
|
def send(self, msg):
|
|
|
|
LOG.debug('header = %s', msg.get_header())
|
|
LOG.debug('message = %s', msg.get_body())
|
|
|
|
if msg.event_type.endswith('Alert'):
|
|
url = 'http://%s:%s%s/alert' % (self.host, self.port, self.root)
|
|
elif msg.event_type == 'Heartbeat':
|
|
url = 'http://%s:%s%s/heartbeat' % (self.host, self.port, self.root)
|
|
else:
|
|
LOG.error('Message type %s not supported by this API endpoint.', msg.event_type)
|
|
raise
|
|
|
|
payload = json.dumps(msg.get_body(), ensure_ascii=False, cls=DateEncoder)
|
|
headers = {'Content-Type': 'application/json'}
|
|
|
|
if CONF.dry_run:
|
|
print "curl -v '%s' -H 'Content-Type: application/json' -d '%s'" % (url, payload)
|
|
sys.exit(0)
|
|
|
|
LOG.debug('Sending alert to API endpoint...')
|
|
try:
|
|
r = requests.post(url, data=payload, headers=headers)
|
|
except requests.Timeout, e:
|
|
LOG.warning('API request timed out: %s', e)
|
|
raise
|
|
except requests.ConnectionError, e:
|
|
LOG.error('API request connection error: %s', e)
|
|
raise
|
|
except requests.TooManyRedirects, e:
|
|
LOG.error('Too many redirects: %s', e)
|
|
raise
|
|
except requests.RequestException, e:
|
|
LOG.error('API request send failed: %s', e)
|
|
raise
|
|
except requests.ConnectionError, e:
|
|
LOG.error('API request send failed: %s', e)
|
|
raise
|
|
|
|
LOG.debug('HTTP status: %s', r.status_code)
|
|
LOG.debug('HTTP response: %s', r.text)
|
|
|
|
try:
|
|
r.raise_for_status()
|
|
except requests.HTTPError, e:
|
|
LOG.error('API request HTTP error: %s', e)
|
|
raise
|
|
|
|
try:
|
|
response = r.json()
|
|
except Exception, e:
|
|
LOG.error('API bad response - %s: %s', e, r.text)
|
|
raise
|
|
|
|
if response.get('status', None) == 'ok':
|
|
LOG.debug('API request successful: %s', response)
|
|
return response['id']
|
|
elif response.get('status', None) == 'error' and 'message' in response:
|
|
LOG.error('API request failed: %s', response['message'])
|
|
raise
|
|
else:
|
|
LOG.error('API request unknown error: %s', response)
|
|
raise
|
|
|
|
def query(self, query=None):
|
|
|
|
LOG.debug('query = %s', query)
|
|
|
|
url = 'http://%s:%s%s/alerts' % (self.host, self.port, self.root)
|
|
|
|
if query:
|
|
url = "%s?%s" % (url, urllib.urlencode(query, doseq=1))
|
|
|
|
if CONF.dry_run:
|
|
print "curl -v '%s' -H 'Content-Type: application/json'" % url
|
|
sys.exit(0)
|
|
|
|
LOG.debug('Querying API endpoint...')
|
|
try:
|
|
r = requests.get(url)
|
|
except requests.Timeout, e:
|
|
LOG.warning('API request timed out: %s', e)
|
|
raise
|
|
except requests.ConnectionError, e:
|
|
LOG.error('API request connection error: %s', e)
|
|
raise
|
|
except requests.TooManyRedirects, e:
|
|
LOG.error('Too many redirects: %s', e)
|
|
raise
|
|
except requests.RequestException, e:
|
|
LOG.error('API request send failed: %s', e)
|
|
raise
|
|
|
|
LOG.debug('HTTP status: %s', r.status_code)
|
|
LOG.debug('HTTP response: %s', r.text)
|
|
|
|
try:
|
|
r.raise_for_status()
|
|
except requests.HTTPError, e:
|
|
LOG.error('API request HTTP error: %s', e)
|
|
raise
|
|
|
|
try:
|
|
response = r.json()
|
|
except Exception, e:
|
|
LOG.error('API bad response - %s: %s', e, r.text)
|
|
raise
|
|
|
|
if response.get('status', None) == 'ok':
|
|
LOG.debug('API request successful: %s', response)
|
|
return response
|
|
elif response.get('status', None) == 'error' and 'message' in response:
|
|
LOG.error('API request failed: %s', response['message'])
|
|
raise
|
|
else:
|
|
LOG.error('API request unknown error: %s', response)
|
|
raise
|
|
|
|
def ack(self, alertid):
|
|
|
|
LOG.debug('alertid = %s', alertid)
|
|
|
|
url = 'http://%s:%s%s/alert/%s/status' % (self.host, self.port, self.root, alertid)
|
|
|
|
data = {"status": "ack", "text": "ack via API"}
|
|
|
|
payload = json.dumps({"status": "ack", "text": "ack via API"})
|
|
headers = {'Content-Type': 'application/json'}
|
|
|
|
if CONF.dry_run:
|
|
print "curl -v -X PUT '%s' -H 'Content-Type: application/json' -d '%s'" % (url, data)
|
|
sys.exit(0)
|
|
|
|
LOG.debug('Acking via API endpoint...')
|
|
try:
|
|
r = requests.post(url, data=payload, headers=headers)
|
|
except requests.Timeout, e:
|
|
LOG.warning('API request timed out: %s', e)
|
|
raise
|
|
except requests.ConnectionError, e:
|
|
LOG.error('API request connection error: %s', e)
|
|
raise
|
|
except requests.TooManyRedirects, e:
|
|
LOG.error('Too many redirects: %s', e)
|
|
raise
|
|
except requests.RequestException, e:
|
|
LOG.error('API request send failed: %s', e)
|
|
raise
|
|
|
|
LOG.debug('HTTP status: %s', r.status_code)
|
|
LOG.debug('HTTP response: %s', r.text)
|
|
|
|
try:
|
|
r.raise_for_status()
|
|
except requests.HTTPError, e:
|
|
LOG.error('API request HTTP error: %s', e)
|
|
raise
|
|
|
|
try:
|
|
response = r.json()
|
|
except Exception, e:
|
|
LOG.error('API bad response - %s: %s', e, r.text)
|
|
raise
|
|
|
|
if response.get('status', None) == 'ok':
|
|
LOG.debug('API request successful: %s', response)
|
|
return response
|
|
elif response.get('status', None) == 'error' and 'message' in response:
|
|
LOG.error('API request failed: %s', response['message'])
|
|
raise
|
|
else:
|
|
LOG.error('API request unknown error: %s', response)
|
|
raise
|
|
|
|
def delete(self, alertid):
|
|
|
|
LOG.debug('alertid = %s', alertid)
|
|
|
|
url = 'http://%s:%s%s/alert/%s' % (self.host, self.port, self.root, alertid)
|
|
|
|
if CONF.dry_run:
|
|
print "curl -v -X DELETE '%s' -H 'Content-Type: application/json'" % url
|
|
sys.exit(0)
|
|
|
|
LOG.debug('Deleting via API endpoint...')
|
|
try:
|
|
r = requests.delete(url)
|
|
except requests.Timeout, e:
|
|
LOG.warning('API request timed out: %s', e)
|
|
raise
|
|
except requests.ConnectionError, e:
|
|
LOG.error('API request connection error: %s', e)
|
|
raise
|
|
except requests.TooManyRedirects, e:
|
|
LOG.error('Too many redirects: %s', e)
|
|
raise
|
|
except requests.RequestException, e:
|
|
LOG.error('API request send failed: %s', e)
|
|
raise
|
|
|
|
LOG.debug('HTTP status: %s', r.status_code)
|
|
LOG.debug('HTTP response: %s', r.text)
|
|
|
|
try:
|
|
r.raise_for_status()
|
|
except requests.HTTPError, e:
|
|
LOG.error('API request HTTP error: %s', e)
|
|
raise
|
|
|
|
try:
|
|
response = r.json()
|
|
except Exception, e:
|
|
LOG.error('API bad response - %s: %s', e, r.text)
|
|
raise
|
|
|
|
if response.get('status', None) == 'ok':
|
|
LOG.debug('API request successful: %s', response)
|
|
return response
|
|
elif response.get('status', None) == 'error' and 'message' in response:
|
|
LOG.error('API request failed: %s', response['message'])
|
|
raise
|
|
else:
|
|
LOG.error('API request unknown error: %s', response)
|
|
raise
|