这样不错

This commit is contained in:
shenjack 2024-11-08 22:06:51 +08:00
parent dd776c69b2
commit c59757824b
Signed by: shenjack
GPG Key ID: 7B1134A979775551

View File

@ -15,6 +15,20 @@ CMD_PREFIX = "/sr"
LAST_CMD = f"{CMD_PREFIX} last" LAST_CMD = f"{CMD_PREFIX} last"
LAST_SHIP_CMD = f"{CMD_PREFIX} last ship" LAST_SHIP_CMD = f"{CMD_PREFIX} last ship"
LAST_SAVE_CMD = f"{CMD_PREFIX} last save" LAST_SAVE_CMD = f"{CMD_PREFIX} last save"
INFO_CMD = f"{CMD_PREFIX} info" # info xxxxx(int)
def data_type_fmt(data_type: str) -> str:
if data_type == "ship":
return "飞船"
elif data_type == "save":
return "存档"
elif data_type == "none":
return "无数据"
elif data_type == "unknown":
return "未知"
else:
return f"未知类型: {data_type}"
def format_data_size(data_bytes: float) -> str: def format_data_size(data_bytes: float) -> str:
@ -42,13 +56,36 @@ def last_data(path: str) -> str:
ship = data["data"] ship = data["data"]
if path == "data": if path == "data":
data_type = "飞船" if ship["save_type"] == "ship" else "存档" d_type_str = f"类型: {data_type_fmt(ship['save_type'])}"
d_type_str = f"类型: {data_type}"
else: else:
d_type_str = "" d_type_str = ""
return f"ID: {ship['save_id']}\n{d_type_str}\n数据长度: {format_data_size(ship['len'])}\nblake3 hash: {ship['blake_hash']}" return f"ID: {ship['save_id']}\n{d_type_str}\n数据长度: {format_data_size(ship['len'])}\nblake3 hash: {ship['blake_hash']}"
def get_ship_info(msg: IcaNewMessage, client: IcaClient):
if len(msg.content) <= len(INFO_CMD) + 1:
client.send_message(msg.reply_with("参数不足"))
return
ship_id = msg.content[len(INFO_CMD) + 1:]
if not ship_id.isdigit():
client.send_message(msg.reply_with("ID 必须是数字"))
return
try:
res = requests.get(f"{API_URL}/info/{ship_id}", timeout=5)
data = res.json()
except (requests.RequestException, requests.Timeout) as e:
client.send_and_warn(msg.reply_with(f"请求中出现问题: {e} {res}"))
return
if data["code"] != 200:
client.send_and_warn(msg.reply_with(f"请求失败: {data['msg']}"))
return
ship = data["data"]
formatted = f"ID: {ship['save_id']}\n类型: {data_type_fmt(ship['save_type'])}\n数据长度: {format_data_size(ship['len'])}\nblake3 hash: {ship['blake_hash']}"
client.send_message(msg.reply_with(formatted))
def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None: def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None:
if msg.is_from_self or not msg.is_room_msg: if msg.is_from_self or not msg.is_room_msg:
return return
@ -62,3 +99,5 @@ def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None:
client.send_message(msg.reply_with(last_data("save"))) client.send_message(msg.reply_with(last_data("save")))
elif msg.content == LAST_CMD: elif msg.content == LAST_CMD:
client.send_message(msg.reply_with(last_data("data"))) client.send_message(msg.reply_with(last_data("data")))
elif msg.content.startswith(INFO_CMD):
get_ship_info(msg, client)