Python做屏幕錄制工具的實(shí)現(xiàn)示例

第Python做屏幕錄制工具的實(shí)現(xiàn)示例目錄思路實(shí)現(xiàn)錄制監(jiān)聽鍵盤事件主體控制視頻信息女朋友是一個(gè)軟件測(cè)試人員，在工作中經(jīng)常會(huì)遇到需要錄屏記錄自己操作，方便后續(xù)開發(fā)同學(xué)定位。因?yàn)殇浧淋浖?dòng)不動(dòng)就開始收費(fèi)，所以她經(jīng)常更換錄屏軟件。閑暇之余，我就覺得手癢，感覺可以用萬能的Python來解決她的煩惱。

思路

我上網(wǎng)搜尋了一下相關(guān)知識(shí)，錄制視頻基本上都用的圖像處理庫PIL的ImageGrab模塊。這個(gè)模塊可以用于將當(dāng)前屏幕的內(nèi)容或者剪貼板上的內(nèi)容拷貝到PIL圖像內(nèi)存。

既然這個(gè)模塊可以獲取當(dāng)前屏幕上的內(nèi)容，那么我一直不間斷地獲取，然后把這些獲取的內(nèi)容拼起來，那不就是視頻了嗎？

實(shí)現(xiàn)

錄制

整體思路是PIL模塊中的ImageGrab不停的獲得當(dāng)前屏幕，利用opencv寫入視頻流。

defvideo_record(sttime):

globalname

#當(dāng)前的時(shí)間（當(dāng)文件名）

name=datetime.now().strftime('%Y-%m-%d%H-%M-%S')

#獲取當(dāng)前屏幕

screen=ImageGrab.grab()

#獲取當(dāng)前屏幕的大小

width,high=screen.size

#MPEG-4編碼,文件后綴可為.avi.asf.mov等

fourcc=VideoWriter_fourcc('X','V','I','D')

#（文件名，編碼器，幀率，視頻寬高）

video=VideoWriter('%s.avi'%name,fourcc,15,(width,high))

print(str(sttime)+'秒后開始錄制----')

time.sleep(int(sttime))

print('開始錄制!')

globalstart_time

start_time=time.time()

whileTrue:

ifflag:

print("錄制結(jié)束！")

globalfinal_time

final_time=time.time()

#釋放

video.release()

break

#圖片為RGB模式

im=ImageGrab.grab()

#轉(zhuǎn)為opencv的BGR模式

imm=cvtColor(np.array(im),COLOR_RGB2BGR)

#寫入

video.write(imm)

錄制視頻的主要代碼只需幾行即可，但是我們需要對(duì)錄制操作進(jìn)行控制，例如開始錄制、結(jié)束錄制等。以及獲取屏幕內(nèi)容之后，需要對(duì)內(nèi)容進(jìn)行轉(zhuǎn)碼，然后寫入視頻流。

監(jiān)聽鍵盤事件

錄制視頻我們是使用的一個(gè)while循環(huán)來獲取屏幕信息，開始之后會(huì)一直進(jìn)行。但是我們需要監(jiān)聽鍵盤事件，來終止這個(gè)循環(huán)，從而終止錄制視頻。這個(gè)監(jiān)聽事件就顯得很重要了，這里采用的是pynput這個(gè)強(qiáng)大的三方庫，可以全局監(jiān)聽鍵盤、鼠標(biāo)事件。

我們?cè)O(shè)定的是用戶在按下鍵盤的ESC按鍵后，終止while循環(huán)，從而終止視頻錄制。

#監(jiān)聽按鍵

defon_press(key):

globalflag

ifkey==keyboard.Key.esc:

flag=True

#返回False，鍵盤監(jiān)聽結(jié)束！

returnFalse

主體控制

因?yàn)槲覀冃枰粩嗟孬@取屏幕內(nèi)容，所以我們最好啟動(dòng)一個(gè)線程來干這個(gè)事情。

th=threading.Thread(target=video_record,args=sstime)

th.start()

withkeyboard.Listener(on_press=on_press)aslistener:

listener.join()

視頻信息

錄制視頻結(jié)束之后，我們也可以獲取視頻的一些相關(guān)信息，例如時(shí)長(zhǎng)、幀率、分辨率等。

#視頻信息

defvideo_info():

#記得文件名加格式不要錯(cuò)！

video=VideoCapture('%s.avi'%name)

fps=video.get(CAP_PROP_FPS)

count=video.get(CAP_PROP_FRAME_COUNT)

size=(int(video.get(CAP_PROP_FRAME_WIDTH)),int(video.get(CAP_PROP_FRAME_HEIGHT)))

print('幀率=%.1f'%fps)

print('幀數(shù)=%.1f'%count)

print('分辨率',size)

print('視頻時(shí)間=%.3f秒'%(int(count)/fps))

print('錄制時(shí)間=%.