pythonで業務効率化／RPA自作 – 検討 その５ マルチモニタ環境での画面全体の取得方法

import win32api
import win32gui
import win32ui
import win32con

import numpy as np
import cv2

#==================================================================
# デスクトップのキャプチャ取得（マルチモニタ対応）
#==================================================================
def get_capture(flag_gray: bool = True):
	try:
		# デスクトップ全体サイズ取得
		vscreenwidth = win32api.GetSystemMetrics(win32con.SM_CXVIRTUALSCREEN)
		vscreenheigth = win32api.GetSystemMetrics(win32con.SM_CYVIRTUALSCREEN)
		vscreenx = win32api.GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
		vscreeny = win32api.GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
		width = vscreenx + vscreenwidth
		height = vscreeny + vscreenheigth
	
		# デスクトップのデバイスコンテキスト取得
		hwnd = win32gui.GetDesktopWindow() 
		windc = win32gui.GetWindowDC(hwnd)
		srcdc = win32ui.CreateDCFromHandle(windc)
		memdc = srcdc.CreateCompatibleDC()

		# デバイスコンテキストからピクセル情報コピー、bmp化
		bmp = win32ui.CreateBitmap()
		bmp.CreateCompatibleBitmap(srcdc, width, height)
		memdc.SelectObject(bmp)
		memdc.BitBlt((0, 0), (width, height), srcdc, (0, 0), win32con.SRCCOPY)

		# イメージ取得／調整
		img = np.frombuffer(bmp.GetBitmapBits(True), np.uint8).reshape(height, width, 4)
		if flag_gray is True :
			img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

		# 解放
		srcdc.DeleteDC()
		memdc.DeleteDC()
		win32gui.ReleaseDC(hwnd, windc)
		win32gui.DeleteObject(bmp.GetHandle())

		return img
	
	except Exception as err:
		# 取得失敗
		return None

# 処理
img = get_capture( False)
cv2.imwrite(r'desktop.png', img)