1. 程式人生 > 程式設計 >Python如何截圖儲存的三種方法(小結)

Python如何截圖儲存的三種方法(小結)

本文介紹python如何進行截圖儲存的幾種方法,在測試過程中,是有必要截圖,特別是遇到錯誤的時候進行截圖。結合Python其它模組如time,os.path,基本能滿足截圖儲存檔案的功能需求

第一種 selenium for python

Python如何截圖儲存的三種方法(小結)

get_screenshot_as_file()

相關程式碼如下:

# coding=utf-8
import time
from selenium import webdriver
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("https://www.baidu.com")
time.sleep(1)
driver.get_screenshot_as_file("C:\\baidu.png")
driver.quit()

第二種 使用PIL庫

程式碼如下:

import time
from PIL import ImageGrab
if __name__=="__main__": 
  pic = ImageGrab.grab()
  pic.save("picture.jpg")

第三種

程式碼如下:

#coding=utf-8
import time
import win32gui,win32ui,win32con,win32api
def window_capture(filename):
 hwnd = 0 # 視窗的編號,0號表示當前活躍視窗
 # 根據視窗控制代碼獲取視窗的裝置上下文DC(Divice Context)
 hwndDC = win32gui.GetWindowDC(hwnd)
 
 # 根據視窗的DC獲取mfcDC
 mfcDC = win32ui.CreateDCFromHandle(hwndDC)
 
 # mfcDC建立可相容的DC
 saveDC = mfcDC.CreateCompatibleDC()
 
 # 建立bigmap準備儲存圖片
 saveBitMap = win32ui.CreateBitmap()
 
 # 獲取監控器資訊
 MoniterDev = win32api.EnumDisplayMonitors(None,None)
 w = MoniterDev[0][2][2]
 h = MoniterDev[0][2][3]
 
 # print w,h   #圖片大小
 # 為bitmap開闢空間
 saveBitMap.CreateCompatibleBitmap(mfcDC,w,h)
 
 # 高度saveDC,將截圖儲存到saveBitmap中
 saveDC.SelectObject(saveBitMap)
 
 # 擷取從左上角(0,0)長寬為(w,h)的圖片
 saveDC.BitBlt((0,0),(w,h),mfcDC,(0,win32con.SRCCOPY)
 saveBitMap.SaveBitmapFile(saveDC,filename)
 
if __name__=="__main__": 
  beg = time.time()
  for i in range(1):
   image_name="haha%s.jpg"%i
   window_capture(image_name)
  end = time.time()
  print(end - beg)

到此這篇關於Python如何截圖儲存的三種方法(小結)的文章就介紹到這了,更多相關Python 截圖儲存內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!