python 自动录屏初步实现
偶尔需要用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版本的录屏工具。
修正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文件