diff --git a/handler/utils.py b/handler/utils.py index 2f2ad36..aa8ab92 100644 --- a/handler/utils.py +++ b/handler/utils.py @@ -1,6 +1,7 @@ import os from handler.common import MESSAGE_QUEUE +from messages.control import ControlMessage from protobuf import message_pb2 from protobuf import wss_pb2 import gzip diff --git a/messages/control.py b/messages/control.py new file mode 100644 index 0000000..f435656 --- /dev/null +++ b/messages/control.py @@ -0,0 +1,12 @@ +import time + +from messages.base import Base +from protobuf import message_pb2 + + +class ControlMessage(Base): + def __init__(self): + self.instance = message_pb2.ChatMessage() + + def __str__(self): + return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + '【直播间信息】' diff --git a/output/IOutput.py b/output/IOutput.py index 1491288..7006c06 100644 --- a/output/IOutput.py +++ b/output/IOutput.py @@ -1,5 +1,6 @@ from messages.base import Base from messages.chat import ChatMessage +from messages.control import ControlMessage from messages.gift import GiftMessage from messages.like import LikeMessage from messages.member import MemberMessage @@ -29,8 +30,11 @@ class IOutput(): def userseq_output(self, message: RoomUserSeqMessage): ... + def control_output(self, message: ControlMessage): + ... + def other_output(self, message_type: str, message_raw: bytes): ... - def error_output(self, exception: Exception): + def error_output(self, message_type: str, message_raw: bytes, exception: Exception): ... \ No newline at end of file