168 lines
3.8 KiB
Python
168 lines
3.8 KiB
Python
from enum import Enum, IntEnum
|
|
from typing import Optional, Dict, Any, List, Union
|
|
from dataclasses import dataclass, field
|
|
|
|
# 枚举类型定义
|
|
class PackMsgType(IntEnum):
|
|
"""消息类型枚举"""
|
|
无 = 0
|
|
弹幕消息 = 1
|
|
点赞消息 = 2
|
|
进直播间 = 3
|
|
关注消息 = 4
|
|
礼物消息 = 5
|
|
直播间统计 = 6
|
|
粉丝团消息 = 7
|
|
直播间分享 = 8
|
|
下播 = 9
|
|
|
|
class Gender(IntEnum):
|
|
"""性别枚举"""
|
|
未知 = 0
|
|
男 = 1
|
|
女 = 2
|
|
|
|
def to_string(self) -> str:
|
|
"""将性别枚举转换为中文字符串"""
|
|
if self == Gender.男:
|
|
return "男"
|
|
elif self == Gender.女:
|
|
return "女"
|
|
else:
|
|
return "妖"
|
|
|
|
class FansclubType(IntEnum):
|
|
"""粉丝团消息类型"""
|
|
无 = 0
|
|
粉丝团升级 = 1
|
|
加入粉丝团 = 2
|
|
|
|
class ShareType(IntEnum):
|
|
"""直播间分享目标"""
|
|
未知 = 0
|
|
微信 = 1
|
|
朋友圈 = 2
|
|
微博 = 3
|
|
QQ空间 = 4
|
|
QQ = 5
|
|
抖音好友 = 112
|
|
|
|
class EnterType(IntEnum):
|
|
"""观众的进入方式"""
|
|
正常进入 = 0
|
|
通过分享进入 = 6
|
|
|
|
# 数据类定义
|
|
@dataclass
|
|
class FansClubInfo:
|
|
"""粉丝团信息"""
|
|
ClubName: str = ""
|
|
Level: int = 0
|
|
|
|
@dataclass
|
|
class MsgUser:
|
|
"""用户信息"""
|
|
Id: int = 0
|
|
IsAdmin: bool = False
|
|
IsAnchor: bool = False
|
|
ShortId: int = 0
|
|
DisplayId: str = ""
|
|
Nickname: str = "匿名用户"
|
|
Level: int = 0
|
|
PayLevel: int = 0
|
|
Gender: Gender = Gender.未知
|
|
HeadImgUrl: str = ""
|
|
SecUid: str = ""
|
|
FansClub: Optional[FansClubInfo] = None
|
|
FollowerCount: int = 0
|
|
FollowStatus: int = 0
|
|
FollowingCount: int = 0
|
|
|
|
def gender_to_string(self) -> str:
|
|
"""返回性别的中文表示"""
|
|
return Gender(self.Gender).to_string()
|
|
|
|
@dataclass
|
|
class RoomAnchorInfo:
|
|
"""直播间主播信息"""
|
|
UserId: str = ""
|
|
SecUid: str = ""
|
|
Nickname: str = ""
|
|
HeadUrl: str = ""
|
|
FollowStatus: int = 0
|
|
|
|
@dataclass
|
|
class Msg:
|
|
"""基础消息类"""
|
|
MsgId: int = 0
|
|
User: MsgUser = field(default_factory=MsgUser)
|
|
Owner: Optional[RoomAnchorInfo] = None # 修正字段名称
|
|
Content: str = ""
|
|
RoomId: str = ""
|
|
WebRoomId: str = ""
|
|
RoomTitle: str = ""
|
|
IsAnonymous: bool = False
|
|
Appid: str = ""
|
|
|
|
@dataclass
|
|
class LikeMsg(Msg):
|
|
"""点赞消息"""
|
|
Count: int = 0
|
|
Total: int = 0
|
|
|
|
@dataclass
|
|
class GiftMsg(Msg):
|
|
"""礼物消息"""
|
|
GiftId: int = 0
|
|
GiftName: str = ""
|
|
GroupId: int = 0
|
|
GiftCount: int = 0
|
|
RepeatCount: int = 0
|
|
DiamondCount: int = 0 # 抖币价格
|
|
Combo: bool = False # 该礼物是否可连击
|
|
ImgUrl: str = "" # 礼物图片地址
|
|
ToUser: Optional[MsgUser] = None # 送礼目标(连麦直播间有用)
|
|
GiftValue: int = 0 # 在原Python代码中使用的礼物价值字段
|
|
|
|
@dataclass
|
|
class MemberMessage(Msg):
|
|
"""进入直播间消息"""
|
|
CurrentCount: int = 0
|
|
EnterTipType: int = 0
|
|
|
|
@dataclass
|
|
class UserSeqMsg(Msg):
|
|
"""直播间统计消息"""
|
|
OnlineUserCount: int = 0
|
|
TotalUserCount: int = 0
|
|
TotalUserCountStr: str = "0"
|
|
OnlineUserCountStr: str = "0"
|
|
|
|
@dataclass
|
|
class FansclubMsg(Msg):
|
|
"""粉丝团消息"""
|
|
Type: int = 0 # 粉丝团消息类型，升级1，加入2
|
|
Level: int = 0 # 粉丝团等级
|
|
FansClubName: str = "" # 粉丝团名称
|
|
|
|
@dataclass
|
|
class ShareMessage(Msg):
|
|
"""直播间分享"""
|
|
ShareType: ShareType = ShareType.未知
|
|
|
|
# 全局统计数据类
|
|
@dataclass
|
|
class LiveStats:
|
|
"""直播间统计数据"""
|
|
total_likes: int = 0
|
|
total_users: int = 0
|
|
male_users: int = 0
|
|
female_users: int = 0
|
|
|
|
def reset(self) -> None:
|
|
"""重置统计数据"""
|
|
self.total_likes = 0
|
|
self.total_users = 0
|
|
self.male_users = 0
|
|
self.female_users = 0
|