python缓存直播到到本地

这里用了一个很好用的包，streamlink,可在github上查看源代码~
几乎所有直播平台都可以缓存~

from streamlink import Streamlink
import streamlink
from urllib import request
import os,time
movie=input('[INFO]输入你要下载的直播视频地址>>>>')
if not os.path.exists('D:/zhibo/'):
    os.makedirs('D:/zhibo/')
print('文件将会下载到D:/zhibo/')
session=Streamlink()#创建一个会话
try:
    streams=session.streams(movie)#在会话里输入url会返回直播的flv缓存地址
except:
    try:
        streams=streamlink.streams(movie)
    except:
        print('[INFO]该网站不存在plug接口')
        exit(0)
# print(streams)
print('[INFO]获取了视屏流地址.....')
list=['source','medium','best','worse']
for l in list:
    if streams[l].url:
        print('[INFO]获得视频%s'%l)
        source=streams[l].url
        if 'm3u8'in str(source):
            print('[ERROR]%s存在m3u8,暂不支持下载，'%l)
            continue
        else:
            print('[INFO]正在缓存直播哦')
            request.urlretrieve(source,os.path.join('D:/zhibo/',str(int(time.time())))+'.flv')
            print('[INFO]您缓存的直播已下播......')
            print('[INFO]按回车键退出缓存')
            input('[INFO]Please Enter!')