1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# -*- coding: utf-8 -*-
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
#
# Author: Mauro Soria
import json
import time
import sys
from lib.reports.base import FileBaseReport
class JSONReport(FileBaseReport):
def generate(self):
report = {"info": {"args": ' '.join(sys.argv), "time": time.ctime()}, "results": []}
for entry in self.entries:
result = {}
header_name = "{0}://{1}:{2}/{3}".format(
entry.protocol, entry.host, entry.port, entry.base_path
)
result[header_name] = []
for e in entry.results:
path_entry = {
"status": e.status,
"path": "/" + e.path,
"content-length": e.get_content_length(),
"redirect": e.response.redirect,
}
result[header_name].append(path_entry)
report["results"].append(result)
return json.dumps(report, sort_keys=True, indent=4)
def save(self):
self.file.seek(0)
self.file.truncate(0)
self.file.flush()
self.file.writelines(self.generate())
self.file.flush()
|
