新增企鹅电竞弹幕

README.md

@@ -23,7 +23,9 @@
 
 ## 更新
 
-### 2020.06.19：新增火猫直播弹幕获取
+### 2020.06.25：新增🐧企鹅电竞弹幕获取
+
+2020.06.19：新增火猫直播弹幕获取
 
 2020.06.18：新增弹幕功能
 

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

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']

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