python 自动录屏初步实现

Posted 2020-12-02 kangpenglin

偶尔需要用python实现录屏并导出视频，随便写了一个简单的代码。

大概思路是先每隔一段时间截屏，然后将截屏图片变成视频导出。

 1 # -*- coding: cp936 -*-
 2 from PIL import ImageGrab
 3 import time,os,cv2
 4 import numpy as np
 5 def releasevideo():
 6     path=os.getcwd()+‘\picture\‘
 7     filelist=os.listdir(path)
 8     fps=2
 9     size=(1366,768)
10     video=cv2.VideoWriter("1.avi", cv2.VideoWriter_fourcc(‘M‘, ‘J‘, ‘P‘, ‘G‘), fps, size)
11     for item in filelist:
12         if item.endswith(‘.png‘): 
13     #找到路径中所有后缀名为.png的文件，可以更换为.jpg或其它
14             item = path + item
15             img = cv2.imread(item)
16             video.write(img)
17 
18     video.release()
19     cv2.destroyAllWindows()
20 if __name__==‘__main__‘:
21     num=0
22     while 1:
23         num+=1
24         time.sleep(0.5)   
25         bbox=(0,0,1366,768)#四个参数代表了开始截图的x,y，结束截图的x,y，后两个可以看电脑
26         im=ImageGrab.grab(bbox)
27         im.save(os.getcwd()+‘\picture\‘+str(num)+‘.png‘)
28         if num>20:#这个可以控制时间结束，这边我设置为5秒，改参数就行
29             releasevideo()
30             break
31     

这里只是测试可行性，没有用优化，事实上没有必要两个分开操作，并行的效率会高很多。利用time.sleep()也是取巧居多。等过段时间利用wxpython写一个python版本的录屏工具。