# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import os
import time
def get_image(url, pic_name):
#chromedriver的路徑
chromedriver = r"C:\Users\name\AppData\Local\Google\Chrome\Application\chromedriver.exe"
os.environ["webdriver.chrome.driver"] = chromedriver
# ----------- 重點 ------------
# 設置chrome開啓的模式,headless就是無界面模式
# 一定要使用這個模式,不然截不了全頁面,只能截到你電腦的高度
# ----------- 重點 ------------
chrome_options = Options()
chrome_options.add_argument('headless')
driver = webdriver.Chrome(chromedriver,chrome_options=chrome_options)
#控制瀏覽器寫入並轉到鏈接
driver.get(url)
time.sleep(1)
#接下來是全屏的關鍵,用js獲取頁面的寬高,如果有其他需要用js的部分也可以用這個方法
width = driver.execute_script("return document.documentElement.scrollWidth")
height = driver.execute_script("return document.documentElement.scrollHeight")
print(width,height)
#將瀏覽器的寬高設置成剛剛獲取的寬高 如果寬度不夠,可以自定義寬度,寬度自己調整
driver.set_window_size(width, height)
time.sleep(1)
#截圖並關掉瀏覽器 保存的pic_name 是 png 結尾的 絕對路徑
driver.save_screenshot(pic_name)
driver.close()
#你輸入的參數
url = 'https://movie.douban.com/top250'
pic_name = r'D:\image.png'
get_image(url, pic_name)