88 lines
3.8 KiB
Python
88 lines
3.8 KiB
Python
|
||
import asyncio
|
||
import websockets
|
||
import time
|
||
import json
|
||
import datetime
|
||
|
||
|
||
|
||
|
||
async def connect():
|
||
uri = "wss://live-ws.immomo.com/ws/im"
|
||
async with websockets.connect(uri) as websocket:
|
||
print(f"链接websocket:[{uri}] 完成")
|
||
|
||
momoid = "1062302597"
|
||
roomid = "17066184578990"
|
||
token = "b458d6fd04ee38a9e5330e76a3802df9"
|
||
ping_msg_id = 2 # 心跳包的id2
|
||
|
||
client_time = str(int(time.time() * 1000))
|
||
data = '{{"msg_id":1,"client_time":{0},"type":"Sauth","data":{{"momoid":"{1}","roomid":"{2}","role":6,"isVisitor":false,"token":"{3}","ua":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36 Edg/122.0.0.0"}}}}'.format(
|
||
client_time, momoid, roomid, token)
|
||
await websocket.send(data)
|
||
print("Send Sauth")
|
||
|
||
while True:
|
||
try:
|
||
response = await websocket.recv()
|
||
current_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||
#print(f"[{current_time}] 收到原始消息 => " + response)
|
||
# print("收到原始消息 => " + response)
|
||
|
||
try:
|
||
json_data = json.loads(response)
|
||
# print("收到JSON消息 => ", json_data)
|
||
|
||
message_type = json_data['groups'][0]['units'][0]['type']
|
||
|
||
if message_type == "Message":
|
||
nick = json_data['groups'][0]['nick']
|
||
text = json_data['groups'][0]['units'][0]['Msg.Message.data']['text']
|
||
print(f"[{current_time}] {nick} : {text}")
|
||
elif message_type == "EnterRoom":
|
||
text = json_data['groups'][0]['units'][0]['Set.EnterRoom.data']['text']
|
||
print(f"[{current_time}] 有新人: {text}")
|
||
elif message_type == "BiliBili":
|
||
nick = json_data['groups'][0]['nick']
|
||
text = json_data['groups'][0]['units'][0]['Pay.BiliBili.data']['text']
|
||
print(f"[{current_time}] {nick} : {text}")
|
||
elif message_type == "GiftV3":
|
||
nick = json_data['groups'][0]['nick']
|
||
text = json_data['groups'][0]['units'][0]['GiftV3.giftV3']['text']
|
||
product_id = json_data['groups'][0]['units'][0]['GiftV3.giftV3']['productId']
|
||
print(f"[{current_time}] 用户: {nick} 送出礼物，内容为: {text}，产品ID为: {product_id}")
|
||
|
||
except KeyError as e:
|
||
print("JSON数据缺少:", e)
|
||
except json.JSONDecodeError as e:
|
||
print("无法解析JSON格式: ", e)
|
||
|
||
except websockets.exceptions.ConnectionClosedError:
|
||
print(f"[{current_time}] 与服务器断开链接")
|
||
break
|
||
|
||
async def send_ping(websocket):
|
||
ping_msg_id = 2
|
||
await asyncio.sleep(30) # 延迟30秒后开始发送心跳包
|
||
while True:
|
||
client_time = str(int(time.time() * 1000))
|
||
ping_data = '{{"msg_id":{0},"client_time":{1},"type":"Ping","data":{{}}}}'.format(ping_msg_id, client_time)
|
||
try:
|
||
await websocket.send(ping_data)
|
||
print("Send Ping", ping_data)
|
||
ping_msg_id += 1
|
||
await asyncio.sleep(30) # 每隔30秒发送一次
|
||
except websockets.exceptions.ConnectionClosedError:
|
||
print("与服务器断开链接")
|
||
break
|
||
|
||
async def main():
|
||
websocket = await websockets.connect("wss://live-ws.immomo.com/ws/im")
|
||
connect_task = asyncio.create_task(connect())
|
||
ping_task = asyncio.create_task(send_ping(websocket))
|
||
await connect_task
|
||
await ping_task
|
||
|
||
asyncio.run(main()) |