新增搜狐千帆直播弹幕

danmu/danmaku/qf.py

@@ -0,0 +1,65 @@
+import aiohttp
+import json
+import time
+
+
+class QF:
+    heartbeat = '2::'
+    heartbeatInterval = 30
+
+    @staticmethod
+    async def get_ws_info(url):
+        rid = url.split('/')[-1]
+        async with aiohttp.ClientSession() as session:
+            async with session.get('https://conn-chat.qf.56.com/socket.io/1/') as resp:
+                res = await resp.text()
+                sessid = res.split(':')[0]
+                ws_url = 'wss://conn-chat.qf.56.com/socket.io/1/websocket/' + sessid
+                # e = 2
+                t = 'connector-sio.entryHandler.enter'
+                s = {
+                    'userId': '',
+                    'aq': 0,
+                    'roomId': rid,
+                    'token': '',
+                    'ip': '',
+                    'recet': 0,
+                    'params': {
+                        'referFrom': '0'
+                    },
+                    'apType': 0,
+                    'timestamp': int(time.time() * 1e3)
+                }
+                r = json.dumps(s, separators=(',', ':'))
+                if len(t) > 255:
+                    raise Exception('route maxlength is overflow')
+                reg_data = '3:::' + '\x00\x00\x00\x02 ' + t + r
+        return ws_url, [reg_data]
+
+    @staticmethod
+    def decode_msg(message):
+        msgs = []
+        msg = {'name': '', 'content': '', 'msg_type': 'other'}
+        type_ = message[0]
+        if type_ == '3':
+            data = json.loads(message[4:])
+            route = data.get('route', 0)
+            body = data['body']
+            if route == 'onUserLog':  # 入场信息
+                msg['name'] = 'SYS'
+                msg['content'] = body['userName'] + ' 来了'
+                msg['msg_type'] = 'danmaku'
+            elif route == 'onChat':  # 弹幕
+                msg['name'] = body['userName']
+                msg['content'] = body['content']
+                msg['msg_type'] = 'danmaku'
+            elif route == 'onGift':  # 弹幕
+                msg['name'] = 'SYS'
+                msg['content'] = body['userName'] + ' 送礼物 ' + body['giftName']
+                msg['msg_type'] = 'danmaku'
+            elif route == 'onBc':  # 弹幕
+                msg['name'] = 'SYS'
+                msg['content'] = body['userName'] + '：' + body['msg']
+                msg['msg_type'] = 'danmaku'
+        msgs.append(msg.copy())
+        return msgs