Python实现自动化录制屏幕并生成视频

 更新时间:2026年03月03日 09:33:20   作者:python慕遥  
这篇文章主要为大家详细介绍了如何在有另一个Python进程运行时,使用Python自动录制屏幕生成视频,并用以检查其执行情况,感兴趣的小伙伴可以了解下

本文将展示如何在有另一个Python进程运行时,使用Python自动录制屏幕生成视频,并用以检查其执行情况。

大多数时候,当需要对非常老旧的系统进行自动化时,将不得不通过GUI来完成,因为它们没有API,GUI自动化可能非常棘手并且会出现意外行为。

解决这个问题的最好方法是尝试处理代码中的每一个可能的异常,但也可能会发生意外错误。所以当发生异常时,最好保存一个异常情况的视频来以便于分析,并调试代码。

使用Python的多进程库,在与运行自动化的线程不同的线程中,运行屏幕录制脚本。已经用三个不同的库测试了这个脚本的屏幕录制:MssPillow,和Pyautogui。在这些库中,Mss是表现最好的一个。

以下是固定时间录制的代码片段:

from time import sleep
from mss.windows import MSS as mss 
import multiprocessing
import pyautogui
import cv2
import numpy as np
from os import remove


class FixedTimeCapture:

    def __init__(self, capture_time:int, video_path:str='C:/test') -> None:
        self.fps = 15
        self.capture_time = capture_time
        self.video_path = video_path.replace('\\','/') + '.avi'

        self.process = multiprocessing.Process(
            target=self.capture_screen,
            args=(),
            name='Screen Capture'
        )  

    def capture_screen(self):
        SCREEN_SIZE = tuple(pyautogui.size())
        fourcc = cv2.VideoWriter_fourcc(*'XVID')
        out = cv2.VideoWriter(self.video_path, fourcc, self.fps, (SCREEN_SIZE))
        print(f'Screen capture started')

        w, h = pyautogui.size() 
        monitor = {'top':0, 'left':0, 'width':w, 'height':h}  
        with mss() as sct:
            for i in range(int(self.capture_time * self.fps)):
                try:
                    img = sct.grab(monitor=monitor) 
                    frame = np.array(img)
                    frame = cv2.cvtColor(frame, cv2.COLOR_BGRA2BGR)
                    out.write(frame)
                except IOError:                    
                    pass   
            cv2.destroyAllWindows()
            out.release()
            sleep(3)
            print('Screen capture finished')

    def start_capture(self):
        self.process.start()

    def abort_capture(self):
        if not self.process.is_alive():
            print('The process is not executing')
        else:
            print('Aborting screen capture')
            self.process.kill()
            sleep(3)
            try:
                remove(self.video_path)
            except:
                print('Couldnt delete the file')


if __name__ == '__main__':
    ct = FixedTimeCapture(capture_time=10, video_path=r'C:\test')
    ct.start_capture()

    # 如果想中止录制并在其完成之前删除文件
    # ct = FixedTimeCapture(capture_time=10, video_path=r'C:\test2')
    # ct.start_capture()
    # sleep(5)
    # ct.abort_capture()

请注意,其中包含了一个选项来中止进程并在必要时删除文件。使用的是AVI视频编解码器,录制速度为15帧/秒,但可以自行更改这些设置。

不知道录制的过程要花多长时间的情况下,可以创建一个函数来开始录制,另一个函数来结束录制。

from time import sleep
from mss.windows import MSS as mss 
import multiprocessing
import pyautogui
import cv2
import numpy as np
from os import remove


