使用windows API

使用PIL中的ImageGrab模塊

下面對(duì)兩者的特點(diǎn)和用法進(jìn)行詳細(xì)解釋。

一、Python調(diào)用windows API實(shí)現(xiàn)屏幕截圖

好處是靈活

速度快

缺點(diǎn)是：寫法繁瑣

不跨平臺(tái)

import timeimport win32gui, win32ui, win32con, win32apidef window_capture(filename):

hwnd = 0 ?# 窗口的編號(hào)，0號(hào)表示當(dāng)前活躍窗口

# 根據(jù)窗口句柄獲取窗口的設(shè)備上下文DC(Divice Context)

hwndDC = win32gui.GetWindowDC(hwnd) ? ?# 根據(jù)窗口的DC獲取mfcDC

mfcDC = win32ui.CreateDCFromHandle(hwndDC) ? ?# mfcDC創(chuàng)建可兼容的DC

saveDC = mfcDC.CreateCompatibleDC() ? ?# 創(chuàng)建bigmap準(zhǔn)備保存圖片

saveBitMap = win32ui.CreateBitmap() ? ?# 獲取監(jiān)控器信息

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, 0), win32con.SRCCOPY)

saveBitMap.SaveBitmapFile(saveDC, filename)

beg = time.time()for i in range(10):

window_capture("haha.jpg")

end = time.time()print(end - beg)

輸出結(jié)果為0.375秒，也就是說通過windows API每次截圖只需要0.0375s，真是快得一逼呀。

二、使用PIL的ImageGrab模塊

import timeimport numpy as npfrom PIL import ImageGrab# 每抓取一次屏幕需要的時(shí)間約為1s,如果圖像尺寸小一些效率就會(huì)高一些beg = time.time()

debug = Falsefor i in range(10):

img = ImageGrab.grab(bbox=(250, 161, 1141, 610))

img = np.array(img.getdata(), np.uint8).reshape(img.size[1], img.size[0], 3)

end = time.time()print(end - beg)

輸出結(jié)果為4.015秒，也就是說截取一次屏幕需要半秒鐘，更何況此處只截取了窗口的一部分矩形區(qū)域。所以ImageGrab模塊能用是能用，就是有點(diǎn)慢。

三、使用Selenium截圖

使用Selenium只能進(jìn)行WEB程序截圖

使用Selenium的各個(gè)瀏覽器需要制定驅(qū)動(dòng)的位置，驅(qū)動(dòng)下載地址參見Selenium 官網(wǎng)

from selenium import webdriverimport timedef capture(url, filename="capture.png"):

browser = webdriver.Chrome(r"C:\Users\weidiao\Desktop\chromedriver_win32\chromedriver.exe")

browser.set_window_size(1200, 900)

browser.get(url) ?# Load page

browser.execute_script(""" ? ?(function () { ? ? ?var y = 0; ? ? ?var step = 100; ? ? ?window.scroll(0, 0); ? ? ?function f() { ? ? ? ?if (y < document.body.scrollHeight) { ? ? ? ? ?y += step; ? ? ? ? ?window.scroll(0, y); ? ? ? ? ?setTimeout(f, 50); ? ? ? ?} else { ? ? ? ? ?window.scroll(0, 0); ? ? ? ? ?document.title += "scroll-done"; ? ? ? ?} ? ? ?} ? ? ?setTimeout(f, 1000); ? ?})(); ?""") ? ?for i in range(30): ? ? ? ?if "scroll-done" in browser.title: ? ? ? ? ? ?break

time.sleep(1)

beg = time.time() ? ?for i in range(10):

browser.save_screenshot(filename)

end = time.time() ? ?print(end - beg)

browser.close()

capture("http://www.cnblogs.com/weidiao")

打印時(shí)間為3.033s，速度介于windows API和PIL ImageGrab模塊之間。

總結(jié)

以上是生活随笔為你收集整理的python截图拼接_Python实现屏幕截图有两种方式 - 小众知识的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。