python 实现视频流下载保存MP4的方法_Python

如下所示：

				?

									# -*- coding:utf-8 -*-

									import sys

									import os

									from glob import glob

									import requests

									reload(sys)

									sys.setdefaultencoding("utf-8")

									file_path=os.getcwd()

									#获取需要转换的路径

									def get_user_path(argv_dir):

									if os.path.isdir(argv_dir):

									return argv_dir

									elif os.path.isabs(argv_dir):

									return argv_dir

									else:

									return False

									#对转换的TS文件进行排序   

									def get_sorted_ts(user_path):

									ts_list = glob(os.path.join(user_path,'*.ts'))

									boxer = []

									for ts in ts_list:

									if os.path.exists(ts):

									# print(os.path.splitext(os.path.basename(ts)))

									file,_ = os.path.splitext(os.path.basename(ts))

									print file

									boxer.append(file)

									# boxer.append(int(file))

									boxer.sort()

									print(boxer)

									return boxer

									#文件合并   

									def convert_m3u8(file_path,boxer,o_file_name):

									print u"开始拼接视频"

									new_path=file_path+"/"+u"视频"

									try:

									os.chdir(new_path)

									except Exception, e:

									os.mkdir(new_path)

									tmp = []

									for ts in boxer:

									print ts

									tmp.append(str(ts)+'.ts')

									cmd_str = '+'.join(tmp)

									exec_str = "copy /b "+cmd_str+' '+o_file_name

									print("copy /b "+cmd_str+' '+o_file_name)

									os.chdir(user_path)

									os.system(exec_str)

									print "go home path"

									import shutil

									shutil.move(o_file_name,new_path+"/"+o_file_name)

									os.chdir(file_path)

									reload(sys)

									sys.setdefaultencoding('utf-8')

									# 功能：爬取m3u8格式的视频

									# 检查存储路径是否正常

									def check_path(_path):

									# 判断存储路径是否存在

									if os.path.isdir(_path) or os.path.isabs(_path):

									# 判断存储路径是否为空

									if not os.listdir(_path):

									return _path

									else:

									print u'>>>[-] 目标文件不为空，将清空目标文件，是否更换路径？'

									flag = raw_input('>>>[*] Yes:1 No:2 \n>>>[+] [2]')

									try:

									if flag == '1':

									_path = raw_input(unicode('>>>[+] 请输入目标文件路径。\n>>>[+] ').encode('gbk'))

									check_path(_path)

									else:

									# 清空存储路径

									os.system('rd /S /Q ' + _path)

									os.system('mkdir ' + _path)

									return _path

									except Exception as e:

									print e

									exit(0)

									else:

									os.makedirs(_path)

									return _path

									# 获取ts视频的爬取位置

									def get_url(_url, _path):

									all_url = _url.split('/')

									url_pre = '/'.join(all_url[:-1]) + '/'

									url_next = all_url[-1]

									os.chdir(_path)

									# 获取m3u8文件

									m3u8_txt = requests.get(_url, headers = {'Connection':'close'})

									with open(url_next, 'wb') as m3u8_content:

									m3u8_content.write(m3u8_txt.content)

									# 提取ts视频的url

									movies_url = []

									_urls = open(url_next, 'rb')

									for line in _urls.readlines():

									if '.ts' in line:

									movies_url.append(url_pre + line[:-1])

									else:

									continue

									_urls.close()

									return movies_url

									# 爬取ts视频

									def download_movie(num,movie_url, _path):

									os.chdir(_path)

									print u'>>>[+] 第{}个视频 downloading...'.format(num)

									print '-' * 60

									error_get = []

									for _url in movie_url:

									# ts视频的名称

									movie_name = _url.split('/')[-1][-6:]

									movie_name1 = _url.split('_')[-1][-6:]

									# print movie_name2

									if len(str(movie_name1))<=4:

									movie_name1="00"+str(movie_name1)

									elif len(str(movie_name1))<=5:

									movie_name1="0"+str(movie_name1)

									else :

									movie_name1=str(movie_name1)

									try:

									# 'Connection':'close' 防止请求端口占用

									# timeout=30 防止请求时间超长连接

									movie = requests.get(_url, headers = {'Connection':'close'}, timeout=60)

									with open(movie_name1, 'wb') as movie_content:

									movie_content.writelines(movie)

									print u'>>>[+] 视频片段 ' + movie_name1 + u' 下载完成'

									# 捕获异常，记录失败请求

									except:

									error_get.append(_url)

									continue

									# 如果没有不成功的请求就结束

									if error_get:

									print u'共有%d个请求失败' % len(file_list)

									print '-' * 60

									download_movie(error_get, _path)

									else:

									print '>>>[+] Download successfully!!!'

									url=["http://video.renrenjiang.cn/record/alilive/7684990805-1512186274.m3u8"]

									if __name__ == '__main__':

									try:

									# _url = raw_input(unicode('>>>[+] 请输入指定的[.m3u8]目标URL。\n>>>[+] ').encode('gbk'))

									# _path = raw_input(unicode('>>>[+] 请输入存储目标文件路径。\n>>>[+] ').encode('gbk'))

									for i in range(len(url)):

									_url=url[i]

									_path=os.getcwd()+"/"+"ts"+str(i+5)

									try:

									os.chdir(_path)

									except Exception, e:

									os.mkdir(_path)

									storage_path = check_path(_path)

									movie_url = get_url(_url, storage_path)

									download_movie(i+1,movie_url, storage_path)#下载视频

									user_path =_path

									o_file_name=str(i+5)+".mp4"

									boxer = get_sorted_ts(user_path)

									convert_m3u8(file_path,boxer,o_file_name)#拼接视频

									except Exception as e:

									print e