q792602257/XiguaLiveDanmakuHelper
Archived
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

改进部分代码，增加Linux可运行的版本，可自己选择想要进入的房间

Browse Source 
Signed-off-by: Jerry Yan <792602257@qq.com>
This commit is contained in:
Jerry Yan 2019-01-20 23:10:35 +08:00
parent c3015eba65
commit ab06868f06
5 changed files with 316 additions and 191 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
Chat.py Normal file
View File

@ -0,0 +1,23 @@
from User import User
class Chat:
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 "content" in json["Msg"]:
self.content = json["Msg"]['content']
def __str__(self):
return "{} : {}".format(self.user,self.content)
def __unicode__(self):
return self.__str__()

19
GiftStruct.py → Gift.py
View File

@ -1,17 +1,21 @@
import requests
from User import User
class GiftStruct:
ID = 0
count = 0
giftList={10001: {"Name": "西瓜", "Price": 0}}
amount = 0
class Gift:
ID:int = 0
count:int = 0
roomID:int = 0
giftList:dict = {10001: {"Name": "西瓜", "Price": 0}}
amount:int = 0
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 "present_end_info" in json["Msg"]:
self.ID = json["Msg"]['present_end_info']['id']
@ -24,20 +28,21 @@ class GiftStruct:
@staticmethod
def update(roomID):
Gift.roomID = roomID
p = requests.get("https://live.ixigua.com/api/gifts/{roomID}".format(roomID= roomID))
d = p.json()
if isinstance(d, int) or "data" not in d:
print("错误：礼物更新失败")
else:
for i in d["data"]:
GiftStruct.giftList[i["ID"]] = {"Name": i["Name"], "Price": i["DiamondCount"]}
Gift.giftList[i["ID"]] = {"Name": i["Name"], "Price": i["DiamondCount"]}
def __str__(self):
if self.ID in self.giftList:
giftN = self.giftList[self.ID]["Name"]
else:
giftN = "未知礼物[{}]".format(self.ID)
return "{count} 个 {name}".format(count= self.count, name= giftN)
return "感谢 {user} 送出的 {count} 个 {name}".format(user= self.user, count= self.count, name= giftN)
def __unicode__(self):
return self.__str__()

26
UserStruct.py → User.py
View File

@ -1,9 +1,9 @@
class UserStruct:
ID = 0
name = ""
brand= ""
level= 0
type = 0
class User:
ID: int = 0
name: str = ""
brand: str = ""
level: int = 0
type: int = 0
def __init__(self, json=None):
if json:
@ -18,14 +18,22 @@ class UserStruct:
if "discipulus_info" in json["Msg"]:
self.level = json["Msg"]["discipulus_info"]["level"]
self.brand = json["Msg"]["discipulus_info"]["discipulus_group_title"]
elif "data" in json:
if "anchorInfo" in json["data"]:
self.ID = json["data"]['anchorInfo']['user_id']
self.name = json["data"]['anchorInfo']['name']
self.type = 0
if self.type is None:
self.type = 0
def __str__(self):
if self.level == 0:
if self.type != 0:
return "[]{}".format(self.name)
return "{}".format(self.name)
if self.type == 1:
return "[房管]{}".format(self.name)
elif self.type == 2:
return "[主播]{}".format(self.name)
else:
return "{}".format(self.name)
else:
if self.type != 0:
return "[{}{}]{}".format(self.brand, self.level, self.name)

193
WinMain.py Normal file
View File

