斗鱼直播

danmu/danmaku/douyu.py

@@ -0,0 +1,48 @@
+import re
+import json
+import aiohttp
+from struct import pack
+
+
+class Douyu:
+    wss_url = 'wss://danmuproxy.douyu.com:8503/'
+    heartbeat = b'\x14\x00\x00\x00\x14\x00\x00\x00\xb1\x02\x00\x00\x74\x79\x70\x65\x40\x3d\x6d\x72\x6b\x6c' \
+                b'\x2f\x00 '
+    heartbeatInterval = 60
+
+    @staticmethod
+    async def get_ws_info(url):
+        room_id = url.split('/')[-1]
+        async with aiohttp.ClientSession() as session:
+            async with session.get('https://m.douyu.com/' + str(room_id)) as resp:
+                room_page = await resp.text()
+                room_id = re.findall(r'"rid":(\d{1,7})', room_page)[0]
+        reg_datas = []
+        data = f'type@=loginreq/roomid@={room_id}/'
+        s = pack('i', 9 + len(data)) * 2
+        s += b'\xb1\x02\x00\x00'  # 689
+        s += data.encode('ascii') + b'\x00'
+        reg_datas.append(s)
+        data = f'type@=joingroup/rid@={room_id}/gid@=-9999/'
+        s = pack('i', 9 + len(data)) * 2
+        s += b'\xb1\x02\x00\x00'  # 689
+        s += data.encode('ascii') + b'\x00'
+        reg_datas.append(s)
+        return Douyu.wss_url, reg_datas
+
+    @staticmethod
+    def decode_msg(data):
+        msgs = []
+        for msg in re.findall(b'(type@=.*?)\x00', data):
+            try:
+                msg = msg.replace(b'@=', b'":"').replace(b'/', b'","')
+                msg = msg.replace(b'@A', b'@').replace(b'@S', b'/')
+                msg = json.loads((b'{"' + msg[:-2] + b'}').decode('utf8', 'ignore'))
+                msg['name'] = msg.get('nn', '')
+                msg['content'] = msg.get('txt', '')
+                msg['msg_type'] = {'dgb': 'gift', 'chatmsg': 'danmaku',
+                                   'uenter': 'enter'}.get(msg['type'], 'other')
+                msgs.append(msg)
+            except Exception as e:
+                pass
+        return msgs