上rust!

This commit is contained in:
shenjack 2024-02-22 13:06:30 +08:00
parent bbaf8cf0c9
commit 75e7e0085a
Signed by: shenjack
GPG Key ID: 7B1134A979775551
4 changed files with 106 additions and 7 deletions

View File

@ -61,9 +61,14 @@ class IcaClient:
@staticmethod @staticmethod
async def send_message_a(client: "IcaClient", message: SendMessage) -> bool: async def send_message_a(client: "IcaClient", message: SendMessage) -> bool:
... ...
def send_message(self, 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: def on_message(msg: NewMessage, client: IcaClient) -> None:

View File

@ -6,14 +6,9 @@ else:
NewMessage = TypeVar("NewMessage") NewMessage = TypeVar("NewMessage")
IcaClient = TypeVar("IcaClient") IcaClient = TypeVar("IcaClient")
_version_ = "0.1.0" _version_ = "1.0.0"
def on_message(msg: NewMessage, client: IcaClient) -> None: 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 not msg.is_from_self:
if msg.content == "/bot-rs-py": if msg.content == "/bot-rs-py":
reply = msg.reply_with(f"ica-async-rs-sync-py {_version_}") reply = msg.reply_with(f"ica-async-rs-sync-py {_version_}")

86
ica-rs/plugins/bmcl.py Normal file
View 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)

View File

@ -1,4 +1,5 @@
use pyo3::prelude::*; use pyo3::prelude::*;
use tracing::{debug, info, warn};
use rust_socketio::asynchronous::Client; use rust_socketio::asynchronous::Client;
use tokio::runtime::Runtime; use tokio::runtime::Runtime;
@ -225,6 +226,18 @@ impl IcaClientPy {
Ok(send_message(&client.client, &message.msg).await) 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 { impl IcaClientPy {