|
|
from protobuf.douyin import *
|
|
|
import random
|
|
|
from loguru import logger
|
|
|
from helper import LiveChatConfig
|
|
|
|
|
|
live_chat_config = LiveChatConfig()
|
|
|
|
|
|
|
|
|
def parse_chat_msg(payload, queue):
|
|
|
"""聊天消息"""
|
|
|
if not random.random() < live_chat_config.chat_prob / 100:
|
|
|
return
|
|
|
message = ChatMessage().parse(payload)
|
|
|
user_name = message.user.nick_name
|
|
|
user_id = message.user.id
|
|
|
content = message.content
|
|
|
prompt = live_chat_config.chat_prompt.format(content=content)
|
|
|
queue.put((prompt, content))
|
|
|
# logger.info(f"【聊天msg】[{user_id}]{user_name}: {content}")
|
|
|
# logger.info(f"队列数量: {queue.qsize()}")
|
|
|
|
|
|
|
|
|
def parse_gif_msg(payload, queue):
|
|
|
"""礼物消息"""
|
|
|
if not random.random() < live_chat_config.gift_prob / 100:
|
|
|
return
|
|
|
message = GiftMessage().parse(payload)
|
|
|
user_name = message.user.nick_name
|
|
|
gift_name = message.gift.name
|
|
|
gift_count = message.combo_count
|
|
|
prompt = live_chat_config.gift_prompt.format(user_name=user_name, gift_count=gift_count, gift_name=gift_name)
|
|
|
queue.put((prompt, None))
|
|
|
# logger.info(f"【礼物msg】{user_name} 送出了 {gift_name}x{gift_count}")
|
|
|
# logger.info(f"队列数量: {queue.qsize()}")
|
|
|
|
|
|
|
|
|
def parse_like_msg(payload, queue):
|
|
|
'''点赞消息'''
|
|
|
if not random.random() < live_chat_config.like_prob / 100:
|
|
|
return
|
|
|
message = LikeMessage().parse(payload)
|
|
|
user_name = message.user.nick_name
|
|
|
count = message.count
|
|
|
prompt = live_chat_config.like_prompt.format(user_name=user_name, count=count)
|
|
|
queue.put((prompt, None))
|
|
|
# logger.info(f"【点赞msg】{user_name} 点了{count}个赞")
|
|
|
# logger.info(f"队列数量: {queue.qsize()}")
|
|
|
|
|
|
|
|
|
def parse_member_msg(payload, queue):
|
|
|
'''进入直播间消息'''
|
|
|
if not random.random() < live_chat_config.enter_live_room_prob / 100:
|
|
|
return
|
|
|
message = MemberMessage().parse(payload)
|
|
|
user_name = message.user.nick_name
|
|
|
user_id = message.user.id
|
|
|
gender = message.user.gender
|
|
|
if gender in (0, 1):
|
|
|
gender = ["女", "男"][gender]
|
|
|
prompt = live_chat_config.enter_live_room_prompt.format(user_name=user_name)
|
|
|
queue.put((prompt, None))
|
|
|
# logger.info(f"【进场msg】[{user_id}][{gender}]{user_name} 进入了直播间")
|
|
|
# logger.info(f"队列数量: {queue.qsize()}")
|
|
|
|
|
|
|
|
|
def parse_social_msg(payload, queue):
|
|
|
'''关注消息'''
|
|
|
if not random.random() < live_chat_config.follow_prob / 100:
|
|
|
return
|
|
|
message = SocialMessage().parse(payload)
|
|
|
user_name = message.user.nick_name
|
|
|
user_id = message.user.id
|
|
|
prompt = live_chat_config.follow_prompt.format(user_name=user_name)
|
|
|
queue.put((prompt, None))
|
|
|
# logger.info(f"【关注msg】[{user_id}]{user_name} 关注了主播")
|
|
|
# logger.info(f"队列数量: {queue.qsize()}")
|
|
|
|
|
|
|
|
|
def parse_room_user_seq_msg(payload):
|
|
|
'''直播间统计'''
|
|
|
message = RoomUserSeqMessage().parse(payload)
|
|
|
current = message.total
|
|
|
total = message.total_pv_for_anchor
|
|
|
# print(f"【统计msg】当前观看人数: {current}, 累计观看人数: {total}")
|
|
|
|
|
|
|
|
|
def parse_fansclub_msg(payload):
|
|
|
'''粉丝团消息'''
|
|
|
message = FansclubMessage().parse(payload)
|
|
|
content = message.content
|
|
|
# print(f"【粉丝团msg】 {content}")
|
|
|
|
|
|
|
|
|
def parse_emoji_chat_msg(payload):
|
|
|
'''聊天表情包消息'''
|
|
|
message = EmojiChatMessage().parse(payload)
|
|
|
emoji_id = message.emoji_id
|
|
|
user = message.user
|
|
|
common = message.common
|
|
|
default_content = message.default_content
|
|
|
# print(f"【聊天表情包id】 {emoji_id},user：{user},common:{common},default_content:{default_content}")
|
|
|
|
|
|
|
|
|
def parse_room_msg(payload):
|
|
|
message = RoomMessage().parse(payload)
|
|
|
common = message.common
|
|
|
room_id = common.room_id
|
|
|
# print(f"【直播间msg】直播间id:{room_id}")
|
|
|
|
|
|
|
|
|
def parse_room_stats_msg(payload):
|
|
|
message = RoomStatsMessage().parse(payload)
|
|
|
display_long = message.display_long
|
|
|
# print(f"【直播间统计msg】{display_long}")
|
|
|
|
|
|
|
|
|
def parse_rank_msg(payload):
|
|
|
message = RoomRankMessage().parse(payload)
|
|
|
ranks_list = message.ranks_list
|
|
|
# print(f"【直播间排行榜msg】{ranks_list}")
|
|
|
|
|
|
|
|
|
def parse_room_stream_adaptation_msg(payload):
|
|
|
message = RoomStreamAdaptationMessage().parse(payload)
|
|
|
adaptationType = message.adaptation_type
|
|
|
# print(f'直播间adaptation: {adaptationType}')
|