From 3c51cb5591fe314eaf12173249665cd147a37ea6 Mon Sep 17 00:00:00 2001 From: wbt5 Date: Sat, 12 Jun 2021 11:35:15 +0800 Subject: [PATCH] =?UTF-8?q?:bug:=20Fix=20=E6=96=97=E9=B1=BC=E7=9B=B4?= =?UTF-8?q?=E6=92=AD=20(#185=20#189=20#191=20#193=20#195=20#198=20#200)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 感谢两位大佬@wjxgzz @4bbu6j5885o3gpv6ss8找到的的CDN和@moozik @BoneAsh @wangyusen提交的代码 --- douyu.py | 70 ++++++++++++++++++++++++++++++++++---------------------- 1 file changed, 43 insertions(+), 27 deletions(-) diff --git a/douyu.py b/douyu.py index 716c688..ca9710c 100644 --- a/douyu.py +++ b/douyu.py @@ -1,17 +1,31 @@ -# 获取斗鱼直播间的真实流媒体地址，默认最高画质。 +# 获取斗鱼直播间的真实流媒体地址，默认最高画质 +# 使用 https://github.com/wbt5/real-url/issues/185 中两位大佬@wjxgzz @4bbu6j5885o3gpv6ss8找到的的CDN，在此感谢！ import hashlib import re import time -import json import execjs import requests class DouYu: + """ + 可用来替换返回链接中的主机部分 + 两个阿里的CDN： + dyscdnali1.douyucdn.cn + dyscdnali3.douyucdn.cn + 墙外不用带尾巴的akm cdn： + hls3-akm.douyucdn.cn + hlsa-akm.douyucdn.cn + hls1a-akm.douyucdn.cn + """ def __init__(self, rid): - # 房间号通常为1~7位纯数字，浏览器地址栏中看到的房间号不一定是真实rid. + """ + 房间号通常为1~7位纯数字，浏览器地址栏中看到的房间号不一定是真实rid. + Args: + rid: + """ self.did = '10000000000000000000000000001501' self.t10 = str(int(time.time())) self.t13 = str(int((time.time() * 1000))) @@ -25,9 +39,6 @@ class DouYu: else: raise Exception('房间号错误') - def set_rate(self, rate): - self.live_rate = rate - @staticmethod def md5(data): return hashlib.md5(data.encode('utf-8')).hexdigest() @@ -45,9 +56,13 @@ class DouYu: 'auth': auth } res = self.s.post(url, headers=headers, data=data).json() - if 0 != res['error']: - raise Exception(res["msg"]) - return res['data']['rtmp_url'] + '/' + res['data']['rtmp_live'] + error = res['error'] + data = res['data'] + key = '' + if data: + rtmp_live = data['rtmp_live'] + key = re.search(r'(\d{1,7}[0-9a-zA-Z]+)_?\d{0,4}(/playlist|.m3u8)', rtmp_live).group(1) + return error, key def get_js(self): result = re.search(r'(function ub98484234.*)\s(var.*)', self.res).group() @@ -64,16 +79,15 @@ class DouYu: js = execjs.compile(func_sign) params = js.call('sign', self.rid, self.did, self.t10) - # params += '&ver=219032101&rid={}&rate=-1'.format(self.rid) - params += '&ver=219032101&rid={}&rate={}'.format(self.rid, self.live_rate) + params += '&ver=219032101&rid={}&rate=-1'.format(self.rid) url = 'https://m.douyu.com/api/room/ratestream' - res = self.s.post(url, params=params).json() - if 0 != res['code']: - return None - return res['data']['url'] + res = self.s.post(url, params=params).text + key = re.search(r'(\d{1,7}[0-9a-zA-Z]+)_?\d{0,4}(.m3u8|/playlist)', res).group(1) - def get_pc_js(self, cdn='ws-h5'): + return key + + def get_pc_js(self, cdn='ws-h5', rate=0): """ 通过PC网页端的接口获取完整直播源。 :param cdn: 主线路ws-h5、备用线路tct-h5 @@ -96,24 +110,26 @@ class DouYu: js = execjs.compile(func_sign) params = js.call('sign', self.rid, self.did, self.t10) - params += '&cdn={}&rate={}'.format(cdn, self.live_rate) + params += '&cdn={}&rate={}'.format(cdn, rate) url = 'https://www.douyu.com/lapi/live/getH5Play/{}'.format(self.rid) res = self.s.post(url, params=params).json() - if 0 != res['error']: - return None - return res['data']['rtmp_url'] + '/' + res['data']['rtmp_live'] + + return res def get_real_url(self): - ret = [] - ret.append(self.get_pre()) - ret.append(self.get_js()) - ret.append(self.get_pc_js()) - return ret + error, key = self.get_pre() + if error == 0: + pass + elif error == 102: + raise Exception('房间不存在') + elif error == 104: + raise Exception('房间未开播') + else: + key = self.get_js() + return "http://dyscdnali1.douyucdn.cn/live/{}.flv?uuid=".format(key) if __name__ == '__main__': r = input('输入斗鱼直播间号：\n') s = DouYu(r) - r = input('输入画质(1流畅；2高清；3超清；4蓝光4M；0蓝光8M或10M)：\n') - s.set_rate(r) print(s.get_real_url())