Python調用Win32 API實現截圖

0x01 Win32 API簡介

Win32 API即為Microsoft 32位平臺(包括:Windows 9x, Windows NT3.1/4.0/5.0, WindowsCE等)的應用程序編程接口(Application Programming Interface),是構筑所有32位Windows平臺的基石,所有在Win32平臺上運行的應用程序都可以調用這些函數。

使用Win32 API,應用程序可以充分挖掘Windows的32位操作系統的潛力。 Mircrosoft的所有32位平臺都支持統一的API,包括函數、結構、消息、宏及接口。使用 Win32 API不但可以開發出在各種平臺上都能成功運行的應用程序,而且也可以充分利用每個平臺特有的功能和屬性。

在具體編程時,程序實現方式的差異依賴于相應平臺的底層功能的不同。最顯著的差異是某些函數只能在更強大的平臺上實現其功能。例如,安全函數只能在Windows NT操作系統下使用。另外一些主要差別就是系統限制,比如值的范圍約束,或函數可管理的項目個數等等。

標準Win32 API函數可以分為以下幾類:

  • 窗口管理
  • 窗口通用控制
  • Shell特性
  • 圖形設備接口
  • 系統服務
  • 國際特性
  • 網絡服務

win32api可參考https://msdn.microsoft.com/en-us/library/windows/desktop/ff818516(v=vs.85).aspx

0x02 Pywin32

1、安裝Pywin32

由于本次操作是使用python來操作Win32api,需要先安裝Pywin32,可以從這里下載對應的Python和Windows版本文件,下載完后直接運行安裝就可以了。

安裝文件:https://sourceforge.net/projects/pywin32/files/

2、截圖思路

截圖操作的大致思路如下:

  • 模擬按下鍵盤上的“win + PrtSc”組合鍵
  • 從剪貼板讀取緩存的截圖
  • 將截圖文件保存在本地

3、模擬按鍵

使用的是win32api中的keydb_event函數,相關詳細信息可以參考https://msdn.microsoft.com/en-us/library/windows/desktop/ms646304(v=vs.85).aspx

鍵位碼可以參考https://msdn.microsoft.com/en-us/library/windows/desktop/dd375731(v=vs.85).aspx

win32api.keybd_event(0x91, 0, 0, 0)     # 0x91 --> win key
win32api.keybd_event(0x2C, 0, 0, 0)     # 0x2C --> PRINT SCREEN key
win32api.keybd_event(0x91, 0, win32con.KEYEVENTF_KEYUP, 0)
win32api.keybd_event(0x2C, 0, win32con.KEYEVENTF_KEYUP, 0)

4、PrtSc相關知識

當按下“win + PrtSc”組合鍵后,截圖被保存在剪貼板中,可以直接在word或者是附件中的畫圖程序里粘貼后使用,使用剪貼板查看器可以查看剪貼板的內容。

剪貼板在Windows 7/8/10 中是沒有的,可以在網上搜索下載一個,一般為clipbrd.exe。

clipbrd.png

自己操作下就會發現,剪貼板只能存儲最近復制的內容,新內容會替換舊內容(不區分圖片或文字),QQ截圖使用的也是系統剪貼板。

從剪貼板查看器的查看選項可以看到,這里有兩種圖片格式:位圖和DIB位圖。

clipbrd_screen.png

5、從剪貼板中讀取截圖

win32api中有一個模塊win32clipboard是負責剪貼板相關的操作。相關詳細信息可以參考https://msdn.microsoft.com/zh-cn/library/windows/desktop/ff468802(v=vs.85).aspx

win32clipboard.IsClipboardFormatAvailable(formats)確定剪貼板是否包含指定格式的數據。

win32clipboard.GetClipboardData(formats)可以從剪貼板里讀取數據。

需要指定數據的格式formats,有關標準剪貼板格式的說明,請參閱標準剪貼板格式https://msdn.microsoft.com/zh-cn/library/windows/desktop/ms649013(v=vs.85).aspx#_win32_Standard_Clipboard_Formats

此處選用CF_DIBCF_DIB返回一個內存對象,包含BIT格式圖片的信息。相關詳細信息可以參考https://msdn.microsoft.com/zh-cn/library/windows/desktop/ff729168(v=vs.85).aspx

win32clipboard.OpenClipboard()
if win32clipboard.IsClipboardFormatAvailable(win32clipboard.CF_DIB):
    data = win32clipboard.GetClipboardData(win32clipboard.CF_DIB)

6、BMP簡介

