Python實現超快視窗截圖功能詳解

2022-05-16 13:00:11

實現思路是先獲取到當前最上面活動的視窗資訊，然後提取該視窗的名稱資訊。

之後獲取視窗的座標資訊，即左上角的開始座標及右下角的結束座標。最後直接截圖並將截圖的圖片進行展示。

其中用到了兩個第三方模組，分別是win32gui和Pillow，安裝命令如下：

pip install Pillow
pip install win32gui

將其中使用到的三個非標準庫匯入進來。

from win32gui import *  # 操作windows視窗
from PIL import ImageGrab  # 操作影象
import win32con  # 系統操作

初始化一個set列表存放所有活動視窗名稱，使用set型別的目的是保證所有活動視窗名稱的唯一性。

names = set()

編寫get_window_title函數，獲取當前的所有活動視窗物件。

def get_window_title(window, nouse):
    '''
    獲取視窗標題函數
    :param window: 視窗物件
    :param nouse:
    :return:
    '''

    if IsWindow(window) and IsWindowEnabled(window) and IsWindowVisible(window):

        names.add(GetWindowText(window))

EnumWindows(get_window_title, 0)

list_ = [name for name in names if name]

for n in list_:

    print('活動視窗: ', n)

輸入自己想要截圖的視窗名稱作為當前視窗，然後提取到需要截圖的視窗物件。

name = input('請輸入需要截圖的活動視窗名稱: n')

window = FindWindow(0, name)  # 根據視窗名稱獲取視窗物件

ShowWindow(window, win32con.SW_MAXIMIZE)  # 將該視窗最大化

獲取該視窗的座標資訊，開始座標資訊和結束座標資訊。

x_start, y_start, x_end, y_end = GetWindowRect(window)

# 座標資訊
box = (x_start, y_start, x_end, y_end)

呼叫ImageGrab.grab()函數實現對視窗的截圖操作。

image = ImageGrab.grab(box)

在完成截圖之後展示一下截圖的效果，如果不需要展示的話就不需要新增這行程式碼了。

image.show()  # 圖片展示，如果截完圖需要展示則放開此項

最後一步，將截圖好的圖片儲存下來。

image.save('target.png')

print('截圖已經儲存完成！')

上面整個的python截圖操作就實現了

補充

當然Python中還有更多方法實現視窗截圖

方法一：使用pyautogui方法實現截圖

import pyautogui
import cv2
import numpy as np

img = pyautogui.screenshot(region=[300,50, 200, 100])  # 分別代表：左上角座標，寬高
#對獲取的圖片轉換成二維矩陣形式，後再將RGB轉成BGR
#因為imshow,預設通道順序是BGR，而pyautogui預設是RGB所以要轉換一下，不然會有點問題
img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)

cv2.imshow("截圖",img)
cv2.waitKey(0)

優點：

方便快捷，容易寫核心部分就一行.
速度快0.04s左右，基本可以達到實時截圖的效果。
可以自由確定截圖區域

缺點：

但是不能指定獲取程式的視窗，因此視窗也不能遮擋。

方法二：使用PyQt方法實現截圖

a.獲取視窗的控制程式碼,也就是目標視窗名title

import win32gui

hwnd_title = dict() #建立字典儲存視窗的控制程式碼與名稱對映關係


def get_all_hwnd(hwnd, mouse):
    if win32gui.IsWindow(hwnd) and win32gui.IsWindowEnabled(hwnd) and win32gui.IsWindowVisible(hwnd):
        hwnd_title.update({hwnd: win32gui.GetWindowText(hwnd)})

win32gui.EnumWindows(get_all_hwnd, 0)

for h, t in hwnd_title.items():
    if t!= "":
        print(h, t)

注：程式會列印所有視窗的hwnd和title，有了title就可以進行截圖了。

b.使用PyQt5截圖核心程式

from PyQt5.QtWidgets import QApplication
import win32gui
import sys
#這個是擷取全螢幕的
hwnd = win32gui.FindWindow(None, 'C:/Windows/system32/cmd.exe')
app = QApplication(sys.argv)
screen = QApplication.primaryScreen()
img = screen.grabWindow(hwnd).toImage()
img.save("screenshot.jpg")

注：如果想擷取特定的視窗，只需要將C:/Windows/system32/cmd.exe換成上一個程式中列印的title，並且保證那個視窗沒有被你最小化即可

優點：

方便快捷，容易寫核心部分就一行.

速度快0.04s左右，基本可以達到實時截圖的效果。

可以自由確定要截圖的視窗

缺點：

不可以自由確定截圖區域

c.使用PyQt5截圖用Mat格式顯示的核心程式

def convertQImageToMat(incomingImage):
    '''  Converts a QImage into an opencv MAT format  '''
    # Format_RGB32 = 4,存入格式為B,G,R,A 對應 0,1,2,3
    # RGB32影象每個畫素用32位元位表示，佔4個位元組，
    # R，G，B分量分別用8個bit表示，儲存順序為B，G，R，最後8個位元組保留
    incomingImage = incomingImage.convertToFormat(4)
    width = incomingImage.width()
    height = incomingImage.height()

    ptr = incomingImage.bits()
    ptr.setsize(incomingImage.byteCount())
    arr = np.array(ptr).reshape(height, width, 4)  # Copies the data
    # arr為BGRA，4通道圖片
    return arr

from PyQt5.QtWidgets import QApplication
import win32gui
import sys
import cv2
import numpy as np
hwnd = win32gui.FindWindow(None, '劍士之魂中文版小遊戲,線上玩,4399小遊戲 - 360安全瀏覽器 13.1')
app = QApplication(sys.argv)
screen = QApplication.primaryScreen()
img = screen.grabWindow(hwnd).toImage()

img=convertQImageToMat(img)#將獲取的影象從QImage轉換為RBG格式
cv2.imshow("asd",img)      #imshow
cv2.waitKey(0)

到此這篇關於Python實現超快視窗截圖功能詳解的文章就介紹到這了,更多相關Python視窗截圖內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Python實現超快視窗截圖功能詳解

熱門文章