renamed the function name and json_reporter to json

This commit is contained in:
maniish-jaiin
2019-02-21 11:17:06 +01:00
parent 4f9a362e6a
commit 9d68679df0
4 changed files with 16 additions and 3 deletions

View File

@@ -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()

View File

@@ -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,

View 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)

View File

@@ -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)