mirror of
https://github.com/alerta/alerta.git
synced 2025-01-30 11:36:20 +00:00
291 lines
11 KiB
Python
291 lines
11 KiB
Python
|
|
import os
|
|
import sys
|
|
import time
|
|
import _strptime # FIXME: http://stackoverflow.com/questions/2427240
|
|
import datetime
|
|
import json
|
|
|
|
from uuid import uuid4
|
|
from email import utils
|
|
|
|
DEFAULT_SEVERITY = "normal" # "normal", "ok" or "clear"
|
|
DEFAULT_TIMEOUT = 86400
|
|
|
|
prog = os.path.basename(sys.argv[0])
|
|
|
|
|
|
class Alert(object):
|
|
|
|
def __init__(self, resource, event, **kwargs):
|
|
|
|
if not resource:
|
|
raise ValueError('Missing mandatory value for "resource"')
|
|
if not event:
|
|
raise ValueError('Missing mandatory value for "event"')
|
|
if any(['.' in key for key in kwargs.get('attributes', dict()).keys()])\
|
|
or any(['$' in key for key in kwargs.get('attributes', dict()).keys()]):
|
|
raise ValueError('Attribute keys must not contain "." or "$"')
|
|
|
|
self.id = kwargs.get('id', str(uuid4()))
|
|
self.resource = resource
|
|
self.event = event
|
|
self.environment = kwargs.get('environment', None) or ""
|
|
self.severity = kwargs.get('severity', None) or DEFAULT_SEVERITY
|
|
self.correlate = kwargs.get('correlate', None) or list()
|
|
if self.correlate and event not in self.correlate:
|
|
self.correlate.append(event)
|
|
self.status = kwargs.get('status', None) or "unknown"
|
|
self.service = kwargs.get('service', None) or list()
|
|
self.group = kwargs.get('group', None) or "Misc"
|
|
self.value = kwargs.get('value', None) or "n/a"
|
|
self.text = kwargs.get('text', None) or ""
|
|
self.tags = kwargs.get('tags', None) or list()
|
|
self.attributes = kwargs.get('attributes', None) or dict()
|
|
self.origin = kwargs.get('origin', None) or '%s/%s' % (prog, os.uname()[1])
|
|
self.event_type = kwargs.get('event_type', kwargs.get('type', None)) or "exceptionAlert"
|
|
self.create_time = kwargs.get('create_time', None) or datetime.datetime.utcnow()
|
|
self.receive_time = None
|
|
self.timeout = kwargs.get('timeout', None) or DEFAULT_TIMEOUT
|
|
self.raw_data = kwargs.get('raw_data', kwargs.get('rawData', None)) or ""
|
|
|
|
def get_id(self, short=False):
|
|
|
|
if short:
|
|
return self.id[:8]
|
|
else:
|
|
return self.id
|
|
|
|
def get_header(self):
|
|
|
|
return {
|
|
"origin": self.origin,
|
|
"type": self.event_type,
|
|
"correlation-id": self.id
|
|
}
|
|
|
|
def get_body(self):
|
|
|
|
return {
|
|
'id': self.id,
|
|
'resource': self.resource,
|
|
'event': self.event,
|
|
'environment': self.environment,
|
|
'severity': self.severity,
|
|
'correlate': self.correlate,
|
|
'status': self.status,
|
|
'service': self.service,
|
|
'group': self.group,
|
|
'value': self.value,
|
|
'text': self.text,
|
|
'tags': self.tags,
|
|
'attributes': self.attributes,
|
|
'origin': self.origin,
|
|
'type': self.event_type,
|
|
'createTime': self.create_time.replace(microsecond=0).isoformat() + ".%03dZ" % (self.create_time.microsecond // 1000),
|
|
'timeout': self.timeout,
|
|
'rawData': self.raw_data
|
|
}
|
|
|
|
def get_type(self):
|
|
return self.event_type
|
|
|
|
def receive_now(self):
|
|
self.receive_time = datetime.datetime.utcnow()
|
|
|
|
def __repr__(self):
|
|
return 'Alert(id=%r, environment=%r, resource=%r, event=%r, severity=%r, status=%r)' % (
|
|
self.id, self.environment, self.resource, self.event, self.severity, self.status)
|
|
|
|
def __str__(self):
|
|
return json.dumps(self.get_body(), indent=4)
|
|
|
|
@staticmethod
|
|
def parse_alert(alert):
|
|
|
|
try:
|
|
alert = json.loads(alert)
|
|
except ValueError, e:
|
|
raise ValueError('Could not parse alert - %s: %s' % (e, alert))
|
|
|
|
for k, v in alert.iteritems():
|
|
if k in ['createTime', 'receiveTime', 'lastReceiveTime', 'expireTime']:
|
|
try:
|
|
alert[k] = datetime.datetime.strptime(v, '%Y-%m-%dT%H:%M:%S.%fZ')
|
|
except ValueError, e:
|
|
raise ValueError('Could not parse date time string: %s' % e)
|
|
|
|
return Alert(
|
|
resource=alert.get('resource', None),
|
|
event=alert.get('event', None),
|
|
environment=alert.get('environment', None),
|
|
severity=alert.get('severity', None),
|
|
correlate=alert.get('correlate', list()),
|
|
status=alert.get('status', None),
|
|
service=alert.get('service', list()),
|
|
group=alert.get('group', None),
|
|
value=alert.get('value', None),
|
|
text=alert.get('text', None),
|
|
tags=alert.get('tags', list()),
|
|
attributes=alert.get('attributes', dict()),
|
|
origin=alert.get('origin', None),
|
|
event_type=alert.get('type', None),
|
|
create_time=alert.get('createTime', None),
|
|
timeout=alert.get('timeout', None),
|
|
raw_data=alert.get('rawData', None),
|
|
)
|
|
|
|
|
|
class AlertDocument(object):
|
|
|
|
def __init__(self, id, resource, event, environment, severity, correlate, status, service, group, value, text,
|
|
tags, attributes, origin, event_type, create_time, timeout, raw_data, duplicate_count, repeat,
|
|
previous_severity, trend_indication, receive_time, last_receive_id, last_receive_time, history):
|
|
|
|
self.id = id
|
|
self.resource = resource
|
|
self.event = event
|
|
self.environment = environment or ""
|
|
self.severity = severity
|
|
self.correlate = correlate or list()
|
|
self.status = status
|
|
self.service = service or list()
|
|
self.group = group or 'Misc'
|
|
self.value = value or 'n/a'
|
|
self.text = text or ""
|
|
self.tags = tags or list()
|
|
self.attributes = attributes or dict()
|
|
self.origin = origin or '%s/%s' % (prog, os.uname()[1])
|
|
self.event_type = event_type or 'exceptionAlert'
|
|
self.create_time = create_time or datetime.datetime.utcnow()
|
|
self.timeout = timeout or DEFAULT_TIMEOUT
|
|
self.raw_data = raw_data
|
|
|
|
self.duplicate_count = duplicate_count
|
|
self.repeat = repeat
|
|
self.previous_severity = previous_severity
|
|
self.trend_indication = trend_indication
|
|
self.receive_time = receive_time
|
|
self.last_receive_id = last_receive_id
|
|
self.last_receive_time = last_receive_time
|
|
self.history = history
|
|
|
|
def get_id(self, short=False):
|
|
|
|
if short:
|
|
return self.id[:8]
|
|
else:
|
|
return self.id
|
|
|
|
def get_header(self):
|
|
|
|
return {
|
|
"origin": self.origin,
|
|
"type": self.event_type,
|
|
"correlation-id": self.id
|
|
}
|
|
|
|
def get_date(self, attr, fmt='iso'):
|
|
|
|
if hasattr(self, attr):
|
|
if fmt == 'local':
|
|
return getattr(self, attr).strftime('%Y/%m/%d %H:%M:%S')
|
|
#return getattr(self, attr).astimezone(self.tz).strftime('%Y/%m/%d %H:%M:%S')
|
|
elif fmt == 'iso' or fmt == 'iso8601':
|
|
return getattr(self, attr).replace(microsecond=0).isoformat() + ".%03dZ" % (getattr(self, attr).microsecond // 1000)
|
|
elif fmt == 'rfc' or fmt == 'rfc2822':
|
|
return utils.formatdate(time.mktime(getattr(self, attr).timetuple()))
|
|
elif fmt == 'short':
|
|
return getattr(self, attr).strftime('%a %d %H:%M:%S')
|
|
#return getattr(self, attr).astimezone(self.tz).strftime('%a %d %H:%M:%S')
|
|
elif fmt == 'epoch':
|
|
return time.mktime(getattr(self, attr).timetuple())
|
|
elif fmt == 'raw':
|
|
return getattr(self, attr)
|
|
else:
|
|
raise ValueError("Unknown date format %s" % fmt)
|
|
else:
|
|
return ValueError("Attribute %s not a date" % attr)
|
|
|
|
def get_body(self):
|
|
|
|
return {
|
|
'id': self.id,
|
|
'resource': self.resource,
|
|
'event': self.event,
|
|
'environment': self.environment,
|
|
'severity': self.severity,
|
|
'correlate': self.correlate,
|
|
'status': self.status,
|
|
'service': self.service,
|
|
'group': self.group,
|
|
'value': self.value,
|
|
'text': self.text,
|
|
'tags': self.tags,
|
|
'attributes': self.attributes,
|
|
'origin': self.origin,
|
|
'type': self.event_type,
|
|
'createTime': self.create_time.replace(microsecond=0).isoformat() + ".%03dZ" % (self.create_time.microsecond // 1000),
|
|
'timeout': self.timeout,
|
|
'rawData': self.raw_data,
|
|
'duplicateCount': self.duplicate_count,
|
|
'repeat': self.repeat,
|
|
'previousSeverity': self.previous_severity,
|
|
'trendIndication': self.trend_indication,
|
|
'receiveTime': self.receive_time.replace(microsecond=0).isoformat() + ".%03dZ" % (self.receive_time.microsecond // 1000),
|
|
'lastReceiveId': self.last_receive_id,
|
|
'lastReceiveTime': self.last_receive_time.replace(microsecond=0).isoformat() + ".%03dZ" % (self.last_receive_time.microsecond // 1000),
|
|
'history': self.history
|
|
}
|
|
|
|
def __repr__(self):
|
|
return 'AlertDocument(id=%r, environment=%r, resource=%r, event=%r, severity=%r, status=%r)' % (
|
|
self.id, self.environment, self.resource, self.event, self.severity, self.status)
|
|
|
|
def __str__(self):
|
|
return json.dumps(self.get_body(), indent=4)
|
|
|
|
@staticmethod
|
|
def parse_alert(alert):
|
|
|
|
for k, v in alert.iteritems():
|
|
if k in ['createTime', 'receiveTime', 'lastReceiveTime', 'expireTime']:
|
|
if '.' in v:
|
|
try:
|
|
alert[k] = datetime.datetime.strptime(v, '%Y-%m-%dT%H:%M:%S.%fZ')
|
|
except ValueError, e:
|
|
raise ValueError('Could not parse date time string: %s' % e)
|
|
else:
|
|
try:
|
|
alert[k] = datetime.datetime.strptime(v, '%Y-%m-%dT%H:%M:%SZ') # if us = 000000
|
|
except ValueError, e:
|
|
raise ValueError('Could not parse date time string: %s' % e)
|
|
|
|
return AlertDocument(
|
|
id=alert.get('id', None),
|
|
resource=alert.get('resource', None),
|
|
event=alert.get('event', None),
|
|
environment=alert.get('environment', None),
|
|
severity=alert.get('severity', None),
|
|
correlate=alert.get('correlate', list()),
|
|
status=alert.get('status', None),
|
|
service=alert.get('service', list()),
|
|
group=alert.get('group', None),
|
|
value=alert.get('value', None),
|
|
text=alert.get('text', None),
|
|
tags=alert.get('tags', list()),
|
|
attributes=alert.get('attributes', dict()),
|
|
origin=alert.get('origin', None),
|
|
event_type=alert.get('type', None),
|
|
create_time=alert.get('createTime', None),
|
|
timeout=alert.get('timeout', None),
|
|
raw_data=alert.get('rawData', None),
|
|
duplicate_count=alert.get('duplicateCount', None),
|
|
repeat=alert.get('repeat', None),
|
|
previous_severity=alert.get('previousSeverity', None),
|
|
trend_indication=alert.get('trendIndication', None),
|
|
receive_time=alert.get('receiveTime', None),
|
|
last_receive_id=alert.get('lastReceiveId', None),
|
|
last_receive_time=alert.get('lastReceiveTime', None),
|
|
history=alert.get('history', None)
|
|
)
|