@ -0,0 +1,193 @@
import os
import sys
import time
from Gift import Gift
from User import User
from Chat import Chat
from api import XiGuaLiveApi as Api
import msvcrt
import ctypes
def readInput(caption, default, timeout: int = 5):
start_time = time.time()
print('{}({})\r\n>'.format(caption, default), end="")
input = ''
while True:
if msvcrt.kbhit():
chr = msvcrt.getche()
if ord(chr) == 13: # enter_key
break
elif ord(chr) == 27:
break
elif ord(chr) == 8:
if input != "":
input = input[:-1]
msvcrt.putch(b" ")
msvcrt.putch(b"\b")
if len(input) == 0:
start_time = time.time()
elif 32 <= ord(chr) <= 126: # space_char
input += chr.decode("utf8")
if len(input) == 0 and (time.time() - start_time) > timeout:
break
if len(input) > 0:
print()
return input
else:
print("使用默认值")
return default
STD_INPUT_HANDLE = -10
STD_OUTPUT_HANDLE = -11
STD_ERROR_HANDLE = -12
std_out_handle = ctypes.windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
# 字体颜色定义 ,关键在于颜色编码，由2位十六进制组成，分别取0~f，前一位指的是背景色，后一位指的是字体色
# 由于该函数的限制，应该是只有这16种，可以前景色与背景色组合。也可以几种颜色通过或运算组合，组合后还是在这16种颜色中
# Windows CMD命令行 字体颜色定义 text colors
FOREGROUND_BLACK = 0x00 # black.
FOREGROUND_DARKBLUE = 0x01 # dark blue.
FOREGROUND_DARKGREEN = 0x02 # dark green.
FOREGROUND_DARKSKYBLUE = 0x03 # dark skyblue.
FOREGROUND_DARKRED = 0x04 # dark red.
FOREGROUND_DARKPINK = 0x05 # dark pink.
FOREGROUND_DARKYELLOW = 0x06 # dark yellow.
FOREGROUND_DARKWHITE = 0x07 # dark white.
FOREGROUND_DARKGRAY = 0x08 # dark gray.
FOREGROUND_BLUE = 0x09 # blue.
FOREGROUND_GREEN = 0x0a # green.
FOREGROUND_SKYBLUE = 0x0b # skyblue.
FOREGROUND_RED = 0x0c # red.
FOREGROUND_PINK = 0x0d # pink.
FOREGROUND_YELLOW = 0x0e # yellow.
FOREGROUND_WHITE = 0x0f # white.
# Windows CMD命令行 背景颜色定义 background colors
BACKGROUND_BLUE = 0x10 # dark blue.
BACKGROUND_GREEN = 0x20 # dark green.
BACKGROUND_DARKSKYBLUE = 0x30 # dark skyblue.
BACKGROUND_DARKRED = 0x40 # dark red.
BACKGROUND_DARKPINK = 0x50 # dark pink.
BACKGROUND_DARKYELLOW = 0x60 # dark yellow.
BACKGROUND_DARKWHITE = 0x70 # dark white.
BACKGROUND_DARKGRAY = 0x80 # dark gray.
BACKGROUND_BLUE = 0x90 # blue.
BACKGROUND_GREEN = 0xa0 # green.
BACKGROUND_SKYBLUE = 0xb0 # skyblue.
BACKGROUND_RED = 0xc0 # red.
BACKGROUND_PINK = 0xd0 # pink.
BACKGROUND_YELLOW = 0xe0 # yellow.
BACKGROUND_WHITE = 0xf0 # white.
def set_cmd_text_color(color, handle=std_out_handle):
Bool = ctypes.windll.kernel32.SetConsoleTextAttribute(handle, color)
return Bool
def resetColor():
set_cmd_text_color(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE)
class WinMain(Api):
_tmp = 0
def getTitle(self):
self._tmp += 1
if self._tmp > 10:
self._tmp = 0
if self._tmp < 5 :
return "{} {} --弹幕助手 by JerryYan".format(self.roomLiver, "的直播间")
else:
if self.roomPopularity == 0:
self._tmp = 0
return self.getTitle()
else:
if self.roomMember > 0:
return "观看人数:{} 人气:{} --弹幕助手 by JerryYan".format(self.roomMember, self.roomPopularity)
else:
return "观看人数:待刷新数据 人气:{} --弹幕助手 by JerryYan".format(self.roomPopularity)
def onMessage(self, msg: str):
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", msg)
resetColor()
def onJoin(self, user: User):
set_cmd_text_color(BACKGROUND_WHITE | FOREGROUND_BLACK)
print("感谢", user, "加入了粉丝团")
resetColor()
def onSubscribe(self, user: User):
return
def onEnter(self, user: User, content: str == ""):
if content == "":
if user.name == "三国空白" or user.name == "四维v":
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", user, "进入直播间")
resetColor()
else:
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", content.format(user))
resetColor()
def onChat(self, chat: Chat):
print(chat)
def onPresent(self, gift: Gift):
return
def onPresentEnd(self, gift: Gift):
set_cmd_text_color(BACKGROUND_WHITE | FOREGROUND_BLACK)
print(gift)
resetColor()
def onLike(self, user: User):
return
def onLeave(self, json: any):
return
def warning(*args):
print(*args)
if __name__ == "__main__":
room = 97621754276 # 永恒
# room = 75366565294
# room = 83940182312 #Dae
resetColor()
print("西瓜直播弹幕助手 by JerryYan")
if len(sys.argv) > 1:
room = int(sys.argv[1])
else:
try:
room = int(readInput("请输入房间号，默认为永恒的直播间", room, 3))
except ValueError:
pass
api = WinMain(room)
print("进入", api.roomLiver, "的直播间")
if not api.isValidRoom:
input("房间不存在")
sys.exit()
os.system("title {}".format(api.getTitle()))
print("=" * 30)
while True:
if api.isLive:
try:
os.system("title {}".format(api.getTitle()))
api.getDanmaku()
except Exception as e:
warning(e)
time.sleep(1)
else:
print("主播未开播，等待1分钟后重试")
time.sleep(60)
api.updRoomInfo()

