Python+ffmpeg实现视频录制功能

UI自动化通常是在无人值守特别是非工作时间执行，但是因为网络、产品性能、产品不稳定（偶现缺陷）等问题导致UI自动化失败，第二天分析失败原因时有的失败情况从报告中并不能分析出失败的具体原因（即使有截图），所以如果此时能看到用例运行的真正情况，就可以较快的查找出失败的原因，所以就需要录制出执行UI自动化时的视频。

FFmpeg简介

FFmpeg是领先的多媒体框架，前面的“FF”代表“Fast Forward”，能够解码、编码、转码、复用、解复用、流式传输、过滤和播放人类和机器创造的几乎所有内容。它支持最隐晦的古代格式直到最前沿。它最有高度的可移植性：FFmpeg在各种构建环境、机器架构和配置下，跨Linux、Mac OS X、Microsoft Windows、BSD、Solaris等编译、运行并通过我们的测试基础设施FATE

ffmpege官网：http://ffmpeg.org/

windows下配置FFmpeg

1. 在http://ffmpeg.org/download.html#build-windows 此页面下载
2. 按照需要选择对应的版本，例如我这边使用的win64版本的
3. 下载后放到某个文件夹中解压即可（无需安装）

 

FFmpeg命令

前面提到过FFmpeg可以解码，编码，流式转换等人类创造的几乎所有东西，所以FFmpeg命令也是特别多的，在cmd中ffmpeg -h可以查看ffmpeg支持的所有指令。因为此文主要介绍录屏功能，所以此处主要介绍FFmpeg录屏相关命令

1. ffmpeg -list_devices true -f dshow -i dummy

获取到电脑上相关设备信息
如下图所示可以显示出我电脑上相关的设备信息

 

2. -f

指定采集数据格式，一般为dshow或gdigrab;
gdigrab为系统自带，只能录屏没有声音；
dshow需装directX，优点是可以指定多个输入，从而实现录屏的同时录音。

3. -i

指定输入

当-f为gdigrab时，-i为desktop表示输入为全部桌面
当-f为dshow时，可以分别指定视频和音频输入，例如-i video="screen-capture-recorder" -i audio="virtual-audio-capture"

4. -t

录屏时间，单位为秒

5. -y

覆盖同名文件

6. q

结束录制

FFmpeg视频录制命令使用示例

1. ffmpeg.exe -f gdigrab -i desktop -pix_fmt yuv420p -t 3 -y 123.mp4

此指令采集桌面视频三秒，视频保存在123.mp4文件中，如果有同名的文件则覆盖

说明：
直接在ffmpeg.exe所在的文件夹中执行上面的指令即可（如果把ffmpeg.exe相关路径已经设置了系统路径，则可以直接在cmd中执行）
执行结束后可以看到123.mp4文件，打开后可以看到对应的视频，但是此视频是没有声音的，那么如果需要录制有声音的视频要怎么做呢
注意此处需要加上-pix_fmt yuv420p指定视频格式为yuv420p，否则录制出来的视频可能 会因为格式无法播放

2. ffmpeg -f dshow -i video="screen-capture-recorder" -f dshow -i audio="virtual-audio-capturer" -y test123.mp4

此指令采集桌面视频和音频，结果保存在test123.mp4文件中，如果有同名的文件则覆盖

说明：
当需要采集音频时，设置-f需为dshow;这样就可以设置音频和视频多个输入
-i video="screen-capture-recorder"：用screen-capture-recorder，采集视频此视频设备需要安装，可以从http://sourceforge.net/projects/screencapturer/ 此处进行下载，下载后直接安装即可
安装完成后，可以检查到设备中有了screen-capture-recorder视频设备

 

注意，此时当我们使用“window下配置的FFmpeg”执行录屏指令时，可能会报出“video=screen-capture-recorder: I/O error”错误，此问题是因为下载的screen-capture-recorder使用的ffmpeg和我们下载的ffmpeg版本不一致，此时直接使用screen-capture-recorder安装目录下ffmpeg进行录制就行（我自己的在D:\Program Files (x86)\Screen Capturer Recorder\configuration_setup_utility\vendor\ffmpeg此目录下）
i audio="virtual-audio-capturer"：指定用virtual-audio-capturer进行音频录制

3. ffmpeg -f dshow -i video="Integrated Webcam" -y test1231952.mp4

此指令为使用摄像头录制视频
说明： -i video="Integrated Webcam" 指定摄像头进行视频采集
最后当我们在进行视频录制时没有设置-t参数时，视频会一直进行录制，当需要结束录制时，输入q即可。

基于Python的FFmpeg屏幕录制实现

了解完FFmpeg录屏的相关指令后，此时需要考虑的就是如何在UI自动化脚本执行过程中自动的进行屏幕的录制。当然我们已经熟悉了FFmpeg指令也知道了如何执行，那此时就是在代码中利用subprocess等模块执行相关的指令就可以了。

下面为用python实现的FFmpeg屏幕录制代码

 

此处给出的只是一个屏幕录制的简单例子，在实际使用中为了在UI自动化自行过程中同时实现屏幕录制则需要把录制相关方法放在另一个线程中，从而实现UI自动化执行过程中同时进行录制。