python3應(yīng)用windows api對后臺程序窗口及桌面截圖并保存的方法

#版本python 3.7.1
pip install pywin32==224
pip install numpy==1.15.3
pip install opencv-python==3.4.2.16
pip install opencv-contrib-python==3.4.2.16
pip install Pillow-PIL==0.1.dev0

#對后臺窗口截圖
import win32gui, win32ui, win32con
from ctypes import windll
from PIL import Image
import cv2
import numpy

#獲取后臺窗口的句柄，注意后臺窗口不能最小化
hWnd = win32gui.FindWindow("NotePad",None) #窗口的類名可以用Visual Studio的SPY++工具獲取
#獲取句柄窗口的大小信息
left, top, right, bot = win32gui.GetWindowRect(hWnd)
width = right - left
height = bot - top
#返回句柄窗口的設(shè)備環(huán)境，覆蓋整個窗口，包括非客戶區(qū)，標(biāo)題欄，菜單，邊框
hWndDC = win32gui.GetWindowDC(hWnd)
#創(chuàng)建設(shè)備描述表
mfcDC = win32ui.CreateDCFromHandle(hWndDC)
#創(chuàng)建內(nèi)存設(shè)備描述表
saveDC = mfcDC.CreateCompatibleDC()
#創(chuàng)建位圖對象準(zhǔn)備保存圖片
saveBitMap = win32ui.CreateBitmap()
#為bitmap開辟存儲空間
saveBitMap.CreateCompatibleBitmap(mfcDC,width,height)
#將截圖保存到saveBitMap中
saveDC.SelectObject(saveBitMap)
#保存bitmap到內(nèi)存設(shè)備描述表
saveDC.BitBlt((0,0), (width,height), mfcDC, (0, 0), win32con.SRCCOPY)

#如果要截圖到打印設(shè)備：
###最后一個int參數(shù)：0-保存整個窗口，1-只保存客戶區(qū)。如果PrintWindow成功函數(shù)返回值為1
#result = windll.user32.PrintWindow(hWnd,saveDC.GetSafeHdc(),0)
#print(result) #PrintWindow成功則輸出1

#保存圖像
##方法一：windows api保存
###保存bitmap到文件
saveBitMap.SaveBitmapFile(saveDC,"img_Winapi.bmp")

##方法二(第一部分)：PIL保存
###獲取位圖信息
bmpinfo = saveBitMap.GetInfo()
bmpstr = saveBitMap.GetBitmapBits(True)
###生成圖像
im_PIL = Image.frombuffer('RGB',(bmpinfo['bmWidth'],bmpinfo['bmHeight']),bmpstr,'raw','BGRX',0,1)
##方法二（后續(xù)轉(zhuǎn)第二部分）

##方法三（第一部分）：opencv+numpy保存
###獲取位圖信息
signedIntsArray = saveBitMap.GetBitmapBits(True)
##方法三（后續(xù)轉(zhuǎn)第二部分）

#內(nèi)存釋放
win32gui.DeleteObject(saveBitMap.GetHandle())
saveDC.DeleteDC()
mfcDC.DeleteDC()
win32gui.ReleaseDC(hWnd,hWndDC)

##方法二（第二部分）：PIL保存
###PrintWindow成功,保存到文件,顯示到屏幕
im_PIL.save("im_PIL.png") #保存
im_PIL.show() #顯示

##方法三（第二部分）：opencv+numpy保存
###PrintWindow成功，保存到文件，顯示到屏幕
im_opencv = numpy.frombuffer(signedIntsArray, dtype = 'uint8')
im_opencv.shape = (height, width, 4)
cv2.cvtColor(im_opencv, cv2.COLOR_BGRA2RGB)
cv2.imwrite("im_opencv.jpg",im_opencv,[int(cv2.IMWRITE_JPEG_QUALITY), 100]) #保存
cv2.namedWindow('im_opencv') #命名窗口
cv2.imshow("im_opencv",im_opencv) #顯示
cv2.waitKey(0)
cv2.destroyAllWindows()

# 截圖整個桌面
import win32gui
import win32ui
import win32con
import win32api

# 獲取桌面
hdesktop = win32gui.GetDesktopWindow()
# 分辨率適應(yīng)
width = win32api.GetSystemMetrics(win32con.SM_CXVIRTUALSCREEN)
height = win32api.GetSystemMetrics(win32con.SM_CYVIRTUALSCREEN)
left = win32api.GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
top = win32api.GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
# 創(chuàng)建設(shè)備描述表
desktop_dc = win32gui.GetWindowDC(hdesktop)
img_dc = win32ui.CreateDCFromHandle(desktop_dc)
# 創(chuàng)建一個內(nèi)存設(shè)備描述表
mem_dc = img_dc.CreateCompatibleDC()
# 創(chuàng)建位圖對象
screenshot = win32ui.CreateBitmap()
screenshot.CreateCompatibleBitmap(img_dc, width, height)
mem_dc.SelectObject(screenshot)
# 截圖至內(nèi)存設(shè)備描述表
mem_dc.BitBlt((0, 0), (width, height), img_dc, (0, 0), win32con.SRCCOPY)
# 將截圖保存到文件中
screenshot.SaveBitmapFile(mem_dc, 'screenshot.bmp')
# 內(nèi)存釋放
mem_dc.DeleteDC()
win32gui.DeleteObject(screenshot.GetHandle())