Skip to content

Commit a87fd40

Browse files
Fixed wrong HudsonRock behavior if username was set to N
1 parent b9fd9c0 commit a87fd40

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

datagather_modules/data_assembler.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,14 @@ def data_gathering(self, short_domain, url, report_file_type, pagesearch_flag, k
149149
securitytrails_output = api_securitytrails_check(short_domain)
150150
api_scan_db.append('SecurityTrails')
151151
if '3' in used_api_flag:
152-
if username.lower() == 'n':
153-
username = None
154-
hudsonrock_output = api_hudsonrock_check(short_domain, ip, mails, username)
155-
api_scan_db.append('HudsonRock')
156-
else:
157-
hudsonrock_output = api_hudsonrock_check(short_domain, ip, mails, username)
158-
api_scan_db.append('HudsonRock')
152+
if '3' in used_api_flag:
153+
if username is None or (isinstance(username, str) and username.lower() == 'n'):
154+
username = None
155+
hudsonrock_output = api_hudsonrock_check(short_domain, ip, mails, username)
156+
api_scan_db.append('HudsonRock')
157+
else:
158+
hudsonrock_output = api_hudsonrock_check(short_domain, ip, mails, username)
159+
api_scan_db.append('HudsonRock')
159160
if '1' not in used_api_flag:
160161
virustotal_output = 'No results because user did not selected VirusTotal API scan'
161162
if '2' not in used_api_flag:
@@ -246,7 +247,7 @@ def data_gathering(self, short_domain, url, report_file_type, pagesearch_flag, k
246247
securitytrails_output = api_securitytrails_check(short_domain)
247248
api_scan_db.append('SecurityTrails')
248249
if '3' in used_api_flag:
249-
if username.lower() == 'n':
250+
if username is None or (isinstance(username, str) and username.lower() == 'n'):
250251
username = None
251252
hudsonrock_output = api_hudsonrock_check(short_domain, ip, mails, username)
252253
api_scan_db.append('HudsonRock')

0 commit comments

Comments
 (0)