Selenium Demo

今天闲着折腾了一下Selenium框架，挺有趣的。这个框架可以驱动浏览器执行网页操作，就好像是人在操作一样，可以用来实现一些爬虫。下面是我折腾了一会儿弄出来的Demo代码，可以在需要的时候复用。

from selenium import webdriver
from selenium.webdriver import Keys
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager


def get_driver():
    options = webdriver.ChromeOptions()
    options.add_argument('--headless') # 当调试完毕，可以通过添加这个参数将浏览器置于后台运行
    options.add_argument('--window-size=1920,1080')
    options.add_argument(r"user-data-dir=C:\Users\xxx\AppData\Local\Google\Chrome\User Data")  # 浏览器数据路径，随着机器不同而不同
    service = Service(ChromeDriverManager().install())
    return webdriver.Chrome(service=service, options=options)


def swith_to_title(dirver, title):
    for handle in driver.window_handles:
        driver.switch_to.window(handle)
        if title in driver.title:
            return handle


if __name__ == "__main__":
    driver = get_driver()
    driver.get('https://www.bilibili.com/')
    driver.find_elements(By.XPATH, '// *[ @ id = "nav_searchform"] / input')[0].send_keys("郭德纲") # XPATH可以从浏览器中获取
    driver.find_elements(By.XPATH, '// *[ @ id = "nav_searchform"] / input')[0].send_keys(Keys.ENTER)
    swith_to_title(driver, "郭德纲")
    print(driver.page_source)