Comparing sensitive data, confidential files or internal emails?

Most legal and privacy policies prohibit uploading sensitive data online. Diffchecker Desktop ensures your confidential information never leaves your computer. Work offline and compare documents securely.

Untitled diff

Created Diff never expires
46 removals
204 lines
37 additions
194 lines
#!/usr/bin/env python
#!/usr/bin/env python


import httplib
import httplib
try:
try:
import simplejson as json
import simplejson as json
except ImportError:
except ImportError:
import json
import json
import os
import os
import sys
import sys
from urlparse import urljoin
from urlparse import urljoin


try:
try:
import requests
import requests
except ImportError:
except ImportError:
raise ImportError('Missing dependency "requests". Do ``pip install requests``.')
raise ImportError('Missing dependency requests. Do ``pip install requests``.')


try:
try:
import yaml
import yaml
except ImportError:
except ImportError:
raise ImportError('Missing dependency "pyyaml". Do ``pip install pyyaml``.')
raise ImportError('Missing dependency pyyaml. Do ``pip install pyyaml``.')


# ST2 configuration
# ST2 configuration
ST2_CONFIG_FILE = './config.yaml'


ST2_API_BASE_URL = 'http://localhost:9101/v1'
ST2_API_BASE_URL = 'http://localhost:9101/v1'
ST2_AUTH_BASE_URL = 'http://localhost:9100'
ST2_AUTH_BASE_URL = 'http://localhost:9100'
ST2_USERNAME = None
ST2_USERNAME = None
ST2_PASSWORD = None
ST2_PASSWORD = None
ST2_AUTH_TOKEN = None
ST2_AUTH_TOKEN = None


ST2_AUTH_PATH = 'tokens'
ST2_AUTH_PATH = 'tokens'
ST2_WEBHOOKS_PATH = 'webhooks/st2/'
ST2_WEBHOOKS_PATH = 'webhooks/st2/'
ST2_TRIGGERS_PATH = 'triggertypes/'
ST2_TRIGGERS_PATH = 'triggertypes/'
ST2_TRIGGERTYPE_PACK = 'e_nagios'
ST2_TRIGGERTYPE_PACK = 'nagios'
ST2_TRIGGERTYPE_NAME = 'service_state_change'
ST2_TRIGGERTYPE_NAME = 'service-state-change'
ST2_TRIGGERTYPE_REF = '.'.join([ST2_TRIGGERTYPE_PACK, ST2_TRIGGERTYPE_NAME])
ST2_TRIGGERTYPE_REF = '.'.join([ST2_TRIGGERTYPE_PACK, ST2_TRIGGERTYPE_NAME])


STATE_MESSAGE = {
'OK': 'All is well on the Western front.',
'WARNING': 'We gots a warning yo!',
'UNKNOWN': 'It be unknown...',
'CRITICAL': 'Critical!'
}

REGISTERED_WITH_ST2 = False
REGISTERED_WITH_ST2 = False


OK_CODES = [httplib.OK, httplib.CREATED, httplib.ACCEPTED, httplib.CONFLICT]
OK_CODES = [httplib.OK, httplib.CREATED, httplib.ACCEPTED, httplib.CONFLICT]


STATE_MESSAGE = {
'OK': '[OK] Service/Host is back to normal.',
'WARNING': '[WARNING] Service/Host warning alert!',
'UNKNOWN': '[UNKNOWN] Service/Host is in the unknown state.',
'CRITICAL': '[CRITICAL] Pay attention immediately!'
}


def _create_trigger_type():
def _create_trigger_type():
try:
try:
url = _get_st2_triggers_url()
url = _get_st2_triggers_url()
payload = {
payload = {
'name': ST2_TRIGGERTYPE_NAME,
'name': ST2_TRIGGERTYPE_NAME,
'pack': ST2_TRIGGERTYPE_PACK,
'pack': ST2_TRIGGERTYPE_PACK,
'description': 'Trigger type for nagios event handler.'
'description': 'Trigger type for nagios event handler.'
}
}
# sys.stdout.write('POST: %s: Body: %s\n' % (url, payload))
# sys.stdout.write('POST: %s: Body: %s\n' % (url, payload))
headers = {}
headers = {}
headers['Content-Type'] = 'application/json; charset=utf-8'
headers['Content-Type'] = 'application/json; charset=utf-8'


