diff --git a/bin/send_alert.py b/bin/send_alert.py index fce2e2c..e6fa612 100755 --- a/bin/send_alert.py +++ b/bin/send_alert.py @@ -10,19 +10,7 @@ import configuration import requests import argparse -if __name__ == '__main__': - - ap = argparse.ArgumentParser() - ap.add_argument("-t", "--title", required=True, default=None, help="The title of the incident.") - ap.add_argument("-d", "--description", required=False, default=None, help="A high-level summary of the problem.") - ap.add_argument("-s", "--service", required=False, default=None, help="The affected service.") - ap.add_argument("-l", "--severity", required=False, default="critical", help="The severity of the alert. Must be one of critical, high, medium, low, info, unknown. Default is critical.") - ap.add_argument("-I", "--stdin", required=False, default=None, help="Contents of standard input") - ap.add_argument("-O", "--stdout", required=False, default=None, help="Contents of standard output") - ap.add_argument("-E", "--stderr", required=False, default=None, help="Contents of standard error") - - - args = vars(ap.parse_args()) +def send_alert (data): cfg = configuration.read() @@ -40,6 +28,23 @@ if __name__ == '__main__': "Authorization": "Bearer " + authkey } + requests.post(url, headers=headers, json=data) + + +if __name__ == '__main__': + + ap = argparse.ArgumentParser() + ap.add_argument("-t", "--title", required=True, default=None, help="The title of the incident.") + ap.add_argument("-d", "--description", required=False, default=None, help="A high-level summary of the problem.") + ap.add_argument("-s", "--service", required=False, default=None, help="The affected service.") + ap.add_argument("-l", "--severity", required=False, default="critical", help="The severity of the alert. Must be one of critical, high, medium, low, info, unknown. Default is critical.") + ap.add_argument("-I", "--stdin", required=False, default=None, help="Contents of standard input") + ap.add_argument("-O", "--stdout", required=False, default=None, help="Contents of standard output") + ap.add_argument("-E", "--stderr", required=False, default=None, help="Contents of standard error") + + + args = vars(ap.parse_args()) + data = { "title": args["title"], "description": args["description"], @@ -51,6 +56,6 @@ if __name__ == '__main__': "error": args["stderr"] } - requests.post(url, headers=headers, json=data) + send_alert(data) print("Done")