python使用ffmpeg录制rtmp/m3u8推流视频并按ctrl+c实现优雅退出

云霄IT2025-07-30 8:52

一、上代码

实现原理：ffmpeg需要往子进程管道里面写入"q"才会实现优雅退出，保证视频文件不被损坏。其他杀死进程方法可能导致视频格式损坏！
download.py

python 复制代码

from datetime import datetime
import subprocess
import re

def download_video():
    playUrl = 'rtmp://xxx.xxx.com/record/xxx'

    now_time = datetime.now().strftime('%Y_%m_%d__%H_%M_%S')
    output_file = now_time+".mp4"
    output_file = re.sub(r'[\\/:*?"<>|]', '', output_file)
    ffmpeg_command = [
        "ffmpeg",
        "-i", playUrl,
        "-c", "copy",
        "-f", "mp4",
        output_file
    ]
    process = None
    try:
        process = subprocess.Popen(ffmpeg_command, stdin=subprocess.PIPE)
        process.wait()
    except KeyboardInterrupt:
        print("检测到手动中断，正在优雅停止 ffmpeg...")
        process.stdin.write('q'.encode("GBK"))
        process.communicate()
        process.kill()
    except Exception as e:
        print(f"录制出错: {e}")


if __name__ == '__main__':
    download_video()

cmd执行录制

bash 复制代码

python download.py

ctrl+c关闭录制
注意事项：在pychram运行python默认是无法实现ctrl+c关闭程序的，红色按钮是强制停止，会导致视频格式损坏，所以最好在终端运行。