if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
headers['X-Auth-Token'] = ST2_AUTH_TOKEN


post_resp = requests.post(url, data=json.dumps(payload), headers=headers, verify=False)
post_resp = requests.post(url, data=json.dumps(payload), headers=headers)
except:
except:
sys.stderr.write('Unable to register trigger type with st2.')
sys.stderr.write('Unable to register trigger type with st2.')
raise
raise
else:
else:
status = post_resp.status_code
status = post_resp.status_code
if status not in OK_CODES:
if status not in OK_CODES:
sys.stderr.write('Failed to register trigger type with st2. HTTP_CODE: %d\n' %
sys.stderr.write('Failed to register trigger type with st2. HTTP_CODE: %d\n' %
status)
status)
raise
raise
else:
else:
sys.stdout.write('Registered trigger type with st2.\n')
sys.stdout.write('Registered trigger type with st2.\n')




def _get_auth_url():
def _get_auth_url():
return urljoin(ST2_AUTH_BASE_URL, ST2_AUTH_PATH)
return urljoin(ST2_AUTH_BASE_URL, ST2_AUTH_PATH)




def _get_auth_token():
def _get_auth_token():
global ST2_AUTH_TOKEN
global ST2_AUTH_TOKEN
auth_url = _get_auth_url()
auth_url = _get_auth_url()
try:
try:
resp = requests.post(auth_url, json.dumps({'ttl': 5 * 60}),
resp = requests.post(auth_url, json.dumps({'ttl': 5 * 60}),
auth=(ST2_USERNAME, ST2_PASSWORD), verify=False)
auth=(ST2_USERNAME, ST2_PASSWORD))
except:
except:
sys.stderr.write('Cannot get auth token from st2. Trying unauthed.')
raise Exception('Cannot get auth token from st2. Will try unauthed.')
else:
else:
if resp.status_code not in OK_CODES:
raise Exception("Cannot authenticate: %s\n" % resp.text)

ST2_AUTH_TOKEN = resp.json()['token']
ST2_AUTH_TOKEN = resp.json()['token']




def _register_with_st2():
def _register_with_st2():
global REGISTERED_WITH_ST2
global REGISTERED_WITH_ST2
try:
try:
url = urljoin(_get_st2_triggers_url(), ST2_TRIGGERTYPE_REF)
url = urljoin(_get_st2_triggers_url(), ST2_TRIGGERTYPE_REF)
# sys.stdout.write('GET: %s\n' % url)
# sys.stdout.write('GET: %s\n' % url)
if not ST2_AUTH_TOKEN:
if not ST2_AUTH_TOKEN:
_get_auth_token()
_get_auth_token()


if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
get_resp = requests.get(url, headers={'X-Auth-Token': ST2_AUTH_TOKEN}, verify=False)
get_resp = requests.get(url, headers={'X-Auth-Token': ST2_AUTH_TOKEN})
else:
else:
get_resp = requests.get(url, verify=False)
get_resp = requests.get(url)


if get_resp.status_code != httplib.OK:
if get_resp.status_code != httplib.OK:
_create_trigger_type()
_create_trigger_type()
else:
else:
body = json.loads(get_resp.text)
body = json.loads(get_resp.text)
if len(body) == 0:
if len(body) == 0:
_create_trigger_type()
_create_trigger_type()
except:
except:
raise
raise
else:
else:
REGISTERED_WITH_ST2 = True
REGISTERED_WITH_ST2 = True




def _get_st2_triggers_url():
def _get_st2_triggers_url():
url = urljoin(ST2_API_BASE_URL, ST2_TRIGGERS_PATH)
url = urljoin(ST2_API_BASE_URL, ST2_TRIGGERS_PATH)
return url
return url




def _get_st2_webhooks_url():
def _get_st2_webhooks_url():
url = urljoin(ST2_API_BASE_URL, ST2_WEBHOOKS_PATH)
url = urljoin(ST2_API_BASE_URL, ST2_WEBHOOKS_PATH)
return url
return url




