Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

removed useless open

Browse Source
This commit is contained in:
nate
2021-08-03 13:01:22 -05:00
parent e25141261c
commit 183e3b3e72
1 changed files with 48 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

98
vulnwhisp/frameworks/qualys_web.py
View File

@ -266,7 +266,7 @@ class qualysUtils:
): ):
temp_list = [] temp_list = []
max_col_count = 0 max_col_count = 0
with open(report, 'rb') as csvfile: with open(report, 'rt') as csvfile:
q_report = csv.reader(csvfile, delimiter=',', quotechar='"') q_report = csv.reader(csvfile, delimiter=',', quotechar='"')
for line in q_report: for line in q_report:
if set(line) == set(section): if set(line) == set(section):
@ -345,7 +345,6 @@ class qualysScanReport:
if file_stream: if file_stream:
self.open_file = file_in.splitlines() self.open_file = file_in.splitlines()
elif file_in: elif file_in:
self.open_file = open(file_in, 'rb') self.open_file = open(file_in, 'rb')
self.downloaded_file = None self.downloaded_file = None
@ -353,56 +352,55 @@ class qualysScanReport:
def grab_sections(self, report): def grab_sections(self, report):
all_dataframes = [] all_dataframes = []
dict_tracker = {} dict_tracker = {}
with open(report, 'rb') as csvfile: dict_tracker['WEB_SCAN_VULN_BLOCK'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['WEB_SCAN_VULN_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, self.WEB_SCAN_VULN_BLOCK,
self.WEB_SCAN_VULN_BLOCK, end=[
end=[ self.WEB_SCAN_SENSITIVE_BLOCK,
self.WEB_SCAN_SENSITIVE_BLOCK, self.WEB_SCAN_INFO_BLOCK],
self.WEB_SCAN_INFO_BLOCK], pop_last=True),
pop_last=True), columns=self.WEB_SCAN_VULN_HEADER)
columns=self.WEB_SCAN_VULN_HEADER) dict_tracker['WEB_SCAN_SENSITIVE_BLOCK'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['WEB_SCAN_SENSITIVE_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, self.WEB_SCAN_SENSITIVE_BLOCK,
self.WEB_SCAN_SENSITIVE_BLOCK, end=[
end=[ self.WEB_SCAN_INFO_BLOCK,
self.WEB_SCAN_INFO_BLOCK, self.WEB_SCAN_SENSITIVE_BLOCK],
self.WEB_SCAN_SENSITIVE_BLOCK], pop_last=True),
pop_last=True), columns=self.WEB_SCAN_SENSITIVE_HEADER)
columns=self.WEB_SCAN_SENSITIVE_HEADER) dict_tracker['WEB_SCAN_INFO_BLOCK'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['WEB_SCAN_INFO_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, self.WEB_SCAN_INFO_BLOCK,
self.WEB_SCAN_INFO_BLOCK, end=[self.QID_HEADER],
end=[self.QID_HEADER], pop_last=True),
pop_last=True), columns=self.WEB_SCAN_INFO_HEADER)
columns=self.WEB_SCAN_INFO_HEADER) dict_tracker['QID_HEADER'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['QID_HEADER'] = pd.DataFrame(self.utils.grab_section(report, self.QID_HEADER,
self.QID_HEADER, end=[self.GROUP_HEADER],
end=[self.GROUP_HEADER], pop_last=True),
pop_last=True), columns=self.QID_HEADER)
columns=self.QID_HEADER) dict_tracker['GROUP_HEADER'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['GROUP_HEADER'] = pd.DataFrame(self.utils.grab_section(report, self.GROUP_HEADER,
self.GROUP_HEADER, end=[self.OWASP_HEADER],
end=[self.OWASP_HEADER], pop_last=True),
pop_last=True), columns=self.GROUP_HEADER)
columns=self.GROUP_HEADER) dict_tracker['OWASP_HEADER'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['OWASP_HEADER'] = pd.DataFrame(self.utils.grab_section(report, self.OWASP_HEADER,
self.OWASP_HEADER, end=[self.WASC_HEADER],
end=[self.WASC_HEADER], pop_last=True),
pop_last=True), columns=self.OWASP_HEADER)
columns=self.OWASP_HEADER) dict_tracker['WASC_HEADER'] = pd.DataFrame(self.utils.grab_section(report,
dict_tracker['WASC_HEADER'] = pd.DataFrame(self.utils.grab_section(report, self.WASC_HEADER, end=[['APPENDIX']],
self.WASC_HEADER, end=[['APPENDIX']], pop_last=True),
pop_last=True), columns=self.WASC_HEADER)
columns=self.WASC_HEADER)
dict_tracker['SCAN_META'] = pd.DataFrame(self.utils.grab_section(report, dict_tracker['SCAN_META'] = pd.DataFrame(self.utils.grab_section(report,
self.SCAN_META, self.SCAN_META,
end=[self.CATEGORY_HEADER], end=[self.CATEGORY_HEADER],
pop_last=True), pop_last=True),
columns=self.SCAN_META) columns=self.SCAN_META)
dict_tracker['CATEGORY_HEADER'] = pd.DataFrame(self.utils.grab_section(report, dict_tracker['CATEGORY_HEADER'] = pd.DataFrame(self.utils.grab_section(report,
self.CATEGORY_HEADER), self.CATEGORY_HEADER),
columns=self.CATEGORY_HEADER) columns=self.CATEGORY_HEADER)
all_dataframes.append(dict_tracker) all_dataframes.append(dict_tracker)
return all_dataframes return all_dataframes