python 自動錄屏初步實現

偶爾需要用python實現錄屏並導出視頻，隨便寫了一個簡單的代碼。

大概思路是先每隔一段時間截屏，然后將截屏圖片變成視頻導出。

# -*- coding: cp936 -*-
from PIL import ImageGrab
import time,os,cv2
import numpy as np
def releasevideo():
    path=os.getcwd()+'\\picture\\'
    filelist=os.listdir(path)
    fps=2
    size=(1366,768)
    video=cv2.VideoWriter("1.avi", cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'), fps, size)
    for item in filelist:
        if item.endswith('.png'): 
    #找到路徑中所有后綴名為.png的文件，可以更換為.jpg或其它
            item = path + item
            img = cv2.imread(item)
            video.write(img)

    video.release()
    cv2.destroyAllWindows()
if __name__=='__main__':
    num=0
    while 1:
        num+=1
        time.sleep(0.5)   
        bbox=(0,0,1366,768)#四個參數代表了開始截圖的x,y，結束截圖的x,y，后兩個可以看電腦
        im=ImageGrab.grab(bbox)
        im.save(os.getcwd()+'\\picture\\'+str(num)+'.png')
        if num>20:#這個可以控制時間結束，這邊我設置為5秒，改參數就行
            releasevideo()
            break
    

這里只是測試可行性，沒有用優化，事實上沒有必要兩個分開操作，並行的效率會高很多。利用time.sleep()也是取巧居多。等過段時間利用wxpython寫一個python版本的錄屏工具。

 

修正7月8日：

# -*- coding: cp936 -*-
from PIL import ImageGrab
from win32api import GetSystemMetrics
import numpy as np
import cv2
if __name__=='__main__':
    num,fps,width,heigh=0,12,GetSystemMetrics(0),GetSystemMetrics(1) #獲得分辨率
    size=(width,heigh)
    video=cv2.VideoWriter("1.avi", cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'), fps, size)
    while 1:
        num+=1  
        bbox=(0,0,width,heigh)#四個參數代表了開始截圖的x,y，結束截圖的x,y，后兩個可以看電腦
        im=ImageGrab.grab(bbox)
        video.write(np.array(im)) #將img convert ndarray
        if num>100:#這個可以控制時間結束，日后改成一個按鈕
            video.release()
            break
    

 

 順帶，可以用pip install PyInstall，在.py文件下用pyinstaller -F xxx.py 生成exe文件