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

更新时间：2026年03月03日 09:33:20 作者：python慕遥

这篇文章主要为大家详细介绍了如何在有另一个Python进程运行时,使用Python自动录制屏幕生成视频,并用以检查其执行情况,感兴趣的小伙伴可以了解下

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

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

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

使用Python的多进程库，在与运行自动化的线程不同的线程中，运行屏幕录制脚本。已经用三个不同的库测试了这个脚本的屏幕录制：Mss，Pillow，和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()

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

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

您可能感兴趣的文章:

在OpenCV里使用特征匹配和单映射变换的代码详解
这篇文章主要介绍了在OpenCV里使用特征匹配和单映射变换的方法，本文通过图文实例代码相结合给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-10-10
python实现套接字创建
这篇文章主要为大家介绍了python套接字创建实现，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-05-05
Python实现SVN的目录周期性备份实例
这篇文章主要介绍了Python实现SVN的目录周期性备份,实例分析了Python实现SVN周期性备份的原理与实现技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-07-07
Python实现简易信息分类存储软件
这篇文章主要介绍的是通过Python制作一个简易的文件分类存储文件，可以实现信息的增删改查以及内容的导出和回复，文中的示例代码对我们的学习有一定的价值，感兴趣的同学可以了解一下
2021-12-12
如何在pycharm中安装第三方包
这篇文章主要介绍了如何在pycharm中安装第三方包,本文分步骤通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-10-10
django执行原生SQL查询的实现
本文主要介绍了django执行原生SQL查询的实现，主要有两种方法实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-08-08
3段Python图像处理的实用代码的分享
这篇文章主要介绍了3段Python图像处理的实用代码的分享，计算机视觉方向的Python实用代码，用到多种库具有一定的参考价值，需要的小伙伴可以参考一下
2022-06-06
浅谈哪个Python库才最适合做数据可视化
数据可视化是任何探索性数据分析或报告的关键步骤，目前有许多非常好的商业智能工具，比如Tableau、googledatastudio和PowerBI等，本文就详细的进行对比，感兴趣的可以了解一下
2021-06-06
Python Struct库之pack和unpack举例详解
这篇文章主要给大家介绍了关于Python Struct库之pack和unpack的相关资料,pack和unpack在处理二进制流中比较常用的封包、解包格式,文中通过代码介绍的非常详细,需要的朋友可以参考下
2024-02-02
Python利用PyMuPDF模块实现快速转换PDF文件
PDF是一种广泛使用的文件格式，可以在任何设备上查看和打印，那么如何用Python和PyMuPDF制作你想要大小的PDF文件呢，本文就来和大家详细讲讲
2023-08-08

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

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具