mirror of
http://shenjack.top:5100/shenjack/icalingua-python-bot.git
synced 2024-11-23 12:41:05 +08:00
上rust!
This commit is contained in:
parent
bbaf8cf0c9
commit
75e7e0085a
|
@ -61,9 +61,14 @@ class IcaClient:
|
|||
@staticmethod
|
||||
async def send_message_a(client: "IcaClient", message: SendMessage) -> bool:
|
||||
...
|
||||
|
||||
def send_message(self, message: SendMessage) -> bool:
|
||||
...
|
||||
def debug(self, message: str) -> None:
|
||||
...
|
||||
def info(self, message: str) -> None:
|
||||
...
|
||||
def warn(self, message: str) -> None:
|
||||
...
|
||||
|
||||
|
||||
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
||||
|
|
|
@ -6,14 +6,9 @@ else:
|
|||
NewMessage = TypeVar("NewMessage")
|
||||
IcaClient = TypeVar("IcaClient")
|
||||
|
||||
_version_ = "0.1.0"
|
||||
_version_ = "1.0.0"
|
||||
|
||||
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
||||
# print(msg)
|
||||
# reply = msg.reply_with("Hello, world!")
|
||||
# print(reply)
|
||||
# print(msg.content)
|
||||
# print(msg.is_from_self)
|
||||
if not msg.is_from_self:
|
||||
if msg.content == "/bot-rs-py":
|
||||
reply = msg.reply_with(f"ica-async-rs-sync-py {_version_}")
|
86
ica-rs/plugins/bmcl.py
Normal file
86
ica-rs/plugins/bmcl.py
Normal file
|
@ -0,0 +1,86 @@
|
|||
|
||||
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)
|
|
@ -1,4 +1,5 @@
|
|||
use pyo3::prelude::*;
|
||||
use tracing::{debug, info, warn};
|
||||
use rust_socketio::asynchronous::Client;
|
||||
use tokio::runtime::Runtime;
|
||||
|
||||
|
@ -225,6 +226,18 @@ impl IcaClientPy {
|
|||
Ok(send_message(&client.client, &message.msg).await)
|
||||
})
|
||||
}
|
||||
|
||||
pub fn debug(&self, content: String) {
|
||||
debug!("{}", content);
|
||||
}
|
||||
|
||||
pub fn info(&self, content: String) {
|
||||
info!("{}", content);
|
||||
}
|
||||
|
||||
pub fn warn(&self, content: String) {
|
||||
warn!("{}", content);
|
||||
}
|
||||
}
|
||||
|
||||
impl IcaClientPy {
|
||||
|
|
Loading…
Reference in New Issue
Block a user