-
Notifications
You must be signed in to change notification settings - Fork 0
/
clair-to-markdown.py
79 lines (66 loc) · 2.6 KB
/
clair-to-markdown.py
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from enum import Enum
from logging import critical
import os
import json
import pystache
class Severity(Enum):
No = 0
Negligible = 1
Low = 2
Medium = 3
High = 4
Critical = 5
Defcon1 = 6
# ignore for summary
Unknown = -1
def get_summary(json_report):
count = 0
neg_count = 0
low_count = 0
medium_count = 0
high_count = 0
critical_count = 0
defcon_count = 0
badge="[Summary](https://img.shields.io/badge/Severity-None-brightgreen)"
highest = Severity.Negligible
for vulnerability in json_report["vulnerabilities"]:
count = count+1
severity=Severity[vulnerability["severity"]]
if severity.value > highest.value :
highest = severity
if severity is Severity.Negligible:
neg_count = neg_count+1
if severity is Severity.Low:
low_count = low_count+1
if severity is Severity.Medium:
medium_count = medium_count+1
if severity is Severity.High:
high_count = high_count+1
if severity is Severity.Critical:
critical_count = critical_count+1
if severity is Severity.Defcon1:
defcon_count = defcon_count+1
if highest is Severity.Negligible:
badge=""
if highest is Severity.Low:
badge=""
if highest is Severity.Medium:
badge=""
if highest is Severity.High:
badge=""
if highest is Severity.Critical:
badge=""
if highest is Severity.Defcon1:
badge=""
return {'badge': badge, 'highest': highest.name, 'count': count, 'neg_count' : neg_count, 'low_count': low_count, 'medium_count': medium_count, 'high_count': high_count, 'critical_count': critical_count, 'defcon_count': defcon_count }
report_file = os.getenv("REPORT_FILE")
markdown_file = os.getenv("MARKDOWN_FILE")
mustache_file = os.getenv("MUSTACHE_TEMPLATE")
json_file = open(report_file)
json_report = json.load(json_file)
json_report["summary"] = get_summary(json_report)
mustache_template = open(mustache_file)
template_string = mustache_template.read()
rendered_report = pystache.render(template_string, json_report)
with open(markdown_file, "w") as markdown_report:
markdown_report.write(rendered_report)