87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
|
|
||
|
|
import time
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING, TypeVar
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from ica_typing import NewMessage, IcaClient
|
||
|
|
else:
|
||
|
|
NewMessage = TypeVar("NewMessage")
|
||
|
|
IcaClient = TypeVar("IcaClient")
|
||
|
|
|
||
|
|
_version_ = "2.0.0-rs"
|
||
|
|
|
||
|
|
def format_data_size(data_bytes: float) -> str:
|
||
|
|
data_lens = ["B", "KB", "MB", "GB", "TB"]
|
||
|
|
data_len = "0B"
|
||
|
|
for i in range(5):
|
||
|
|
if data_bytes < 1024:
|
||
|
|
data_bytes = round(data_bytes, 5)
|
||
|
|
data_len = f"{data_bytes}{data_lens[i]}"
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
data_bytes /= 1024
|
||
|
|
return data_len
|
||
|
|
|
||
|
|
|
||
|
|
def format_hit_count(count: int) -> str:
|
||
|
|
"""数据分段, 四位一个下划线
|
||
|
|
|
||
|
|
Args:
|
||
|
|
count (int): 数据
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
str: 格式化后的数据
|
||
|
|
1 -> 1
|
||
|
|
1000 -> 1000
|
||
|
|
10000 -> 1_0000
|
||
|
|
100000 -> 10_0000
|
||
|
|
1000000 -> 100_0000
|
||
|
|
"""
|
||
|
|
count_str = str(count)
|
||
|
|
count_len = len(count_str)
|
||
|
|
if count_len <= 4:
|
||
|
|
return count_str
|
||
|
|
else:
|
||
|
|
return "_".join(count_str[i:i + 4] for i in range(0, count_len, 4))
|
||
|
|
|
||
|
|
def bmcl(msg: NewMessage, client: IcaClient) -> None:
|
||
|
|
req_time = time.time()
|
||
|
|
# 记录请求时间
|
||
|
|
response = requests.get("https://bd.bangbang93.com/openbmclapi/metric/dashboard")
|
||
|
|
if not response.status_code == 200 or response.reason != "OK":
|
||
|
|
reply = msg.reply_with(f"请求数据失败\n{response.status_code}")
|
||
|
|
client.warn(
|
||
|
|
f"数据请求失败, 请检查网络\n{response.status}"
|
||
|
|
)
|
||
|
|
client.send_message(reply)
|
||
|
|
return
|
||
|
|
data = response.json()
|
||
|
|
data_bytes: float = data["bytes"]
|
||
|
|
data_hits: int = data["hits"]
|
||
|
|
data_bandwidth: float = data["currentBandwidth"]
|
||
|
|
load_str: float = data["load"] * 100
|
||
|
|
online_node: int = data["currentNodes"]
|
||
|
|
online_bandwidth: int = data["bandwidth"]
|
||
|
|
data_len = format_data_size(data_bytes)
|
||
|
|
hits_count = format_hit_count(data_hits)
|
||
|
|
|
||
|
|
report_msg = (
|
||
|
|
f"OpenBMCLAPI 状态面板v{_version_} :\n"
|
||
|
|
f"实时信息: {online_node} 带宽: {online_bandwidth}Mbps\n"
|
||
|
|
f"负载: {load_str:.2f}% 带宽: {data_bandwidth:.2f}Mbps\n"
|
||
|
|
f"当日请求: {hits_count} 数据量: {data_len}\n"
|
||
|
|
f"请求时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(req_time))}\n"
|
||
|
|
"数据源: https://bd.bangbang93.com/pages/dashboard"
|
||
|
|
)
|
||
|
|
client.info(report_msg)
|
||
|
|
reply = msg.reply_with(report_msg)
|
||
|
|
client.send_message(reply)
|
||
|
|
|
||
|
|
|
||
|
|
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
||
|
|
if not msg.is_from_self:
|
||
|
|
if msg.content == "/bmcl-rs":
|
||
|
|
bmcl(msg, client)
|