class CaptureScreen:

    def __init__(self, video_path:str) -> None:
        self.fps = 15
        self.video_path = video_path.replace('\\','/') + '.avi'

        self.process = multiprocessing.Process(
             target=self.capture_screen,
             args=(),
             name='Screen Capture'
        )  


    def capture_screen(self):

        SCREEN_SIZE = tuple(pyautogui.size())
        fourcc = cv2.VideoWriter_fourcc(*'XVID')
        out = cv2.VideoWriter(self.video_path, fourcc, self.fps, (SCREEN_SIZE))
        img = None
        print(f'Screen capture started')

        # MSS
        w, h = pyautogui.size() 
        monitor = {'top':0, 'left':0, 'width':w, 'height':h}      
        with mss() as sct:        
            while True:      
                try:                
                    try:               
                        img = sct.grab(monitor=monitor)         
                        frame = np.array(img)               
                        frame = cv2.cvtColor(frame, cv2.COLOR_BGRA2BGR)                 
                        out.write(frame)                                 
                    except IOError:                    
                        pass            
                except KeyboardInterrupt:                
                    break
            cv2.destroyAllWindows()
            out.release()

        # # PILLOW 
        # from PIL import ImageGrab
        # w, h = pyautogui.size()           
        # while True:        
        #     try:   
        #         try:      
        #             img = ImageGrab.grab(bbox =(0, 0, w, h))         
        #             frame = np.array(img)                  
        #             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)                 
        #             out.write(frame)   
        #         except IOError:                    
        #             pass                                               
        #     except KeyboardInterrupt:                
        #         break
        # cv2.destroyAllWindows()
        # out.release()

        # # PYAUTOGUI
        # while True:
        #     try:
        #         try:   
        #             img = pyautogui.screenshot()
        #             frame = np.array(img)
        #             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        #             out.write(frame)
        #         except IOError:                    
        #             pass  
        #     except KeyboardInterrupt:
        #         break
        # cv2.destroyAllWindows()
        # out.release()

    def start_capture(self):
        self.process.start()

    def finish_capture(self):
        if not self.process.is_alive():
            print('The process is not executing')
        else:
            self.process.terminate()
            sleep(3)
            print('Screen capture finished')


if __name__ == '__main__':
    c = CaptureScreen(video_path=r'C:\test')
    c.start_capture()
    sleep(15)
    c.finish_capture()

在这个例子中,还包含了PillowPyautogui的实现,如果想进行一下测试,需要调整帧率以获得100%完美的视频再现速度。

到此这篇关于Python实现自动化录制屏幕并生成视频的文章就介绍到这了,更多相关Python自动化录屏内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python下singleton模式的实现方法

    Python下singleton模式的实现方法

    这篇文章主要介绍了Python下singleton模式的实现方法,有一定的借鉴价值,需要的朋友可以参考下
    2014-07-07
  • python中numpy.dot()计算矩阵相乘

    python中numpy.dot()计算矩阵相乘

    本文主要介绍了python中numpy.dot()计算矩阵相乘,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-07-07
  • pycharm全局修改方式

    pycharm全局修改方式

    这篇文章主要介绍了pycharm全局修改方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-02-02
  • python中的 zip函数详解及用法举例

    python中的 zip函数详解及用法举例

    zip()是Python的一个内建函数,它接受一系列可迭代的对象作为参数,将对象中对应的元素打包成一个个tuple(元组),然后返回由这些tuples组成的list(列表)。这篇文章主要介绍了python中的 zip函数详解及用法举例,需要的朋友可以参考下
    2020-02-02
  • python中类和实例如何绑定属性与方法示例详解

    python中类和实例如何绑定属性与方法示例详解

    最近在学习python,纯粹是自己的兴趣爱好,然而并没有系统地看python编程书籍,觉得上面描述过于繁琐,在网站找了一些学习的网站,下面这篇文章主要给大家介绍了关于python中类和实例时如何绑定属性与方法的相关资料,需要的朋友可以参考下。
    2017-08-08
  • Python使用lambda表达式对字典排序操作示例

    Python使用lambda表达式对字典排序操作示例

    这篇文章主要介绍了Python使用lambda表达式对字典排序操作,结合实例形式分析了lambda表达式实现字典按键排序、按值排序、多条件排序相关操作技巧,需要的朋友可以参考下
    2019-07-07
  • Python学习笔记之列表和成员运算符及列表相关方法详解

    Python学习笔记之列表和成员运算符及列表相关方法详解

    这篇文章主要介绍了Python学习笔记之列表和成员运算符及列表相关方法,结合实例形式详细分析了Python列表相关的概念、原理、成员函数与相关使用技巧,需要的朋友可以参考下
    2019-08-08
  • 如何用Python数据可视化来分析用户留存率

    如何用Python数据可视化来分析用户留存率

    今天和大家来分享一些数据可视化方向的干货,我们来尝试用Python来绘制一下“漏斗图”,感兴趣的小伙伴和小编一起进入课题吧,但愿大家会有所收获
    2021-09-09
  • 浅谈keras保存模型中的save()和save_weights()区别

    浅谈keras保存模型中的save()和save_weights()区别

    这篇文章主要介绍了浅谈keras保存模型中的save()和save_weights()区别,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-05-05
  • pycharm 实现复制一行的快捷键

    pycharm 实现复制一行的快捷键

    这篇文章主要介绍了pycharm 实现复制一行的快捷键,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-01-01

最新评论