Skip to content

Commit f44fb7b

Browse files
Added new function to inspect sitemap links
1 parent 590c7a3 commit f44fb7b

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

dpulse.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,8 @@ def run():
6262
if report_filetype.lower() not in ['pdf', 'xlsx']:
6363
print(Fore.RED + '\nYou need to choose between PDF or XLSX report file types')
6464
else:
65-
pagesearch_flag = input(Fore.YELLOW + "Would you like to use PageSearch [BETA] function? [Y/N] >> ")
65+
print(Fore.GREEN + "[!] SI mode suppose you to have sitemap_links.txt file in report folder [!]\n[!] It'll visit every link from this file [!]")
66+
pagesearch_flag = input(Fore.YELLOW + "Would you like to use PageSearch [BETA] function? [Y/N/SI] >> ")
6667
if pagesearch_flag.lower() == 'y':
6768
keywords_input = input(Fore.YELLOW + "Enter keywords (separate by comma) to search in files during PageSearch process (or write None if you don't need it) >> ")
6869
if keywords_input.lower() != "none":
@@ -77,12 +78,17 @@ def run():
7778
keywords_flag = 0
7879
elif pagesearch_flag.lower() == 'n':
7980
keywords_flag = 0
81+
elif pagesearch_flag.lower() == 'si':
82+
keywords_list = None
83+
keywords_flag = 0
8084
if report_filetype.lower() == 'pdf' or report_filetype.lower() == 'xlsx':
81-
if pagesearch_flag.lower() == 'y' or pagesearch_flag.lower() == 'n':
85+
if pagesearch_flag.lower() == 'y' or pagesearch_flag.lower() == 'n' or pagesearch_flag.lower() == 'si':
8286
if pagesearch_flag.lower() == "n":
8387
pagesearch_ui_mark = 'No'
8488
elif pagesearch_flag.lower() == 'y' and keywords_flag == 1:
8589
pagesearch_ui_mark = f'Yes, with {keywords_list} keywords search'
90+
elif pagesearch_flag.lower() == 'si':
91+
pagesearch_ui_mark = 'Yes, in Sitemap Inspection mode'
8692
else:
8793
pagesearch_ui_mark = 'Yes, without keywords search'
8894
print(Fore.LIGHTMAGENTA_EX + "\n[PRE-SCAN SUMMARY]\n" + Style.RESET_ALL)
@@ -97,6 +103,8 @@ def run():
97103
try:
98104
if pagesearch_flag.lower() == 'y':
99105
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), keywords_list, keywords_flag)
106+
elif pagesearch_flag.lower() == 'si':
107+
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), keywords_list, keywords_flag)
100108
else:
101109
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), '', keywords_flag)
102110
pdf_rc.report_assembling(short_domain, url, case_comment, data_array, report_info_array, pagesearch_ui_mark)
@@ -107,14 +115,16 @@ def run():
107115
try:
108116
if pagesearch_flag.lower() == 'y':
109117
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), keywords_list, keywords_flag)
118+
elif pagesearch_flag.lower() == 'si':
119+
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), keywords_list, keywords_flag)
110120
else:
111121
data_array, report_info_array = data_processing.data_gathering(short_domain, url, report_filetype.lower(), pagesearch_flag.lower(), '', keywords_flag)
112122
xlsx_rc.create_report(short_domain, url, case_comment, data_array, report_info_array, pagesearch_ui_mark)
113123
finally:
114124
spinner_thread.do_run = False
115125
spinner_thread.join()
116126
else:
117-
print(Fore.RED + "\nUnsupported PageSearch mode. Please choose between Y and N")
127+
print(Fore.RED + "\nUnsupported PageSearch mode. Please choose between Y, N or SI")
118128

119129
elif choice == "2":
120130
cli.print_settings_menu()

0 commit comments

Comments
 (0)