def _post_event_to_st2(url, body):
def _post_event_to_st2(url, body):
headers = {}
headers = {}
headers['X-ST2-Integration'] = 'nagios.'
headers['X-ST2-Integration'] = 'nagios.'
headers['Content-Type'] = 'application/json; charset=utf-8'
headers['Content-Type'] = 'application/json; charset=utf-8'
if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
try:
try:
sys.stdout.write('POST: url: %s, body: %s\n' % (url, body))
# sys.stdout.write('POST: url: %s, body: %s\n' % (url, body))
r = requests.post(url, data=json.dumps(body), headers=headers, verify=False)
r = requests.post(url, data=json.dumps(body), headers=headers)
except:
except:
sys.stderr.write('Cannot connect to st2 endpoint.')
sys.stderr.write('Cannot connect to st2 endpoint.')
else:
else:
status = r.status_code
status = r.status_code
if status not in OK_CODES:
if status not in OK_CODES:
sys.stderr.write('Failed posting nagios event to st2. HTTP_CODE: %d\n' % status)
sys.stderr.write('Failed posting nagios event to st2. HTTP_CODE: %d\n' % status)
else:
else:
sys.stdout.write('Sent nagios event to st2. HTTP_CODE: %d\n' % status)
sys.stdout.write('Sent nagios event to st2. HTTP_CODE: %d\n' % status)


def _get_payload(host, service, event_id, state, state_id, state_type, attempt):

def _get_payload(host, service, event_id, state, state_type, attempt):
payload = {}
payload = {}
payload['host'] = host
payload['host'] = host
payload['service'] = service
payload['service'] = service
payload['event_id'] = event_id
payload['event_id'] = event_id
payload['state'] = state
payload['state'] = state
payload['state_id'] = state_id
payload['state_type'] = state_type
payload['state_type'] = state_type
payload['attempt'] = attempt
payload['attempt'] = attempt
payload['msg'] = STATE_MESSAGE.get(state, 'Undefined state.')
payload['msg'] = STATE_MESSAGE.get(state, 'Undefined state.')
return payload
return payload



def main(args):
def main(args):
event_id = args[1]
event_id = args[1]
service = args[2]
service = args[2]
state = args[3]
state = args[3]
state_id = args[4]
state_type = args[4]
state_type = args[5]
attempt = args[5]
attempt = args[6]
host = args[6]
host = args[7]

payload = _get_payload(host, service, event_id, state, state_type, attempt)
body = {}
body = {}
body['trigger'] = ST2_TRIGGERTYPE_REF
body['trigger'] = ST2_TRIGGERTYPE_REF
body['payload'] = _get_payload(host, service, event_id, state, state_id, state_type, attempt)
body['payload'] = payload
_post_event_to_st2(_get_st2_webhooks_url(), body)
_post_event_to_st2(_get_st2_webhooks_url(), body)



if __name__ == '__main__':
if __name__ == '__main__':
if len(sys.argv) > 1:
st2_config_file = sys.argv[1]
sys.argv.pop(1)
else:
sys.stderr.write('Error: config file missing.\n')
sys.stderr.write('Usage: %s ST2_CONFIG_FILE\n' % sys.argv[0])
sys.exit(2)

try:
try:
if not os.path.exists(st2_config_file):
if not os.path.exists(ST2_CONFIG_FILE):
sys.stderr.write('Configuration file not found. Exiting.\n')
sys.stderr.write('Configuration file not found. Exiting.\n')
sys.exit(1)
sys.exit(1)


with open(st2_config_file) as f:
with open(ST2_CONFIG_FILE) as f:
config = yaml.safe_load(f)
config = yaml.safe_load(f)
ST2_USERNAME = config['st2_username']
ST2_USERNAME = config['st2_username']
ST2_PASSWORD = config['st2_password']
ST2_PASSWORD = config['st2_password']
ST2_API_BASE_URL = config['st2_api_base_url']
ST2_API_BASE_URL = config['st2_api_base_url']
ST2_AUTH_BASE_URL = config['st2_auth_base_url']
ST2_AUTH_BASE_URL = config['st2_auth_base_url']


if not REGISTERED_WITH_ST2:
if not REGISTERED_WITH_ST2:
_register_with_st2()
_register_with_st2()
except Exception as e:
except:
sys.stderr.write(
sys.stderr.write('Failed registering with st2. Won\'t post event.\n')
'Failed registering with st2. Won\'t post event.\n%s' % e)
# import traceback
# traceback.print_exc()
else:
else:
main(sys.argv)
main(sys.argv)