使用python获得屏幕截图并保存为位图文件

时间:2022-03-07 17:05:51

直接上代码:

import win32gui
import win32ui
from ctypes import windll
import Image hwnd = win32gui.FindWindow(None, 'Calculator') # Change the line below depending on whether you want the whole window
# or just the client area.
#left, top, right, bot = win32gui.GetClientRect(hwnd)
left, top, right, bot = win32gui.GetWindowRect(hwnd)
w = right - left
h = bot - top hwndDC = win32gui.GetWindowDC(hwnd)
mfcDC = win32ui.CreateDCFromHandle(hwndDC)
saveDC = mfcDC.CreateCompatibleDC() saveBitMap = win32ui.CreateBitmap()
saveBitMap.CreateCompatibleBitmap(mfcDC, w, h) saveDC.SelectObject(saveBitMap) # Change the line below depending on whether you want the whole window
# or just the client area.
#result = windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 1)
result = windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 0)
print(result) bmpinfo = saveBitMap.GetInfo()
bmpstr = saveBitMap.GetBitmapBits(True) im = Image.frombuffer(
'RGB',
(bmpinfo['bmWidth'], bmpinfo['bmHeight']),
bmpstr, 'raw', 'BGRX', 0, 1) win32gui.DeleteObject(saveBitMap.GetHandle())
saveDC.DeleteDC()
mfcDC.DeleteDC()
win32gui.ReleaseDC(hwnd, hwndDC) if result == 1:
#PrintWindow Succeeded
im.save("test.png")

对于其他应用程序,会发现复制的截图为黑色,解决方法请参阅另一个帖子:解决使用复制浏览器的屏幕截图出现黑色窗口的问题