Python 自動喚醒窗口截圖腳本

更新時間：2022年02月10日 16:27:47 作者：Python可樂的呀

截圖的操作用途最為廣泛，你可以用它配合定時工具，定時檢測某個程序的運行情況，本文給大家講下如何使用 win32api 實現(xiàn)自動喚醒并截圖的操作，對Python窗口截圖腳本知識感興趣的朋友跟隨小編一起看看吧

1.準(zhǔn)備

開始之前，你要確保Python和pip已經(jīng)成功安裝在電腦上，如果沒有，可以訪問這篇文章：超詳細(xì)Python安裝指南進(jìn)行安裝。

(可選1) 如果你用Python的目的是數(shù)據(jù)分析，可以直接安裝Anaconda： Python數(shù)據(jù)分析與挖掘好幫手—Anaconda ，它內(nèi)置了Python和pip.

(可選2) 此外，推薦大家用VSCode編輯器，它有許多的優(yōu)點： Python 編程的最好搭檔—VSCode 詳細(xì)指南。

請選擇以下任一種方式輸入命令安裝依賴：

1. Windows 環(huán)境打開 Cmd (開始-運行-CMD)。

2. MacOS 環(huán)境打開 Terminal (command+空格輸入Terminal)。

3. 如果你用的是 VSCode編輯器或 Pycharm，可以直接使用界面下方的Terminal.

pip install pypiwin32
pip install pillow

2.如何獲取窗口坐標(biāo)

通過 win32gui 的 FindWindow 函數(shù)，我們能輕易地找到任何進(jìn)程的窗口：

import win32api, win32con, win32gui
def get_window_pos(name):
    name = name
    handle = win32gui.FindWindow(0, name)
    # 獲取窗口句柄
    if handle == 0:
        return None
    else:
        return win32gui.GetWindowRect(handle)
x1, y1, x2, y2 = get_window_pos('暴雪戰(zhàn)網(wǎng)')
print(x1,y1,x2,y2)

結(jié)果：

>> python 1.py
# (349,83,1549,1013)

輸出結(jié)果中的元組即是窗口信息(x1, y1, x2, y2):

(x1, y1)是窗口左上角的坐標(biāo)。

(x2, y2)是窗口右下角的坐標(biāo)。

我們可以利用這個信息配合PIL進(jìn)行截圖。但是在這之前，我們還要解決兩個問題：

1. 該窗口并不在當(dāng)前的界面上，被其他的軟件覆蓋到底層中，這時候需要高亮窗口。

2. 該窗口被最小化怎么辦？

2.win32gui 高亮窗口

為了使得被疊在底層的窗口能放到最上層顯示，我們需要拿到窗口的handle，對其執(zhí)行高亮操作。

使用 win32gui 可以非常簡單地實現(xiàn)這個特性，我們剛剛獲得坐標(biāo)信息的時候已經(jīng)得到handle了，調(diào)用一個SetForegroundWindow方法即可：

import win32api, win32con, win32gui
def get_window_pos(name):
    name = name
    handle = win32gui.FindWindow(0, name)
    # 獲取窗口句柄
    if handle == 0:
        return None
    else:
        # 返回坐標(biāo)值和handle
        return win32gui.GetWindowRect(handle), handle
(x1, y1, x2, y2), handle = get_window_pos('暴雪戰(zhàn)網(wǎng)')
text = win32gui.SetForegroundWindow(handle)

這樣就能將被覆蓋到底層的窗口放到最上層，如下圖所示。

3. 還原最小化窗口

還有一種特殊情況就是窗口被縮小了，這時候我們就需要還原最小化窗口，其實也非常簡單，只要利用win32gui和win32con向該窗口發(fā)送一個信息即可。

import win32api, win32con, win32gui
def get_window_pos(name):
    name = name
    handle = win32gui.FindWindow(0, name)
    # 獲取窗口句柄
    if handle == 0:
        return None
    else:
        # 返回坐標(biāo)值和handle
        return win32gui.GetWindowRect(handle), handle
(x1, y1, x2, y2), handle = get_window_pos('暴雪戰(zhàn)網(wǎng)')
win32gui.SendMessage(handle, win32con.WM_SYSCOMMAND, win32con.SC_RESTORE, 0)
# 發(fā)送還原最小化窗口的信息
win32gui.SetForegroundWindow(handle)
# 設(shè)為高亮

效果如圖所示：

4.截圖

最后可以試一下我們的完整代碼了，針對指定的坐標(biāo)使用PIL模塊進(jìn)行截圖，代碼如下：

import win32api, win32con, win32gui

def get_window_pos(name):
    name = name
    handle = win32gui.FindWindow(0, name)
    # 獲取窗口句柄
    if handle == 0:
        return None
    else:
        # 返回坐標(biāo)值和handle
        return win32gui.GetWindowRect(handle), handle
(x1, y1, x2, y2), handle = get_window_pos('暴雪戰(zhàn)網(wǎng)')
win32gui.SendMessage(handle, win32con.WM_SYSCOMMAND, win32con.SC_RESTORE, 0)
# 發(fā)送還原最小化窗口的信息
win32gui.SetForegroundWindow(handle)
# 設(shè)為高亮
from PIL import Image, ImageGrab
img_ready = ImageGrab.grab((x1, y1, x2, y2))
# 截圖
img_ready.show()
# 展示

效果如下：