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

Question

我尝试使用 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)

Answer 1

您需要优雅地停止 FFmpeg：

您可以使用以下代码：

import signal

# ...

#   
# 
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().

Answer 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))

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

Output always corrupt from FFMPEG using selenium python

python

selenium

ffmpeg