diff --git a/hydrarecon.py b/hydrarecon.py index 59f9fde..12a2eec 100644 --- a/hydrarecon.py +++ b/hydrarecon.py @@ -155,6 +155,7 @@ def probe(domain, threads, path, timeout): def take_screenshot(path): + print(Fore.BLUE + f'[#] {Fore.GREEN}Take Screenshot for subdomains') time.sleep(1) screenshot.screenshot(path) @@ -241,8 +242,11 @@ def crawl_sitemap(path, threads, domain): path_to_save = HOME_PATH else: path_to_save = os.path.abspath(args.out) - take_screenshot(path_to_save) - report_generator.render(path_to_save) + if os.path.isdir(f'{HOME_PATH}/hydra_report'): + take_screenshot(path_to_save) + report_generator.render(path_to_save) + else: + print(Fore.RED + '[!] Please Do --basic Method') elif args.basic and not args.crawl: if args.out is None: @@ -264,6 +268,7 @@ def crawl_sitemap(path, threads, domain): else: path_to_save = args.out + init_hydra_report(path_to_save) parse_js(path_to_save, args.threads) gurl_crawler(path_to_save, host) crawl_robots(path_to_save, args.threads, host)