TotalCloud-CSPM-Remediation-Report/separate_convert.py at main · Qualys/TotalCloud-CSPM-Remediation-Report · GitHub

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import json
import os.path
from airium import Airium
import datetime
import sys

def get_header(head):
    x = 0
    for i in head:
        if i.isupper():
            x = head.index(i)
            break
    return head[:x].capitalize() + " " +  head[x:].capitalize()


def dict_found(i, element, a):
    t = dict(i[element])
    with a.p(id=element):
        a.strong(_t=element.capitalize())
        with a.ul():
            for key in t.keys():
                # print(type(t[key]))
                if type(t[key]) == type([]):
                    with a.li():
                        a.strong(_t=f"{key.capitalize()} : ")
                        list_found(t[key], key, a)
                else:
                    with a.li():
                        a.strong(_t=key.capitalize())
                        a(f" : {i[element][key]}")


def list_found(i, element, a):
    with a.ul():
        for item in i:
            with a.li():
                a(f"{item}")


def getEvaluation(evaluation, a):
    with a.th():
        with a.ul():
            for i in evaluation:
                if evaluation[i] != []:
                    with a.li():
                        a(f"{i} : {evaluation[i]}")


def convert_to_html(CSP, timestamp):
    path = "Resposnes"
    file = open(f"{path}/{CSP}_response_{timestamp}.json")
    file_json = json.load(file)
    item = len(file_json['control'])

    # setup toolbar
    complete_percent = 0
    process = f"Generating Files {CSP}...  "
    sys.stdout.write(f"{process}")
    sys.stdout.flush()
    if os.path.exists(CSP):
        pass
    else:
        os.mkdir(CSP)
    if os.path.exists(f"{CSP}/CID_wise_Metadata"):
        pass
    else:
        os.mkdir(f"{CSP}/CID_wise_Metadata")


    for i in file_json['control']:
        with open(f"{CSP}/CID_wise_Metadata/{CSP}_CID_{i['cid']}_{timestamp}.html", "w") as f:
            a = Airium()
            a('<!DOCTYPE html>')
            with a.html(lang="en"):
                with a.head():
                    a.meta(charset="utf-8")
                    a.title(_t=f"{i['cid']} - {get_header(i['controlName'])}")
                    a.style(_t="""table {
                                  border: 1px solid black;
                                  align: left;
                                  text-align: left;
                                  width : 100%;
                              }
                              th, td {
                                  border: 1px solid black;
                                  align: left;
                                  text-align: left;
                              }""")

                with a.body():
                    with a.h4():
                        a(f"Report Generated Time : {datetime.datetime.now()}")
                    with a.h2():
                        a(f"Qualys CloudView Controls Metadata")
                    with a.h3():
                        a(f"CID - {i['cid']} : {i['controlName']}")
                    with a.table():
                        for element in i:
                            # if element in ["cid", "controlName", "criticality", "manualRemediation"] :
                            if element == "evaluation":
                                with a.tr():
                                    a.th(_t=get_header(element))
                                    getEvaluation(i[element], a)
                                continue
                            with a.tr():
                                a.th(_t=get_header(element))
                                a.th(_t=i[element])
                with a.h4():
                    a(f"Report Generated Time : {datetime.datetime.now()}")

            f.write(str(a))
        # update the bar
        if complete_percent != int(((file_json['control'].index(i) + 1) / item) * 100):
            sys.stdout.write("\b" * (len(str(complete_percent))+1))
            complete_percent = int(((file_json['control'].index(i) + 1) / item) * 100)
            sys.stdout.write(str(complete_percent) + "%")
            sys.stdout.flush()
    sys.stdout.write("\n")