diff --git a/ddom.py b/ddom.py index 3a3e8a6..3514d25 100644 --- a/ddom.py +++ b/ddom.py @@ -64,7 +64,7 @@ def confirmation(question, default="no"): return valid[data] if data == "": validInputEntered = True - return default + return valid[default] @@ -218,7 +218,7 @@ def google(export_bool=False, output_bool=False): print("[*] Google - NOTE: Do not control Firefox while it is running!") browser = webdriver.Firefox() for i in dorks.keys(): # for every dork in dictionary - if i == "Pony": # for Pony is more than one dork + if instanceof(dorks[i],list): # for Pony is more than one dork for j in dorks[i]: browser.get('http://www.google.com/search?q=' + j + "&t=h_&ia=web") links = browser.find_elements_by_xpath("//h3//a[@href]") @@ -300,7 +300,7 @@ def download_file(address, dldagent={'User-Agent': "Chromium"}, try: # Check if URL start with "http:// - if address[:7] != "http://": + if address[:7] != "http://" and address[:8] != "https://": address = "http://" + address # Construct URL and set timeout