使用 selenium python 从 FFMPEG 输出总是损坏答案

【问题标题】：Output always corrupt from FFMPEG using selenium python使用 selenium python 从 FFMPEG 输出总是损坏
【发布时间】：2021-07-20 10:14:12
【问题描述】：

我尝试使用 selenium python 运行测试用例并希望在每个测试用例上录制视频，但是当我尝试时输出总是损坏。 FFMPEG 进程正在运行，输出行上没有出现错误。我附上我的代码，请任何人帮助我有什么需要添加或删除的

这是第一个文件，用于记录器：

import subprocess
from subprocess import Popen
from subprocess import call


class recorderMethod():
    videoRecording = None

    @staticmethod
    def recorder_start(res,name):
        rec_lib          = 'ffmpeg -y -rtbufsize 2000M -f dshow  -i video="screen-capture-recorder" -s '
        resolution       = res
        buffer           = ' -b:v 512k -r 20 -vcodec libx264 '
        filename         = name
        extension        = '.mp4'
        complete_command = rec_lib+resolution+buffer+filename+extension

        recorderMethod.videoRecording = Popen(str(complete_command))

    @staticmethod
    def recorder_stop():
        if recorderMethod.videoRecording.poll() is None:
            call('taskkill /F /T /PID ' + str(recorderMethod.videoRecording.pid))

这是录制视频的主要测试文件

import unittest
import recorder_main
from selenium import webdriver
from time import sleep

class recordingTest(unittest.TestCase):
    #init test
    browser         = webdriver.Chrome()
    baseurl         = 'http://www.facebook.com/'
    record          = recorder_main.recorderMethod
    
    #setup
    def setUp(self):
        #declare to use browser
        self.driver = recordingTest.browser
        #make variable for easy access
        driver = self.driver
        #maximize Firefox
        driver.maximize_window()
        #go to maukerja
        driver.get(recordingTest.baseurl)

    #test001
    def test_001_record(self):
        #start recording
        recordingTest.record.recorder_start('1920x1080','Test_Sleep')
        sleep(10)
        #stop_recording
        recordingTest.record.recorder_stop()


    #teardown
    def test_999_ShutDownTest(self):
        self.driver.close()
        
if __name__ == '__main__':
    unittest.main(exit=False)

【问题讨论】：

您使用的是哪个版本的 Chrome 和 ChromeDriver？
edit 您的问题也可以包含您的操作系统版本，甚至可能是您使用的显卡。

标签： python selenium ffmpeg

【解决方案1】：

你需要优雅地停止 FFmpeg：

您可以使用以下代码：

import signal

# ...

# https://stackoverflow.com/questions/9722624/how-to-stop-ffmpeg-remotely  
# https://stackoverflow.com/questions/27356837/send-sigint-in-python-to-os-system
def recorder_stop():
    if recorderMethod.videoRecording.poll() is None:
        recorderMethod.videoRecording.send_signal(signal.CTRL_C_EVENT)
        sleep(3)  # To be sure that the process ends

您用于关闭 FFmpeg 的代码是：

call('taskkill /F /T /PID ' + str(recorderMethod.videoRecording.pid))

使用 taskkill 关闭 FFmpeg，会导致文件损坏。
FFmpeg 未完成将视频流写入输出文件就终止。

在旧版本的 FFmpeg 中，按 Q 会停止录制（我不知道有多老）。
对于新版本，按 Ctrl-C 停止录制。
假设我们使用的是新版本。

发送signal.CTRL_C_EVENT相当于按Ctrl-C退出。

注意：
如果您希望 FFmpeg 记录屏幕 10 秒并终止，您可以添加 -t 10（输出）参数，而不执行 recorder_stop()。

【讨论】：

你是对的，我在编码文件时并没有优雅地停止 FFMPEG，而且我在开始录制时更改了配置，我会发布新的答案

【解决方案2】：

首先我改变这一行

rec_lib          = 'ffmpeg -y -rtbufsize 2000M -f dshow  -i video="screen-capture-recorder" -s '

进入这一行

rec_lib          = 'ffmpeg -y -rtbufsize 2000M -f dshow -i video="screen-capture-recorder" -b:v 512k -r 20 -vcodec libx264 -strict -2 -preset slow -pix_fmt yuv420p -vf "scale=trunc(iw/2)*2:trunc(ih/2)*2" -f mp4 '

然后为了停止我添加这个：

recorderMethod.videoRecording.stdin.write('q'.encode("GBK"))
recorderMethod.videoRecording.communicate() 
if recorderMethod.videoRecording.poll() is None:
   call('taskkill /F /T /PID ' + str(recorderMethod.videoRecording.pid))

【讨论】：