python爬取m3u8连接的视频

本文为大家分享了python爬取m3u8连接的视频方法，供大家参考，具体内容如下要求：输入m3u8

本文为大家分享了python爬取m3u8连接的视频方法，供大家参考，具体内容如下

要求：输入m3u8所在url，且ts视频与其在同一路径下

#!/usr/bin/env/python
#_*_coding:utf-8_*_
#Data:17-10-08
#Auther:苏莫
#Link:http://blog.csdn.net/lingluofengzang
#PythonVersion:python2.7
#filename:download_movie.py

import os
import sys
import requests

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(movie_url, _path):
 os.chdir(_path)
 print '>>>[+] downloading...'
 print '-' * 60
 error_get = []

 for _url in movie_url:
 # ts视频的名称
 movie_name = _url.split('/')[-1][-6:]

 try:
 # 'Connection':'close' 防止请求端口占用
 # timeout=30 防止请求时间超长连接
 movie = requests.get(_url, headers = {'Connection':'close'}, timeout=60)
 with open(movie_name, 'wb') as movie_content:
 movie_content.writelines(movie)
 print '>>>[+] File ' + movie_name + ' done'
 # 捕获异常，记录失败请求
 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!!!'

if __name__ == '__main__':
 try:

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

 storage_path = check_path(_path)
 movie_url = get_url(_url, storage_path)
 download_movie(movie_url, storage_path)

 except Exception as e:
 print e

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。