PublicMirrors/you-get
1
0
Fork 0
mirror of https://github.com/soimort/you-get.git synced 2025-02-03 08:43:58 +03:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

fix CI failure by using 3rd packages

Browse Source
This commit is contained in:
steven7851 2016-07-29 18:25:40 +08:00 committed by GitHub
parent 6689f45469
commit 06b57e2db7
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/you_get/extractors/douyutv.py
View File

@ -8,13 +8,13 @@ import hashlib
import time import time
import random import random
import string import string
import requests import urllib.parse, urllib.request
def douyutv_download(url, output_dir = '.', merge = True, info_only = False, **kwargs): def douyutv_download(url, output_dir = '.', merge = True, info_only = False, **kwargs):
room_id = url[url.rfind('/')+1:] room_id = url[url.rfind('/')+1:]
json_request_url = "http://m.douyu.com/html5/live?roomId=%s" % room_id json_request_url = "http://m.douyu.com/html5/live?roomId=%s" % room_id
content = get_html(json_request_url) content = get_content(json_request_url)
data = json.loads(content)['data'] data = json.loads(content)['data']
server_status = data.get('error',0) server_status = data.get('error',0)
if server_status is not 0: if server_status is not 0:
@ -27,14 +27,19 @@ def douyutv_download(url, output_dir = '.', merge = True, info_only = False, **k
tt = int(time.time() / 60) tt = int(time.time() / 60)
did = ''.join([random.choice(string.ascii_uppercase + string.digits) for n in range(32)]) did = ''.join([random.choice(string.ascii_uppercase + string.digits) for n in range(32)])
sign = hashlib.md5((room_id + did + 'A12Svb&%1UUmf@hC' + "%d" % tt).encode("utf-8")).hexdigest() sign_content = '{room_id}{did}A12Svb&%1UUmf@hC{tt}'.format(room_id = room_id, did = did, tt = tt)
sign = hashlib.md5(sign_content.encode('utf-8')).hexdigest()
json_request_url = "http://www.douyu.com/lapi/live/getPlay/%s" % room_id json_request_url = "http://www.douyu.com/lapi/live/getPlay/%s" % room_id
payload = {'cdn': 'ws', 'rate': '0', 'tt': tt, 'did': did, 'sign': sign} payload = {'cdn': 'ws', 'rate': '0', 'tt': tt, 'did': did, 'sign': sign}
r = requests.post(json_request_url, data=payload) json_data = json.dumps(payload).encode('utf8')
content = r.json() postdata = urllib.parse.urlencode(payload)
data = content['data'] req = urllib.request.Request(json_request_url, postdata.encode('utf-8'))
with urllib.request.urlopen(req) as response:
the_page = response.read()
content = json.loads(the_page.decode('utf-8'))
data = content['data']
server_status = data.get('error',0) server_status = data.get('error',0)
if server_status is not 0: if server_status is not 0:
raise ValueError("Server returned error:%s" % server_status) raise ValueError("Server returned error:%s" % server_status)
@ -43,6 +48,7 @@ def douyutv_download(url, output_dir = '.', merge = True, info_only = False, **k
print_info(site_info, title, 'flv', float('inf')) print_info(site_info, title, 'flv', float('inf'))
if not info_only: if not info_only:
print(real_url)
download_url_ffmpeg(real_url, title, 'flv', None, output_dir, merge = merge) download_url_ffmpeg(real_url, title, 'flv', None, output_dir, merge = merge)
site_info = "douyu.com" site_info = "douyu.com"