diff --git a/vulnwhisp/frameworks/qualys_web.py b/vulnwhisp/frameworks/qualys_web.py index 71a4f55..7c54bfb 100644 --- a/vulnwhisp/frameworks/qualys_web.py +++ b/vulnwhisp/frameworks/qualys_web.py @@ -353,54 +353,64 @@ class qualysScanReport: def grab_sections(self, report): all_dataframes = [] dict_tracker = {} - dict_tracker['WEB_SCAN_VULN_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, - self.WEB_SCAN_VULN_BLOCK, - end=[ - self.WEB_SCAN_SENSITIVE_BLOCK, - self.WEB_SCAN_INFO_BLOCK], - pop_last=True), - columns=self.WEB_SCAN_VULN_HEADER) - dict_tracker['WEB_SCAN_SENSITIVE_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, - self.WEB_SCAN_SENSITIVE_BLOCK, - end=[ - self.WEB_SCAN_INFO_BLOCK, - self.WEB_SCAN_SENSITIVE_BLOCK], - pop_last=True), - columns=self.WEB_SCAN_SENSITIVE_HEADER) - dict_tracker['WEB_SCAN_INFO_BLOCK'] = pd.DataFrame(self.utils.grab_section(report, - self.WEB_SCAN_INFO_BLOCK, - end=[self.QID_HEADER], - pop_last=True), - columns=self.WEB_SCAN_INFO_HEADER) - dict_tracker['QID_HEADER'] = pd.DataFrame(self.utils.grab_section(report, - self.QID_HEADER, - end=[self.GROUP_HEADER], - pop_last=True), - columns=self.QID_HEADER) - dict_tracker['GROUP_HEADER'] = pd.DataFrame(self.utils.grab_section(report, - self.GROUP_HEADER, - end=[self.OWASP_HEADER], - pop_last=True), - columns=self.GROUP_HEADER) - dict_tracker['OWASP_HEADER'] = pd.DataFrame(self.utils.grab_section(report, - self.OWASP_HEADER, - end=[self.WASC_HEADER], - pop_last=True), - columns=self.OWASP_HEADER) - dict_tracker['WASC_HEADER'] = pd.DataFrame(self.utils.grab_section(report, - self.WASC_HEADER, end=[['APPENDIX']], - pop_last=True), - columns=self.WASC_HEADER) - - dict_tracker['SCAN_META'] = pd.DataFrame(self.utils.grab_section(report, - self.SCAN_META, - end=[self.CATEGORY_HEADER], - pop_last=True), - columns=self.SCAN_META) - - dict_tracker['CATEGORY_HEADER'] = pd.DataFrame(self.utils.grab_section(report, - self.CATEGORY_HEADER), - columns=self.CATEGORY_HEADER) + dict_tracker['WEB_SCAN_VULN_BLOCK'] = pd.DataFrame( + self.utils.grab_section( + report, + self.WEB_SCAN_VULN_BLOCK, + end=[self.WEB_SCAN_SENSITIVE_BLOCK, self.WEB_SCAN_INFO_BLOCK], + pop_last=True), + columns=self.WEB_SCAN_VULN_HEADER) + dict_tracker['WEB_SCAN_SENSITIVE_BLOCK'] = pd.DataFrame( + self.utils.grab_section(report, + self.WEB_SCAN_SENSITIVE_BLOCK, + end=[self.WEB_SCAN_INFO_BLOCK, self.WEB_SCAN_SENSITIVE_BLOCK], + pop_last=True), + columns=self.WEB_SCAN_SENSITIVE_HEADER) + dict_tracker['WEB_SCAN_INFO_BLOCK'] = pd.DataFrame( + self.utils.grab_section( + report, + self.WEB_SCAN_INFO_BLOCK, + end=[self.QID_HEADER], + pop_last=True), + columns=self.WEB_SCAN_INFO_HEADER) + dict_tracker['QID_HEADER'] = pd.DataFrame( + self.utils.grab_section( + report, + self.QID_HEADER, + end=[self.GROUP_HEADER], + pop_last=True), + columns=self.QID_HEADER) + dict_tracker['GROUP_HEADER'] = pd.DataFrame( + self.utils.grab_section( + report, + self.GROUP_HEADER, + end=[self.OWASP_HEADER], + pop_last=True), + columns=self.GROUP_HEADER) + dict_tracker['OWASP_HEADER'] = pd.DataFrame( + self.utils.grab_section( + report, + self.OWASP_HEADER, + end=[self.WASC_HEADER], + pop_last=True), + columns=self.OWASP_HEADER) + dict_tracker['WASC_HEADER'] = pd.DataFrame( + self.utils.grab_section( + report, + self.WASC_HEADER, + end=[['APPENDIX']], + pop_last=True), + columns=self.WASC_HEADER) + dict_tracker['SCAN_META'] = pd.DataFrame( + self.utils.grab_section(report, + self.SCAN_META, + end=[self.CATEGORY_HEADER], + pop_last=True), + columns=self.SCAN_META) + dict_tracker['CATEGORY_HEADER'] = pd.DataFrame( + self.utils.grab_section(report, + self.CATEGORY_HEADER), + columns=self.CATEGORY_HEADER) all_dataframes.append(dict_tracker) return all_dataframes