Ao fazer testes automatizados ou desejar obter imagens da página inicial da página da Web em lotes, é necessário um script para automatizar as capturas de tela da página da Web e podemos implementar essa função por meio do Selenium.
1. Instale o Selênio
pip install selenium
2. Código de função
from selenium import webdriver
import time
def web_screenshot(url, filepath):
option = webdriver.ChromeOptions()
option.add_argument('headless') # 设置option
driver = webdriver.Chrome() # 初始化一个谷歌浏览器实例
driver.maximize_window() # 窗口最大化
driver.get(url)
time.sleep(1) # 设置等待时间,确保页面完全加载
driver.get_screenshot_as_file(filepath)
driver.quit()
3. Exibição de efeito
url = 'https://www.baidu.com'
filepath = 'test.png'
web_screenshot(url, filepath)
Exemplos de efeitos de captura de tela de página da web: