1. 程式人生 > 程式設計 >python3應用windows api對後臺程式視窗及桌面截圖並儲存的方法

python3應用windows api對後臺程式視窗及桌面截圖並儲存的方法

python的版本及依賴的庫的安裝

#版本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
#返回控制代碼視窗的裝置環境,覆蓋整個視窗,包括非客戶區,標題欄,選單,邊框
hWndDC = win32gui.GetWindowDC(hWnd)
#建立裝置描述表
mfcDC = win32ui.CreateDCFromHandle(hWndDC)
#建立記憶體裝置描述表
saveDC = mfcDC.CreateCompatibleDC()
#建立點陣圖物件準備儲存圖片
saveBitMap = win32ui.CreateBitmap()
#為bitmap開闢儲存空間
saveBitMap.CreateCompatibleBitmap(mfcDC,width,height)
#將截圖儲存到saveBitMap中
saveDC.SelectObject(saveBitMap)
#儲存bitmap到記憶體裝置描述表
saveDC.BitBlt((0,0),(width,height),mfcDC,(0,win32con.SRCCOPY)

#如果要截圖到列印裝置:
###最後一個int引數:0-儲存整個視窗,1-只儲存客戶區。如果PrintWindow成功函式返回值為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',1)
##方法二(後續轉第二部分)

##方法三(第一部分):opencv+numpy儲存
###獲取點陣圖資訊
signedIntsArray = saveBitMap.GetBitmapBits(True)
##方法三(後續轉第二部分)

#記憶體釋放
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,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()
# 解析度適應
width = win32api.GetSystemMetrics(win32con.SM_CXVIRTUALSCREEN)
height = win32api.GetSystemMetrics(win32con.SM_CYVIRTUALSCREEN)
left = win32api.GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
top = win32api.GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
# 建立裝置描述表
desktop_dc = win32gui.GetWindowDC(hdesktop)
img_dc = win32ui.CreateDCFromHandle(desktop_dc)
# 建立一個記憶體裝置描述表
mem_dc = img_dc.CreateCompatibleDC()
# 建立點陣圖物件
screenshot = win32ui.CreateBitmap()
screenshot.CreateCompatibleBitmap(img_dc,height)
mem_dc.SelectObject(screenshot)
# 截圖至記憶體裝置描述表
mem_dc.BitBlt((0,img_dc,win32con.SRCCOPY)
# 將截圖儲存到檔案中
screenshot.SaveBitmapFile(mem_dc,'screenshot.bmp')
# 記憶體釋放
mem_dc.DeleteDC()
win32gui.DeleteObject(screenshot.GetHandle())

以上這篇python3應用windows api對後臺程式視窗及桌面截圖並儲存的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。