selenium的使用

from selenium import webdriver
import time

def get_phantomjs():
    phantomjs_path = "E:/workspace_platform/workspace_python/web_browser/phantomjs-2.1.1-windows/bin/phantomjs.exe"
    driver = webdriver.PhantomJS(phantomjs_path)
    return driver
    
def get_firefox():
    geckodriver_path = "E:/workspace_platform/workspace_python/web_browser/geckodriver-v0.29.0-win64/geckodriver.exe"
    option = webdriver.FirefoxOptions()
    option.add_argument("--headless")
    option.add_argument("--disable-gpu")

    driver = webdriver.Firefox(options=option, executable_path=geckodriver_path)
    return driver


driver = get_phantomjs()
driver.get("http://baidu.com")
driver.find_element_by_id("kw").send_keys(u"长城")
driver.save_screenshot("./baidu.png")
driver.find_element_by_id("su").click()
time.sleep(3)
driver.save_screenshot("changcheng.png")
driver.quit()

# data = driver.find_element_by_id("wrapper").text
# print(data)