@@ -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