增加禁言消息

This commit is contained in:
Jerry Yan 2019-01-22 00:13:49 +08:00
parent 5d5a0a9432
commit e3da5d98b3
4 changed files with 46 additions and 19 deletions

32
MemberMsg.py Normal file
View File

@ -0,0 +1,32 @@
from User import User
class MemberMsg:
type:int = -1
content:str = ""
user:User = None
def __init__(self, json=None):
if json:
self.parse(json)
def parse(self, json):
self.user = User(json)
if "Msg" in json:
if "action" in json["Msg"]:
self.type = json["Msg"]['action']
elif "content" in json["Msg"]:
self.content = json["Msg"]['content']
def __str__(self):
if self.type == 3:
return "{} 被禁言了".format(self.user)
elif self.type == 4:
return "{} 被取消禁言了".format(self.user)
elif self.type == 0:
return "{} 进入了房间".format(self.user)
else:
return self.content
def __unicode__(self):
return self.__str__()

View File

@ -30,7 +30,7 @@ class User:
if self.level == 0: if self.level == 0:
if self.type == 1: if self.type == 1:
return "[房管]{}".format(self.name) return "[房管]{}".format(self.name)
elif self.type == 2: elif self.type == 3:
return "[主播]{}".format(self.name) return "[主播]{}".format(self.name)
else: else:
return "{}".format(self.name) return "{}".format(self.name)

View File

@ -3,6 +3,7 @@ import sys
import time import time
from Gift import Gift from Gift import Gift
from MemberMsg import MemberMsg
from User import User from User import User
from Chat import Chat from Chat import Chat
@ -116,7 +117,7 @@ class WinMain(Api):
def onMessage(self, msg: str): def onMessage(self, msg: str):
set_cmd_text_color(FOREGROUND_DARKGRAY) set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息", msg) print("消息 : ", msg)
resetColor() resetColor()
def onJoin(self, user: User): def onJoin(self, user: User):
@ -127,15 +128,10 @@ class WinMain(Api):
def onSubscribe(self, user: User): def onSubscribe(self, user: User):
return return
def onEnter(self, user: User, content: str == ""): def onEnter(self, msg:MemberMsg):
if content == "": if msg.type != 0:
if user.name == "三国空白" or user.name == "四维v":
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息:", user, "进入直播间")
resetColor()
else:
set_cmd_text_color(FOREGROUND_DARKGRAY) set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息:", content.format(user)) print("提示 : ", msg)
resetColor() resetColor()
def onChat(self, chat: Chat): def onChat(self, chat: Chat):

17
api.py
View File

@ -1,4 +1,6 @@
import sys import sys
from MemberMsg import MemberMsg
from User import User from User import User
from Gift import Gift from Gift import Gift
from Chat import Chat from Chat import Chat
@ -52,26 +54,23 @@ class XiGuaLiveApi:
def onChat(self, chat: Chat): def onChat(self, chat: Chat):
print(chat) print(chat)
def onEnter(self, user: User, content: str == ""): def onEnter(self, msg:MemberMsg):
if content == "": print("提示 : ", msg)
print("消息:", user, "进入直播间")
else:
print("消息:", content.format(user))
def onSubscribe(self, user: User): def onSubscribe(self, user: User):
print("消息", user, "关注了主播") print("消息 : ", user, "关注了主播")
def onJoin(self, user: User): def onJoin(self, user: User):
print("感谢", user, "加入了粉丝团") print("感谢", user, "加入了粉丝团")
def onMessage(self, msg: str): def onMessage(self, msg: str):
print("消息", msg) print("消息 : ", msg)
def onLike(self, user: User): def onLike(self, user: User):
print("用户", user, "点了喜欢") print("用户", user, "点了喜欢")
def onLeave(self, json: any): def onLeave(self, json: any):
print("消息", "主播离开一小会") print("消息 : ", "主播离开一小会")
def _enterRoom(self): def _enterRoom(self):
if not self.isValidRoom: if not self.isValidRoom:
@ -132,7 +131,7 @@ class XiGuaLiveApi:
self.onChat(Chat(i)) self.onChat(Chat(i))
elif i['Method'] == "VideoLiveMemberMessage": elif i['Method'] == "VideoLiveMemberMessage":
self._updateRoomInfo(i) self._updateRoomInfo(i)
self.onEnter(User(i), i["Msg"]["content"]) self.onEnter(MemberMsg(i))
elif i['Method'] == "VideoLiveSocialMessage": elif i['Method'] == "VideoLiveSocialMessage":
self.onSubscribe(User(i)) self.onSubscribe(User(i))
elif i['Method'] == "VideoLiveJoinDiscipulusMessage": elif i['Method'] == "VideoLiveJoinDiscipulusMessage":