tursom/blivedm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
tursom
2026-03-14 07:57:27 +08:00
parent 17947229a1
commit 0e24b9026b
1 changed files with 59 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
live_status_ntf.py
View File

@@ -9,24 +9,22 @@ import db
import re import re
path = "logs" path = "logs"
room_id_re = re.compile(rf"{path}/(\d+).jsonl$") room_id_re = re.compile(rf"{path}/(\d+)\.jsonl$")
notify_qq_group = { notify_qq_group = {
"12571885": lambda msg: notify_group(msg, ["831867573", "138981147"]), "12571885": lambda room, is_online: notify_group(room, ["831867573", "138981147"], is_online),
"147482": lambda msg: notify_group(msg, ["175545447"]), "147482": lambda room, is_online: notify_group(room, ["175545447"], is_online),
"20571": ["891117762"], "20571": lambda room, is_online: notify_group(room, ["891117762"], is_online),
} }
class Room: class Room:
position = 0
# 0 unkonwn, 1 live, 2 pending
state = 0
state_changed = False
def __init__(self, room_id: str) -> None: def __init__(self, room_id: str) -> None:
self.room_id = room_id self.room_id = room_id
self.db_room = db.get_room(room_id) self.db_room = db.get_room(room_id)
self.position = 0
self.state = 0
self.state_changed = False
def reset(self): def reset(self):
self.position = 0 self.position = 0
@@ -41,30 +39,31 @@ class Room:
_rooms: dict[str, Room] = {} _rooms: dict[str, Room] = {}
def notify_group(room: Room, groups: list[str]): def notify_group(room: Room, groups: list[str], is_online: bool):
status = "开锅了" if is_online else "下锅了"
msg = "\n".join([ msg = "\n".join([
f"{room.db_room.liver_name} 下锅了！", f"{room.db_room.liver_name} {status}！",
f"https://live.bilibili.com/{room.room_id}" f"https://live.bilibili.com/{room.room_id}"
]) ])
for qq_group_id in groups: for qq_group_id in groups:
seng_qq_group_msg(qq_group_id, msg) send_qq_group_msg(qq_group_id, msg)
def notify_group_and_mc(room: Room, groups: list[str]): def notify_group_and_mc(room: Room, groups: list[str], is_online: bool):
notify_group(room, groups, is_online)
try:
msg = "\n".join([ msg = "\n".join([
f"{room.db_room.liver_name} 开锅了！", f"{room.db_room.liver_name} {'开锅了' if is_online else '下锅了'}！",
f"https://live.bilibili.com/{room.room_id}" f"https://live.bilibili.com/{room.room_id}"
]) ])
try:
requests.post("http://mc1:5000/msg", json={"msg": msg}) requests.post("http://mc1:5000/msg", json={"msg": msg})
except BaseException as e: except BaseException as e:
print(f"notify mc error: {e}") print(f"notify mc error: {e}")
for qq_group_id in groups:
seng_qq_group_msg(qq_group_id, msg)
def get_room(src_path): def get_room(src_path):
room_id = room_id_re.findall(src_path)[0] src_path_str = str(src_path)
room_id = room_id_re.findall(src_path_str)[0]
room = _rooms.get(room_id, None) room = _rooms.get(room_id, None)
if room is None: if room is None:
room = Room(room_id) room = Room(room_id)
@@ -78,25 +77,27 @@ class MyHandler(FileSystemEventHandler):
if event.is_directory: if event.is_directory:
return return
if not room_id_re.match(event.src_path): src_path_str = str(event.src_path)
if not room_id_re.search(src_path_str):
return return
room: Room = get_room(event.src_path) room: Room = get_room(src_path_str)
try: try:
with open(event.src_path, "r", encoding="utf-8") as f: with open(event.src_path, "r", encoding="utf-8") as f:
if room.position > os.path.getsize(event.src_path): file_size = os.path.getsize(event.src_path)
if room.position > file_size:
room.reset() room.reset()
f.seek(room.position) f.seek(room.position)
room.position = os.path.getsize(event.src_path)
for line in f: for line in f:
try: try:
data = json.loads(line) data = json.loads(line)
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
continue continue
self._handle_data(room, data) self._handle_data(room, data)
room.position = os.path.getsize(event.src_path)
except UnicodeDecodeError: except UnicodeDecodeError:
pass pass
@@ -106,11 +107,11 @@ class MyHandler(FileSystemEventHandler):
if event.is_directory: if event.is_directory:
return return
if not room_id_re.match(event.src_path): src_path_str = str(event.src_path)
if not room_id_re.search(src_path_str):
return return
room: Room = get_room(event.src_path) room: Room = get_room(src_path_str)
room.position = os.path.getsize(event.src_path)
try: try:
with open(event.src_path, "r", encoding="utf-8") as f: with open(event.src_path, "r", encoding="utf-8") as f:
@@ -119,7 +120,9 @@ class MyHandler(FileSystemEventHandler):
data = json.loads(line) data = json.loads(line)
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
continue continue
self._handle_data(data) self._handle_data(room, data)
room.position = os.path.getsize(event.src_path)
except UnicodeDecodeError: except UnicodeDecodeError:
pass pass
@@ -129,10 +132,12 @@ class MyHandler(FileSystemEventHandler):
if event.is_directory: if event.is_directory:
return return
if not room_id_re.match(event.src_path): src_path_str = str(event.src_path)
if not room_id_re.search(src_path_str):
return return
room = _rooms.get(event.src_path, Room()) room_id = room_id_re.findall(src_path_str)[0]
room = _rooms.get(room_id, Room(room_id))
room.reset() room.reset()
def _handle_data(self, room: Room, data): def _handle_data(self, room: Room, data):
@@ -159,32 +164,44 @@ class MyHandler(FileSystemEventHandler):
f"{room.db_room.liver_name} 开锅了！", f"{room.db_room.liver_name} 开锅了！",
f"https://live.bilibili.com/{room.room_id}" f"https://live.bilibili.com/{room.room_id}"
]) ])
try:
requests.post("http://turntf:18846/notify", json={"msg": msg}) requests.post("http://turntf:18846/notify", json={"msg": msg})
except BaseException as e:
print(f"notify turntf error: {e}")
for notify_group in notify_qq_group.get(room.room_id, []): notify_func = notify_qq_group.get(room.room_id)
seng_qq_group_msg(notify_group, msg) if notify_func:
notify_func(room, True)
elif room.state == 2: elif room.state == 2:
msg = "\n".join([ msg = "\n".join([
f"{room.db_room.liver_name} 下锅了！", f"{room.db_room.liver_name} 下锅了！",
f"https://live.bilibili.com/{room.room_id}" f"https://live.bilibili.com/{room.room_id}"
]) ])
try:
requests.post("http://turntf:18846/notify", json={"msg": msg}) requests.post("http://turntf:18846/notify", json={"msg": msg})
except BaseException as e:
print(f"notify turntf error: {e}")
notify_qq_group.get(room.room_id, lambda msg: None)(room) notify_func = notify_qq_group.get(room.room_id)
if notify_func:
notify_func(room, False)
def seng_qq_group_msg(group_id: str, msg: str): def send_qq_group_msg(group_id: str, msg: str):
try:
requests.post("http://napcat:3000/send_group_msg", json={ requests.post("http://napcat:3000/send_group_msg", json={
"group_id": group_id, "group_id": group_id,
"message": msg "message": msg
}) })
except BaseException as e:
print(f"send qq group msg error: {e}")
def main(): def main():
for f in os.listdir(path): for f in os.listdir(path):
f = path+"/"+str(f) f = path+"/"+str(f)
if not room_id_re.match(f): if not room_id_re.search(f):
continue continue
get_room(f).position = os.path.getsize(f) get_room(f).position = os.path.getsize(f)