mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-19 23:59:28 +00:00
renamed the function name and json_reporter to json
This commit is contained in:
@@ -35,7 +35,7 @@ if config.log.lower() != "none":
|
||||
|
||||
from src.modules.report.plain import PlainReporter
|
||||
from src.modules.report.yaml import YAMLReporter
|
||||
from src.modules.report.json_reporter import JSONReporter
|
||||
from src.modules.report.json import JSONReporter
|
||||
|
||||
if config.report.lower() == "yaml":
|
||||
config.reporter = YAMLReporter()
|
||||
|
||||
@@ -22,7 +22,7 @@ class BaseReporter(object):
|
||||
services_lock.release()
|
||||
return services_data
|
||||
|
||||
def get_vulenrabilities(self):
|
||||
def get_vulnerabilities(self):
|
||||
vulnerabilities_lock.acquire()
|
||||
vulnerabilities_data = [{"location": "{}:{}".format(vuln.host, vuln.port) if vuln.host else "",
|
||||
"category": vuln.category.name,
|
||||
|
||||
13
src/modules/report/json.py
Normal file
13
src/modules/report/json.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import StringIO
|
||||
import json
|
||||
from base import BaseReporter
|
||||
|
||||
class JSONReporter(BaseReporter):
|
||||
def get_report(self):
|
||||
report = {
|
||||
"nodes": self.get_nodes(),
|
||||
"services": self.get_services(),
|
||||
"vulnerabilities": self.get_vulnerabilities(),
|
||||
"vulnerability_count": self.get_count()
|
||||
}
|
||||
return json.dumps(report)
|
||||
@@ -9,7 +9,7 @@ class YAMLReporter(BaseReporter):
|
||||
report = {
|
||||
"nodes": self.get_nodes(),
|
||||
"services": self.get_services(),
|
||||
"vulnerabilities": self.get_vulenrabilities()
|
||||
"vulnerabilities": self.get_vulnerabilities()
|
||||
}
|
||||
output = StringIO.StringIO()
|
||||
yaml.dump(report, output)
|
||||
|
||||
Reference in New Issue
Block a user