you-get/src/you_get/extractors/yizhibo.py
2016-12-11 01:49:13 +01:00

38 lines
1.3 KiB
Python

#!/usr/bin/env python
__all__ = ['yizhibo_download']
from ..common import *
import json
import time
def yizhibo_download(url, output_dir = '.', merge = True, info_only = False, **kwargs):
video_id = url[url.rfind('/')+1:].split(".")[0]
json_request_url = 'http://www.yizhibo.com/live/h5api/get_basic_live_info?scid={}'.format(video_id)
content = get_content(json_request_url)
error = json.loads(content)['result']
if (error != 1):
raise ValueError("Error : {}".format(error))
data = json.loads(content)
title = data.get('data')['live_title']
if (title == ''):
title = data.get('data')['nickname']
m3u8_url = data.get('data')['play_url']
m3u8 = get_content(m3u8_url)
base_url = "/".join(data.get('data')['play_url'].split("/")[:7])+"/"
part_url = re.findall(r'([0-9]+\.ts)', m3u8)
real_url = []
for i in part_url:
url = base_url + i
real_url.append(url)
print_info(site_info, title, 'ts', float('inf'))
if not info_only:
if player:
launch_player(player, [m3u8_url])
download_urls(real_url, title, 'ts', float('inf'), output_dir, merge = merge)
site_info = "yizhibo.com"
download = yizhibo_download
download_playlist = playlist_not_supported('yizhibo')