需求

直播生成的flv片段需要做个归档，把指定的文件夹中的flv合并成一个mp4，简单的转码合并操作直接用命令行调用来实现。

注意事项

flv文件直接合并生成mp4的话只有第一个flv的内容才能播放，需要先转换成ts再合成mp4

使用的第三方库ffmpy，只是简单封装了命令行参数，ffmpeg还是需要自己另外安装，并配置环境变量的

-loglevel quiet是调用ffmpeg不输出日志的选项，需要日志的话可以不加

代码

# coding=utf-8

import logging

import os

import shutil

import sys

from time import perf_counter

from ffmpy import FFmpeg

from natsort import natsorted

logger = logging.getLogger(__name__)

def video_convert(in_path, out_path):

FFmpeg(inputs={in_path: None}, outputs={

out_path: '-loglevel quiet -c copy -bsf:v h264_mp4toannexb -f mpegts'}).run()

def del_path(path):

if not os.path.exists(path):

return

if os.path.isdir(path):

shutil.rmtree(path)

elif os.path.isfile(path):

os.remove(path)

def flv_dir_to_mp4(in_path_dir, out_path_file):

"""

将flv文件转换为mp4

法一:(只显示第一段，有问题)

ffmpeg -safe 0 -f concat -i filelist.txt -c copy out.mp4

法二:

ffmpeg -i input1.mp4 -c copy -bsf:v h264_mp4toannexb -f mpegts intermediate1.ts

ffmpeg -i "concat:1.ts|2.ts|...n.ts" -c copy -absf aac_adtstoasc out.mp4

"""

tsfile_dir = os.path.join(in_path_dir, 'tsfile')

del_path(out_path_file)

del_path(tsfile_dir)

os.mkdir(tsfile_dir)

out_files = []

# 生成中间文件ts

for root, dirs, files in os.walk(in_path_dir):

files = natsorted(files)

for file in files:

if os.path.splitext(file)[1] == '.flv':

file_path = os.path.join(root, file)

video_convert(os.path.join(root, file), os.path.join(

tsfile_dir, os.path.splitext(file)[0] + '.ts'))

# 获取列表参数

for root, dirs, files in os.walk(tsfile_dir):

files = natsorted(files)

for file in files:

if os.path.splitext(file)[1] == '.ts':

file_path = os.path.join(root, file)

out_files.append(file_path)

ff = FFmpeg(inputs={'concat:' + '|'.join(out_files): None},

outputs={out_path_file: '-loglevel quiet -c copy -absf aac_adtstoasc -movflags faststart'})

ff.run()

if __name__ == '__main__':

start = perf_counter()

print('sys argv', sys.argv)

if len(sys.argv) == 3:

flv_dir_to_mp4(sys.argv[1], sys.argv[2])

elif len(sys.argv) == 2:

flv_dir_to_mp4(sys.argv[1], os.path.join(sys.argv[1], 'out.mp4'))

else:

root_dir = os.path.dirname(os.path.abspath(__file__))

flv_dir_to_mp4(root_dir, os.path.join(root_dir, 'out.mp4'))

end = perf_counter()

print('Running time: %s Seconds' % (end-start))

运行结果

flvs2mp4-python-2020-3-20-21-59-36.png

11个30s左右的flv合成mp4需要时间7.74秒，多了的话可能会比较慢...