From 0ea4fe43e7a2f39dfdbdb45597adf20e469ca7ef Mon Sep 17 00:00:00 2001 From: wbt5 Date: Thu, 25 Jun 2020 17:11:27 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E4=BC=81=E9=B9=85=E7=94=B5?= =?UTF-8?q?=E7=AB=9E=E5=BC=B9=E5=B9=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 4 +- danmu/danmaku/__init__.py | 4 +- danmu/danmaku/egame.py | 350 ++++++++++++++++++++++++++++++++++++++ danmu/main.py | 2 + 4 files changed, 358 insertions(+), 2 deletions(-) create mode 100644 danmu/danmaku/egame.py diff --git a/README.md b/README.md index 078fcdb..90741da 100644 --- a/README.md +++ b/README.md @@ -23,7 +23,9 @@ ## 更新 -### 2020.06.19：新增火猫直播弹幕获取 +### 2020.06.25：新增🐧企鹅电竞弹幕获取 + +2020.06.19：新增火猫直播弹幕获取 2020.06.18：新增弹幕功能 diff --git a/danmu/danmaku/__init__.py b/danmu/danmaku/__init__.py index cd26fa4..3b50996 100644 --- a/danmu/danmaku/__init__.py +++ b/danmu/danmaku/__init__.py @@ -7,6 +7,7 @@ from .douyu import Douyu from .huya import Huya from .kuaishou import KuaiShou from .huomao import HuoMao +from .egame import eGame __all__ = ['DanmakuClient'] @@ -28,7 +29,8 @@ class DanmakuClient: 'live.bilibili.com': Bilibili, 'huya.com': Huya, 'huomao.com': HuoMao, - 'kuaishou.com': KuaiShou,}.items(): + 'kuaishou.com': KuaiShou, + 'egame.qq.com': eGame}.items(): if re.match(r'^(?:http[s]?://)?.*?%s/(.+?)$' % u, url): self.__site = s break diff --git a/danmu/danmaku/egame.py b/danmu/danmaku/egame.py new file mode 100644 index 0000000..36e7f57 --- /dev/null +++ b/danmu/danmaku/egame.py @@ -0,0 +1,350 @@ +import aiohttp +import struct +import json +import re + + +class eGame: + heartbeats = b'\x00\x00\x00\x12\x00\x12\x00\x01\x00\x07\x00\x00\x00\x01\x00\x00\x00\x00' + heartbeatInterval = 60 + + @staticmethod + async def get_ws_info(url): + rid = url.split('/')[-1] + page_id = aid = rid + headers = { + 'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A372 Safari/604.1' + } + async with aiohttp.ClientSession() as session: + + async with session.get('https://m.egame.qq.com/live?anchorid' + rid, headers=headers) as resp: + res = await resp.text() + res_ = re.findall(r'"videoInfo":(.*),"h5Url"', res)[0] + str_id = json.loads(res_)['pid'] + params = { + 'param': json.dumps({"0":{"module":"pgg.ws_token_go_svr.DefObj","method":"get_token","param":{"scene_flag":16,"subinfo":{"page":{"scene":1,"page_id":int(page_id),"str_id":str(str_id),"msg_type_list":[1,2]}},"version":1,"message_seq":-1,"dc_param":{"params":{"info":{"aid":aid}},"position":{"page_id":"QG_HEARTBEAT_PAGE_LIVE_ROOM"},"refer":{}},"other_uid":0}}}) + } + + async with session.post('https://share.egame.qq.com/cgi-bin/pgg_async_fcgi', data=params, headers=headers) as resp: + res = json.loads(await resp.text()) + token = res['data']['0']['retBody']['data']['token'] + # 开始拼接reg_datas + reg_datas = [] + tokenbuf = token.encode('ascii') + bodybuf = struct.pack('!Bi', 7, len(tokenbuf)) + tokenbuf + headerbuf = struct.pack('!ihhhihh', 18 + len(bodybuf), 18, 1, 1, 0, 0, 0) + data = headerbuf + bodybuf + reg_datas.append(data) + reg_datas.append(eGame.heartbeats) + + return 'wss://barragepush.egame.qq.com/sub', reg_datas + + @staticmethod + def decode_msg(data): + """ + type: 0、3、9用户发言；7、33礼物信息；29、35欢迎信息；24、31系统提醒；23关注信息 + """ + msgs = [] + msg = {} + s = MessageDecode(data) + body = s.v()['body'] + if body: + bin_datas = body['bin_data'] + for bin_data in bin_datas: + if bin_data['type'] in (0, 3, 9): + msg['name'] = bin_data['nick'] + msg['content'] = bin_data['content'] + msg['msg_type'] = 'danmaku' + else: + msg = {'name': '', 'content': '', 'msg_type': 'other'} + msgs.append(msg.copy()) + return msgs + else: + msg = {'name': '', 'content': '', 'msg_type': 'None'} + msgs.append(msg.copy()) + return msgs + + +class MessageDecode: + """ + 数据解包，还原JS中的操作步骤 + """ + + def __init__(self, data): + + self.data = data + + self.ie = { + 'event_id': 0, + 'msg_type': 1, + 'bin_data': 2, + 'params': 3, + 'start_tm': 4, + 'data_list': 6, + 'end_tm': 5, + 'message_seq': 7, + } + + self.ne = { + 'uid': 0, + 'msgid': 1, + 'nick': 2, + 'content': 3, + 'tm': 4, + 'type': 5, + 'scenes_flag': 6, + 'ext': 7, + 'send_scenes': 8 + } + + self.oe = { + 'event_id': 0, + 'event_name': 1, + 'info': 2, + 'params': 3, + 'bin_data': 4 + } + + def v(self): + data = self.data + startPosition = 18 + endPosition, = struct.unpack_from('!i', data, 0) + seq, = struct.unpack_from('!i', data, 10) + operation, = struct.unpack_from('!h', data, 8) + + if endPosition != len(data): + raise Exception('The received packet length is abnormal') + + return { + 'seq': seq, + 'operation': operation, + 'body': self.w(operation, startPosition, endPosition, data) + } + + def w(self, operation, startPosition, endPosition, data): + if operation == 3: + return self.x(startPosition, endPosition, data) + else: + return None + + def x(self, startPosition, endPosition, data): + i, = struct.unpack_from('!i', data, startPosition) + n = data[startPosition: endPosition] + if len(n) >= (4 + i): + o = n[4:(4 + i)] + a = self.S(o) + y = self.ye(a) + return y + else: + return None + + def ye(self, e): + return self.T({ + 'resultObj': e, + 'template': self.ie, + 'afterChange': 1, + + }) + + def afterChange(self, e, t, i, n, o): + if t == 'bin_data': + v = [] + ve = {} + for m in n: + a = self.S(e, m['ext']) + b = o['msg_type'] + if b == 1: + ve = self.T({ + 'resultObj': a, + 'template': self.ne + }) + elif b == 2: + ve = self.T({ + 'resultObj': a, + 'template': self.oe + }) + v.append(ve.copy()) + return v + else: + return n + + def T(self, e): + i = e['resultObj'] + n = e['template'] + o = e.get('beforeChange', '') + r = e.get('afterChange', '') + a = {} + for s in n.keys(): + for t in i[0]: + if t['tag'] == n[s]: + q = t + p = q['value'] + c = q['ext'] + if r: + a[s] = self.afterChange(i[1], s, c, p, a) + else: + a[s] = p + break + return a + + def S(self, e, t=0): + if t == '': + t = 0 + i = [] + n = len(e) + while t < n: + o = self.m(e, t) + dict_ = { + 'value': o['value'], + 'lastPosition': o['position'], + 'ext': o['ext'], + 'tag': o['tag'] + } + i.append(dict_.copy()) + t = o['position'] + return i, e + + def m(self, e, t): + value = position = ext = '' + i = e + a, = struct.unpack_from('!B', i, t) + tag = (240 & a) >> 4 + type = 15 & a + s_position = t + 1 + + if type == 0: + value, position = self.f0(i, s_position) + elif type == 1: + value, position = self.f1(i, s_position) + elif type == 2: + value, position = self.f2(i, s_position) + elif type == 3: + value, position = self.f3(i, s_position) + elif type == 6: + value, position, ext = self.f6(i, s_position) + elif type == 7: + value, position, ext = self.f7(i, s_position) + elif type == 8: + value, position = self.f8(i, s_position) + elif type == 9: + value, position = self.f9(i, s_position) + elif type == 12: + value, position = self.f12(i, s_position) + + i = '' + + return { + 'i': i, + 'tag': tag, + 'type': type, + 'value': value, + 'position': position, + 'ext': ext + } + + def f0(self, e, t): + o = 1 + try: + n, = struct.unpack_from('!B', e, t) + except: + n = '' + return n, t + o + + def f1(self, e, t): + o = 2 + try: + n, = struct.unpack_from('!H', e, t) + except: + n = '' + return n, t + o + + def f2(self, e, t): + o = 4 + try: + n, = struct.unpack_from('!I', e, t) + except: + n = '' + return n, t + o + + def f3(self, e, t): + e = struct.unpack('!8B', e[t:t + 8]) + i = (e[0] << 24) + (e[1] << 16) + (e[2] << 8) + e[3] + o = (e[4] << 24) + (e[5] << 16) + (e[6] << 8) + e[7] + value = (i << 32) + o + position = t + 8 + return value, position + + def f4(self, e, t): + o = 4 + try: + n, = struct.unpack_from('!f', e, t) + except: + n = '' + return n, t + o + + def f5(self, e, t): + o = 8 + try: + n, = struct.unpack_from('!d', e, t) + except: + n = '' + return n, t + o + + def f6(self, e, t): + n, = struct.unpack_from('!B', e, t) + r = t + 1 + s = r + n + value = (e[r:s]).decode('utf8', errors='ignore') + return value, s, r + + def f7(self, e, t): + n, = struct.unpack_from('!I', e, t) + r = t + 4 + s = r + n + value = (e[r:s]).decode('utf8', errors='ignore') + return value, s, r + + def f8(self, e, t): + i = {} + b = self.m(e, t) + o = b['value'] + r = b['position'] + while o > 0: + a = self.m(e, r) + s = self.m(e, a['position']) + if a['tag'] == 0 and s['tag'] == 1: + i[a['value']] = s['value'] + r = s['position'] + o -= 1 + return i, r + + def f9(self, e, t): + i = self.m(e, t) + n = i['value'] + o = i['position'] + r = [] + while n > 0: + a = self.m(e, o) + r.append(a.copy()) + o = a['position'] + n -= 1 + return r, o + + def f10(self, e, t): + i = [] + while True: + n = self.m(e, t) + t = n['position'] + if n['type'] == 11: + return i, t + i.append(n['value'].copy()) + + def f11(self, e, t): + return '', t + + def f12(self, e, t): + return 0, t + + def f13(self, e, t): + i = self.m(e, t) + return e[(t + i['position']):i['value']], t + i['position'] + i['value'] diff --git a/danmu/main.py b/danmu/main.py index aeeab8d..a066d80 100644 --- a/danmu/main.py +++ b/danmu/main.py @@ -27,3 +27,5 @@ asyncio.run(main(a)) # 斗鱼：https://www.douyu.com/85894 # B站：https://live.bilibili.com/70155 # 快手：https://live.kuaishou.com/u/jjworld126 +# 火猫： +# 企鹅电竞：https://egame.qq.com/383204988