From 03760724acfaa53593325906a0be423e2f6bb30b Mon Sep 17 00:00:00 2001 From: daniel_sagi Date: Wed, 27 Jun 2018 14:13:49 +0300 Subject: [PATCH] + added report url handling on end of run, when specifiyng token --- report/reporter.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/report/reporter.py b/report/reporter.py index d0abd93..1f43aab 100644 --- a/report/reporter.py +++ b/report/reporter.py @@ -21,7 +21,8 @@ EVIDENCE_PREVIEW = 40 MAX_WIDTH_VULNS = 70 MAX_WIDTH_SERVICES = 60 -AQUA_PUSH_URL = "https://qlyscbqwl7.execute-api.us-east-1.amazonaws.com/Prod?token={}" +AQUA_PUSH_URL = "https://qlyscbqwl7.execute-api.us-east-1.amazonaws.com/Prod/submit?token={token}" +AQUA_RESULTS_URL = "https://qlyscbqwl7.execute-api.us-east-1.amazonaws.com/Prod/result?token={token}" @handler.subscribe(Service) @handler.subscribe(Vulnerability) @@ -105,6 +106,7 @@ class Reporter(object): return current_list def send_report(self, token): + logging.debug("generating report") def generate_report(): """function generates a report corresponding to specifications of the frontend of kubehunter""" for service in services: @@ -129,9 +131,14 @@ class Reporter(object): 'results': generate_report(), 'metadata': {} } - r = requests.put(AQUA_PUSH_URL.format(token), json=report) + logging.debug("uploading report") + r = requests.put(AQUA_PUSH_URL.format(token=token), json=report) - print "{}: {}".format(r.status_code, r.text) + if r.status_code == 201: # created status + print "\nYour report: \n{}".format(AQUA_RESULTS_URL.format(token=token)) + else: + logging.debug("Failed sending report with:{}, {}".format(r.status_code, r.text)) + print "\nSomething went wrong.\nPlease try hunting again." reporter = Reporter()