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文件

转载于:https://www.cnblogs.com/kangpenglin/p/9278716.html