81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
|
import asyncio
|
|||
|
|
import requests
|
|||
|
|
import websockets
|
|||
|
|
import time
|
|||
|
|
import json
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
# 构建API请求链接
|
|||
|
|
room_id = "17066184578990"
|
|||
|
|
timestamp = int(time.time() * 1000)
|
|||
|
|
api_url = f"https://live-api.immomo.com/open/m/room/profile?roomid={room_id}&momoid=&src=m50011&web_uid=11126216061Xazx3TgC&_= {timestamp}"
|
|||
|
|
|
|||
|
|
# 发起API请求
|
|||
|
|
response = requests.get(api_url)
|
|||
|
|
data = response.json()
|
|||
|
|
ws_token = data["data"].get("ws_token", "")
|
|||
|
|
umomoid = data["data"]["momoid"]
|
|||
|
|
|
|||
|
|
async def connect():
|
|||
|
|
uri = "wss://live-ws.immomo.com/ws/im"
|
|||
|
|
async with websockets.connect(uri) as websocket:
|
|||
|
|
print(f"链接websocket:[{uri}] 完成")
|
|||
|
|
|
|||
|
|
momoid = umomoid
|
|||
|
|
roomid = room_id
|
|||
|
|
token = f"{ws_token}"
|
|||
|
|
|
|||
|
|
data = '{{"msg_id":1,"client_time":{0},"type":"Sauth","data":{{"momoid":"{1}","roomid":"{2}","token":"{3}","barType":"NORMAL"}}}}'.format(
|
|||
|
|
timestamp, momoid, roomid, token)
|
|||
|
|
print(data)
|
|||
|
|
|
|||
|
|
await websocket.send(data)
|
|||
|
|
|
|||
|
|
print("Send Hello")
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
response = await websocket.recv()
|
|||
|
|
current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|||
|
|
#print(f"[{current_time}] 收到原始消息 => " + response)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
json_data = json.loads(response)
|
|||
|
|
#print(f"[{current_time}] 收到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}")
|
|||
|
|
#else:
|
|||
|
|
# 其他消息类型的处理
|
|||
|
|
#print(f"[{current_time}] 收到未处理的消息类型: {message_type}")
|
|||
|
|
|
|||
|
|
except KeyError as e:
|
|||
|
|
print(f"[{current_time}] JSON数据中缺少必要的键:", e)
|
|||
|
|
except json.JSONDecodeError as e:
|
|||
|
|
print(f"[{current_time}] 无法解析为JSON格式: ", e)
|
|||
|
|
|
|||
|
|
except websockets.exceptions.ConnectionClosedError:
|
|||
|
|
print(f"[{current_time}] 与服务器断开链接")
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
await connect()
|
|||
|
|
|
|||
|
|
asyncio.run(main())
|