mirror of
https://github.com/wbt5/real-url.git
synced 2025-08-01 14:48:01 +08:00
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
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,8})', 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
|