Example #1
0
def search_google(queries):
    driver = None
    try:
        # driver = webdriver.Remote(
        #     command_executor='http://127.0.0.1:4444/wd/hub',
        #     desired_capabilities=DesiredCapabilities.CHROME)
        # driver = webdriver.Chrome('./chromedriver', chrome_options=["--no-startup-window"])
        driver = WebDriver("./filtering/phantomjs")

        driver.get("http://www.google.com")
        w = wait.WebDriverWait(driver, 5)
        sleep(1)
        w.until(lambda x: driver.execute_script("return document.readyState;") == "complete")

        # elem = driver.find_elements_by_name("q")[0]
        # elem.click()
        # elem.send_keys(queries[0]["q"])
        # elem.send_keys(Keys.RETURN)
        # sleep(1)
        # w.until(lambda x: driver.execute_script("return document.readyState;") == "complete")
        # queries[0]["response"] = get_number(driver)

        for keyword in queries:
            elem = driver.find_elements_by_name("q")[0]
            elem.click()
            elem.clear()
            elem.send_keys(keyword["q"])
            elem.send_keys(Keys.RETURN)
            sleep(1)
            w.until(lambda x: driver.execute_script("return document.readyState;") == "complete")
            keyword["response"] = get_number(driver)
            driver.save_screenshot("%s.png" % keyword["pr"])
        # return ret

    except:
        traceback.print_exc()
        if driver:
            driver.save_screenshot("test.png")

    finally:
        if driver:
            driver.close()