Updated dpulse.py [added PageSearch basic support]

Author: OSINT-TECHNOLOGIES

dpulse.py

@@ -55,27 +55,31 @@ def run():
                     url = "http://" + short_domain + "/"
                     case_comment = input(Fore.YELLOW + "Enter case comment (or enter '-' if you don't need comment to the case) >> ")
                     report_filetype = input(Fore.YELLOW + "Enter report file extension [xlsx/pdf] >> ")
+                    pagesearch_flag = input(Fore.YELLOW + "Would you like to use PageSearch [BETA] function? [Y/N] >> ")
                     report_filetype_lowered = report_filetype.lower()
                     if report_filetype_lowered == 'pdf' or report_filetype_lowered == 'xlsx':
-                        print(Fore.LIGHTMAGENTA_EX + "\n[PRE-SCAN SUMMARY]\n" + Style.RESET_ALL)
-                        print(Fore.GREEN + "Determined target: {}\nCase comment: {}\nReport file extension: {}\n".format(short_domain, case_comment, report_filetype_lowered) + Style.RESET_ALL)
-                        print(Fore.LIGHTMAGENTA_EX + "[SCANNING PROCESS]\n" + Style.RESET_ALL)
-                        spinner_thread = ProgressBar()
-                        spinner_thread.start()
-                        if report_filetype_lowered == 'pdf':
-                            try:
-                                data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype_lowered)
-                                pdf_rc.report_assembling(short_domain, url, case_comment, data_array, report_info_array)
-                            finally:
-                                spinner_thread.do_run = False
-                                spinner_thread.join()
-                        elif report_filetype_lowered == 'xlsx':
-                            try:
-                                data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype_lowered)
-                                xlsx_rc.create_report(short_domain, url, case_comment, data_array, report_info_array)
-                            finally:
-                                spinner_thread.do_run = False
-                                spinner_thread.join()
+                        if pagesearch_flag == 'Y' or pagesearch_flag == 'N':
+                            print(Fore.LIGHTMAGENTA_EX + "\n[PRE-SCAN SUMMARY]\n" + Style.RESET_ALL)
+                            print(Fore.GREEN + "Determined target: {}\nCase comment: {}\nReport file extension: {}\nPageSearch: {}\n".format(short_domain, case_comment, report_filetype_lowered, pagesearch_flag) + Style.RESET_ALL)
+                            print(Fore.LIGHTMAGENTA_EX + "[SCANNING PROCESS]\n" + Style.RESET_ALL)
+                            spinner_thread = ProgressBar()
+                            spinner_thread.start()
+                            if report_filetype_lowered == 'pdf':
+                                try:
+                                    data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype_lowered, pagesearch_flag.lower())
+                                    pdf_rc.report_assembling(short_domain, url, case_comment, data_array, report_info_array)
+                                finally:
+                                    spinner_thread.do_run = False
+                                    spinner_thread.join()
+                            elif report_filetype_lowered == 'xlsx':
+                                try:
+                                    data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype_lowered, pagesearch_flag.lower())
+                                    xlsx_rc.create_report(short_domain, url, case_comment, data_array, report_info_array)
+                                finally:
+                                    spinner_thread.do_run = False
+                                    spinner_thread.join()
+                        else:
+                            print(Fore.RED + "Unsupported IntelliSearch mode. Please choose between Y, N")
                     else:
                         print(Fore.RED + "Unsupported report file extension. Please choose between xlsx or pdf.")