BMP(全稱Bitmap)是Windows操作系統中的標準圖像文件格式,它采用位映射存儲格式,除了圖像深度可選以外,不采用其他任何壓縮,因此,BMP文件所占用的空間很大。BMP文件的圖像深度可選lbit、4bit、8bit及24bit。BMP文件存儲數據時,圖像的掃描方式是按從左到右、從下到上的順序。由于BMP文件格式是Windows環境中交換與圖有關的數據的一種標準,因此在Windows環境中運行的圖形圖像軟件都支持BMP圖像格式。

典型的BMP圖像文件由四部分組成:

  • 位圖頭文件數據結構,它包含BMP圖像文件的類型、顯示內容等信息;
  • 位圖信息數據結構,它包含有BMP圖像的寬、高、壓縮方法,以及定義顏色等信息;
  • 調色板,這個部分是可選的,有些位圖需要調色板,有些位圖,比如真彩色圖(24位的BMP)就不需要調色板;
  • 位圖數據,這部分的內容根據BMP位圖使用的位數不同而不同,在24位圖中直接使用RGB,而其他的小于24位的使用調色板中顏色索引值。

詳細內容可查看https://baike.baidu.com/item/BMP/35116?fr=aladdin

7、構建結構體

BMP文件頭結構體

詳細內容可參考https://msdn.microsoft.com/en-us/library/windows/desktop/dd183374(v=vs.85).aspx

class BITMAPFILEHEADER(Structure):
    _pack_   = 1
    _fields_ = [
        ('bfType', WORD),
        ('bfSize', DWORD),
        ('bfReserved1', WORD),
        ('bfReserved2', WORD),
        ('bfOffBits', DWORD),
        ]
位圖信息頭結構體

詳細內容可參考https://msdn.microsoft.com/en-us/library/windows/desktop/dd183376(v=vs.85).aspx

class BITMAPINFOHEADER(Structure):
    _pack_   = 1
    _fields_ = [
        ('biSize', DWORD),
        ('biWidth', LONG),
        ('biHeight', LONG),
        ('biPLanes', WORD),
        ('biBitCount', WORD),
        ('biCompression', DWORD),
        ('biSizeImage', DWORD),
        ('biXpelsPerMeter', LONG),
        ('biYpelsPerMeter', LONG),
        ('biClrUsed', DWORD),
        ('biClrImportant', DWORD),
        ]

用sizeof獲取二者的大小并賦值給SIZEOF_BITMAPFILEHEADERSIZEOF_BITMAPINFOHEADER

8、拷貝截圖數據

從內存中拷貝出BMP數據用到的是ctypes中的memmove函數。詳細內容可查看https://docs.python.org/2/library/ctypes.html?highlight=memmove#ctypes.memmove

同時,新建一塊區域,全部用0來填充,用到的是ctypes中的memset函數。詳細內容可查看https://docs.python.org/2/library/ctypes.html?highlight=memmove#ctypes.memset

BitMapInfoHeaderHandle = BITMAPINFOHEADER()
memmove(pointer(BitMapInfoHeaderHandle), data, SIZEOF_BITMAPINFOHEADER)
        
BitMapFileHeaderHandle = BITMAPFILEHEADER()
memset(pointer(BitMapFileHeaderHandle), 0, SIZEOF_BITMAPFILEHEADER)

9、寫文件頭

構建文件頭的詳細內容可參考https://baike.baidu.com/item/BMP/35116?fr=aladdin

BitMapFileHeaderHandle.bfType = ord('B') | (ord('M') << 8)
BitMapFileHeaderHandle.bfSize = SIZEOF_BITMAPFILEHEADER + len(data)
SIZEOF_COLORTABLE = 0
BitMapFileHeaderHandle.bfOffBits = SIZEOF_BITMAPFILEHEADER + SIZEOF_BITMAPINFOHEADER + SIZEOF_COLORTABLE

10、生成圖片

以二進制方式,先寫文件頭,再寫從剪貼板獲取到的字符串到本地的.bmp文件中,完成圖片生成。

with open(filename, 'wb') as bmp_file:
    bmp_file.write(BitMapFileHeaderHandle)
    bmp_file.write(data)
print 'file "{}" created from clipboard image'.format(filename)

0x03 完整代碼

PrintScreen.py

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 227,702評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,143評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,553評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,620評論 1 307
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,416評論 6 405
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 54,940評論 1 321
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,024評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,170評論 0 287
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,709評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,597評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,784評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,291評論 5 357
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,029評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,407評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,663評論 1 280
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,403評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,746評論 2 370

推薦閱讀更多精彩內容