246
api.py
View File

@ -1,194 +1,98 @@
import msvcrt
import sys
from UserStruct import UserStruct
from GiftStruct import GiftStruct
from User import User
from Gift import Gift
from Chat import Chat
import requests
import time
import ctypes
import os
def warning(*args):
print(*args)
def debug(*args):
# print(args)
pass
def readInput(caption, default, timeout:int=5):
start_time = time.time()
print('{}({})\r\n>'.format(caption,default), end="")
input = ''
while True:
if msvcrt.kbhit():
chr = msvcrt.getche()
if ord(chr) == 13: # enter_key
break
elif ord(chr) == 27:
break
elif ord(chr) == 8:
if input != "":
input = input[:-1]
elif 32 <= ord(chr) <= 126: # space_char
input += chr.decode("utf8")
if len(input) == 0 and (time.time() - start_time) > timeout:
break
if len(input) > 0:
print()
return input
else:
print("使用默认值")
return default
STD_INPUT_HANDLE = -10
STD_OUTPUT_HANDLE = -11
STD_ERROR_HANDLE = -12
std_out_handle = ctypes.windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
# 字体颜色定义 ,关键在于颜色编码，由2位十六进制组成，分别取0~f，前一位指的是背景色，后一位指的是字体色
#由于该函数的限制，应该是只有这16种，可以前景色与背景色组合。也可以几种颜色通过或运算组合，组合后还是在这16种颜色中
# Windows CMD命令行 字体颜色定义 text colors
FOREGROUND_BLACK = 0x00 # black.
FOREGROUND_DARKBLUE = 0x01 # dark blue.
FOREGROUND_DARKGREEN = 0x02 # dark green.
FOREGROUND_DARKSKYBLUE = 0x03 # dark skyblue.
FOREGROUND_DARKRED = 0x04 # dark red.
FOREGROUND_DARKPINK = 0x05 # dark pink.
FOREGROUND_DARKYELLOW = 0x06 # dark yellow.
FOREGROUND_DARKWHITE = 0x07 # dark white.
FOREGROUND_DARKGRAY = 0x08 # dark gray.
FOREGROUND_BLUE = 0x09 # blue.
FOREGROUND_GREEN = 0x0a # green.
FOREGROUND_SKYBLUE = 0x0b # skyblue.
FOREGROUND_RED = 0x0c # red.
FOREGROUND_PINK = 0x0d # pink.
FOREGROUND_YELLOW = 0x0e # yellow.
FOREGROUND_WHITE = 0x0f # white.
# Windows CMD命令行 背景颜色定义 background colors
BACKGROUND_BLUE = 0x10 # dark blue.
BACKGROUND_GREEN = 0x20 # dark green.
BACKGROUND_DARKSKYBLUE = 0x30 # dark skyblue.
BACKGROUND_DARKRED = 0x40 # dark red.
BACKGROUND_DARKPINK = 0x50 # dark pink.
BACKGROUND_DARKYELLOW = 0x60 # dark yellow.
BACKGROUND_DARKWHITE = 0x70 # dark white.
BACKGROUND_DARKGRAY = 0x80 # dark gray.
BACKGROUND_BLUE = 0x90 # blue.
BACKGROUND_GREEN = 0xa0 # green.
BACKGROUND_SKYBLUE = 0xb0 # skyblue.
BACKGROUND_RED = 0xc0 # red.
BACKGROUND_PINK = 0xd0 # pink.
BACKGROUND_YELLOW = 0xe0 # yellow.
BACKGROUND_WHITE = 0xf0 # white.
s = requests.Session()
def set_cmd_text_color(color, handle=std_out_handle):
Bool = ctypes.windll.kernel32.SetConsoleTextAttribute(handle, color)
return Bool
def resetColor():
set_cmd_text_color(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE)
class XiGuaLiveApi:
isLive = False
roomInfo = {}
roomID = 0
cursor = ""
isLive: bool = False
isValidRoom: bool = False
_rawRoomInfo = {}
roomID: int = 0
roomTitle: str = ""
roomLiver: User = None
roomPopularity: int = 0
roomMember: int = 0
_cursor = ""
def __init__(self, room: int):
self.room = room
self.updRoomInfo()
GiftStruct.update(self.roomID)
Gift.update(self.roomID)
self._enterRoom()
def notLiveError(self):
print("主播未开播")
def apiChangedError(self, msg:str):
def _updateRoomInfo(self, json):
if "Msg" in json:
if "member_count" in json["Msg"]:
self.roomMember = json["Msg"]["member_count"]
if "popularity" in json["Msg"]:
self.roomPopularity = json["Msg"]["popularity"]
def apiChangedError(self, msg: str, *args):
print(msg)
print(*args)
def onPresent(self, user:UserStruct, gift:GiftStruct):
return
# print("礼物连击：", user, "的", gift)
def onPresent(self, gift: Gift):
print("礼物连击：", gift)
def onPresentEnd(self, user:UserStruct, gift:GiftStruct):
set_cmd_text_color(BACKGROUND_WHITE | FOREGROUND_BLACK)
print("感谢", user, "送出的", gift)
resetColor()
def onPresentEnd(self, gift: Gift):
print("礼物", gift)
def onAd(self, i):
# print(i)
pass
def onChat(self, user:UserStruct, content:str):
print(user, "：", content)
# pass
def onChat(self, chat: Chat):
print(chat)
def onEnter(self, user:UserStruct, content:str == ""):
def onEnter(self, user: User, content: str == ""):
if content == "":
if user.name == "三国空白" or user.name == "四维v":
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", user, "进入直播间")
resetColor()
print("消息：", user, "进入直播间")
else:
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", content.format(user))
resetColor()
def onSubscribe(self, user:UserStruct):
if user.level >= 6 and user.brand == "永恒":
set_cmd_text_color(FOREGROUND_DARKGRAY)
print("消息：", user, "关注了主播")
resetColor()
def onSubscribe(self, user: User):
print("消息：", user, "关注了主播")
def onJoin(self, user:UserStruct):
set_cmd_text_color(BACKGROUND_WHITE | FOREGROUND_BLACK)
def onJoin(self, user: User):
print("感谢", user, "加入了粉丝团")
resetColor()
def onMessage(self, msg:str):
set_cmd_text_color(FOREGROUND_DARKGRAY)
def onMessage(self, msg: str):
print("消息：", msg)
resetColor()
def onLike(self, user:UserStruct):
return
# set_cmd_text_color(FOREGROUND_DARKGRAY)
# print("用户", user, "点了喜欢")
# resetColor()
def onLike(self, user: User):
print("用户", user, "点了喜欢")
def onLeave(self, json:any):
def onLeave(self, json: any):
print("消息：", "主播离开一小会")
debug(json)
return
def enterRoom(self):
def _enterRoom(self):
if not self.isValidRoom:
return
p = s.post("https://live.ixigua.com/api/room/enter/{roomID}".format(roomID=self.roomID))
debug(p.json())
def updRoomInfo(self):
p = s.get("https://live.ixigua.com/api/room/{room}".format(room=self.room))
d = p.json()
debug(d)
if "data" not in d:
self.apiChangedError("数据结构改变，请与我联系")
debug(d)
self.apiChangedError("无法获取RoomID，请与我联系")
return
self.roomInfo = d["data"]
print("进入", self.roomInfo["anchorInfo"]["name"], "的直播间")
self.isValidRoom = True
self._rawRoomInfo = d["data"]
self.roomLiver = User(d)
self.roomTitle = self._rawRoomInfo["Title"]
self.roomPopularity = self._rawRoomInfo["Extra2"]["Popularity"]
if "Id" in d["data"]:
self.roomID = d["data"]["Id"]
else:
warning("无法获取RoomID，请与我联系")
self.apiChangedError("无法获取RoomID，请与我联系")
if "FinishTime" in d["data"]:
self.isLive = False
self.notLiveError()
@ -196,76 +100,68 @@ class XiGuaLiveApi:
self.isLive = True
def getDanmaku(self):
if not self.isValidRoom:
return
p = s.get("https://live.ixigua.com/api/msg/list/{roomID}?AnchorID={room}&Cursor={cursor}".format(
roomID=self.roomID,
room=self.room,
cursor=self.cursor
cursor=self._cursor
))
d = p.json()
debug(d)
if "data" not in d:
self.apiChangedError("数据结构改变，请与我联系")
debug(d)
self.apiChangedError("数据结构改变，请与我联系", d)
return
if "Extra" not in d["data"]:
self.apiChangedError("数据结构改变，请与我联系")
debug(d["data"])
self.apiChangedError("数据结构改变，请与我联系", d)
return
if "Cursor" not in d["data"]["Extra"]:
self.apiChangedError("数据结构改变，请与我联系")
debug(d["data"])
self.apiChangedError("数据结构改变，请与我联系", d)
return
else:
self.cursor = d["data"]["Extra"]["Cursor"]
self._cursor = d["data"]["Extra"]["Cursor"]
if "LiveMsgs" not in d["data"]:
return
for i in d['data']['LiveMsgs']:
if i['Method'] == "VideoLivePresentMessage":
self.onPresent(UserStruct(i), GiftStruct(i))
self.onPresent(Gift(i))
elif i['Method'] == "VideoLivePresentEndTipMessage":
self.onPresentEnd(UserStruct(i), GiftStruct(i))
self.onPresentEnd(Gift(i))
elif i['Method'] == "VideoLiveRoomAdMessage":
self.onAd(i)
elif i['Method'] == "VideoLiveChatMessage":
self.onChat(UserStruct(i), i["Msg"]['content'])
self.onChat(Chat(i))
elif i['Method'] == "VideoLiveMemberMessage":
self.onEnter(UserStruct(i), i["Msg"]["content"])
self._updateRoomInfo(i)
self.onEnter(User(i), i["Msg"]["content"])
elif i['Method'] == "VideoLiveSocialMessage":
self.onSubscribe(UserStruct(i))
self.onSubscribe(User(i))
elif i['Method'] == "VideoLiveJoinDiscipulusMessage":
self.onJoin(UserStruct(i))
self.onJoin(User(i))
elif i['Method'] == "VideoLiveControlMessage":
print("消息：", "主播离开一小会")
elif i['Method'] == "VideoLiveDiggMessage":
self.onLike(UserStruct(i))
self.onLike(User(i))
else:
debug(i)
pass
if __name__ == "__main__":
room = 97621754276 #永恒
room = 97621754276 # 永恒
# room = 75366565294
# room = 83940182312 #Dae
resetColor()
if len(sys.argv)>1:
room = int(sys.argv[1])
else:
try:
room = int(readInput("请输入房间号，默认为永恒的直播间", room, 3))
except ValueError:
pass
print("西瓜直播弹幕助手 by JerryYan")
print("正在进入房间", room)
api = XiGuaLiveApi(room)
os.system("title {} {}".format(api.roomInfo["anchorInfo"]["name"],"的直播间 --西瓜直播弹幕助手 by JerryYan"))
api.enterRoom()
print("="*30)
print("进入", api.roomLiver, "的直播间")
if not api.isValidRoom:
input("房间不存在")
sys.exit()
print("=" * 30)
while True:
if api.isLive:
try:
api.getDanmaku()
except Exception as e:
warning(e)
print(e)
time.sleep(1)
else:
print("主播未开播，等待1分钟后重试")