moved dict_tracker assignments into creation

This commit is contained in:
nate
2021-08-03 14:29:00 -05:00
parent 51234a569f
commit 742a645190

View File

@ -263,67 +263,70 @@ class qualysScanReport:
def grab_sections(self, report): def grab_sections(self, report):
all_dataframes = [] all_dataframes = []
dict_tracker = {} dict_tracker = {
dict_tracker['WEB_SCAN_VULN_BLOCK'] = pd.DataFrame( 'WEB_SCAN_VULN_BLOCK': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, report,
self.WEB_SCAN_VULN_BLOCK, self.WEB_SCAN_VULN_BLOCK,
end=[self.WEB_SCAN_SENSITIVE_BLOCK, self.WEB_SCAN_INFO_BLOCK], end=[self.WEB_SCAN_SENSITIVE_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( 'WEB_SCAN_SENSITIVE_BLOCK': pd.DataFrame(
self.utils.grab_section(report, self.utils.grab_section(report,
self.WEB_SCAN_SENSITIVE_BLOCK, self.WEB_SCAN_SENSITIVE_BLOCK,
end=[self.WEB_SCAN_INFO_BLOCK, self.WEB_SCAN_SENSITIVE_BLOCK], end=[self.WEB_SCAN_INFO_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( 'WEB_SCAN_INFO_BLOCK': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, 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(
'QID_HEADER': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, 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( 'GROUP_HEADER': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, 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( 'OWASP_HEADER': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, 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( 'WASC_HEADER': pd.DataFrame(
self.utils.grab_section( self.utils.grab_section(
report, report,
self.WASC_HEADER, self.WASC_HEADER,
end=[['APPENDIX']], end=[['APPENDIX']],
pop_last=True), pop_last=True),
columns=self.WASC_HEADER) columns=self.WASC_HEADER),
dict_tracker['SCAN_META'] = pd.DataFrame( 'SCAN_META': pd.DataFrame(
self.utils.grab_section(report, 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( 'CATEGORY_HEADER': pd.DataFrame(
self.utils.grab_section(report, 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
def data_normalizer(self, dataframes): def data_normalizer(self, dataframes):