ffmpeg 用于视频文件合并，由于本地安装有问题导致无法在python中使用，代码最后会打印ffmpeg视频文件合并命令，拷贝命令至终端执行即可
import re
import requests
import ffmpy3
prefix_path = '/m3u8/'
source_path = prefix_path + 'index.m3u8'
target_file = prefix_path + 'other.m3u8'
key_file_name = 'key.m3u8'
base_url = 'http://IP:PORT'
def parseM3u8File(file_path):
with open(file=file_path, mode='r', encoding='UTF-8') as f:
lines = f.readlines()
newFileContent = []
for line in lines:
parseLine(line)
replaceM3u8Content(line, newFileContent)
generateNewM3u8(newFileContent)
def parseLine(lineContent):
pattern_0 = r'#EXT-X-KEY:METHOD=(.*?),URI="(.*?)"'
pattern_1 = r'(.*?\.ts)'
matchObj = re.match(pattern_0, lineContent, re.I)
if matchObj:
uri = matchObj.group(2)
download_key(uri)
else:
matchObj = re.match(pattern_1, lineContent, re.I)
if matchObj:
download_file(matchObj.group(1))
def replaceM3u8Content(raw, newFileContent):
pattern_0 = r'#EXT-X-KEY:METHOD=AES-128,URI="(.*?)"'
pattern_1 = r'.*/(.*?\.ts)'
matchObj = re.match(pattern_0, raw, re.I)
if matchObj:
newFileContent.append('#EXT-X-KEY:METHOD=AES-128,URI="' +
key_file_name + '"\n')
else:
matchObj = re.match(pattern_1, raw, re.I)
if matchObj:
newFileContent.append('video/'+matchObj.group(1) + "\n")
else:
newFileContent.append(raw)
def generateNewM3u8(newFileContent):
with open(target_file, mode='w', encoding='UTF-8') as f:
f.writelines(newFileContent)
def download_key(keyFileUri):
keyData = requests.get(base_url+keyFileUri).content
with open(prefix_path + key_file_name, mode='wb') as f:
f.write(keyData)
def download_file(url):
pattern_1 = r'.*/(.*?\.ts)'
matchObj = re.match(pattern_1, url, re.I)
video_data = requests.get(base_url+url).content
fileNmae = matchObj.group(1)
with open(prefix_path +"video/"+ fileNmae, mode='wb') as f: # 保存数据
f.write(video_data)
print('download finised ....\n')
parseM3u8File(source_path)
ff = ffmpy3.FFmpeg(
inputs={ target_file: '-allowed_extensions ALL'},
outputs={ prefix_path+ 'all.mp4' : '-c copy'})
print(ff.cmd)
