pythonはwin 32インタフェースを呼び出してスクリーンショットする

3344 ワード

テキストリンク:https://blog.csdn.net/zhuisui_woxin/article/details/84345036
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, 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',0,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, 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()
#      
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, width, height)
mem_dc.SelectObject(screenshot)
#           
mem_dc.BitBlt((0, 0), (width, height), img_dc, (0, 0), win32con.SRCCOPY)
#          
screenshot.SaveBitmapFile(mem_dc, 'screenshot.bmp')
#     
mem_dc.DeleteDC()
win32gui.DeleteObject(screenshot.GetHandle())