mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-11 01:27:29 +08:00
✨ Add sign status for Lagrange.Core
This commit is contained in:
parent
b3aa5c9e02
commit
75a4d1fdcb
127
liteyuki/plugins/sign_status.py
Normal file
127
liteyuki/plugins/sign_status.py
Normal file
@ -0,0 +1,127 @@
|
|||||||
|
import datetime
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
|
import httpx
|
||||||
|
from nonebot import require
|
||||||
|
from nonebot.plugin import PluginMetadata
|
||||||
|
|
||||||
|
from liteyuki.utils.base.config import get_config
|
||||||
|
from liteyuki.utils.base.data import Database, LiteModel
|
||||||
|
|
||||||
|
require("nonebot_plugin_alconna")
|
||||||
|
require("nonebot_plugin_apscheduler")
|
||||||
|
from nonebot_plugin_apscheduler import scheduler
|
||||||
|
from nonebot_plugin_alconna import Alconna, Subcommand, on_alconna
|
||||||
|
|
||||||
|
__author__ = "snowykami"
|
||||||
|
__plugin_meta__ = PluginMetadata(
|
||||||
|
name="签名服务器状态",
|
||||||
|
description="适用于ntqq的签名状态查看",
|
||||||
|
usage="",
|
||||||
|
type="application",
|
||||||
|
homepage="https://github.com/snowykami/LiteyukiBot",
|
||||||
|
extra={
|
||||||
|
"liteyuki" : True,
|
||||||
|
"toggleable" : True,
|
||||||
|
"default_enable": True,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
SIGN_COUNT_URLS: dict[str, str] = get_config("sign_count_urls", None)
|
||||||
|
SIGN_COUNT_DURATION = get_config("sign_count_duration", 10)
|
||||||
|
|
||||||
|
|
||||||
|
class SignCount(LiteModel):
|
||||||
|
TABLE_NAME: str = "sign_count"
|
||||||
|
time: float = 0.0
|
||||||
|
count: int = 0
|
||||||
|
sid: str = ""
|
||||||
|
|
||||||
|
|
||||||
|
sign_db = Database("data/liteyuki/ntqq_sign.ldb")
|
||||||
|
sign_db.auto_migrate(SignCount())
|
||||||
|
|
||||||
|
sign_status = on_alconna(Alconna(
|
||||||
|
"sign",
|
||||||
|
Subcommand(
|
||||||
|
"chart"
|
||||||
|
),
|
||||||
|
Subcommand(
|
||||||
|
"count"
|
||||||
|
),
|
||||||
|
Subcommand(
|
||||||
|
"data"
|
||||||
|
)
|
||||||
|
))
|
||||||
|
|
||||||
|
|
||||||
|
@sign_status.assign("count")
|
||||||
|
async def _():
|
||||||
|
reply = "Current sign count:"
|
||||||
|
for name, count in (await get_now_sign()).items():
|
||||||
|
reply += f"\n{name}: {count[1]}"
|
||||||
|
await sign_status.send(reply)
|
||||||
|
|
||||||
|
|
||||||
|
@sign_status.assign("data")
|
||||||
|
async def _():
|
||||||
|
query_stamp = [1, 5, 10, 15]
|
||||||
|
|
||||||
|
reply = "Count from last " + ", ".join([str(i) for i in query_stamp]) + "mins"
|
||||||
|
for name, url in SIGN_COUNT_URLS.items():
|
||||||
|
count_data = []
|
||||||
|
for stamp in query_stamp:
|
||||||
|
count_rows = sign_db.all(SignCount(), "sid = ? and time > ?", url, time.time() - 60 * stamp)
|
||||||
|
if len(count_rows) < 2:
|
||||||
|
count_data.append(-1)
|
||||||
|
else:
|
||||||
|
count_data.append(count_rows[-1].count - count_rows[0].count)
|
||||||
|
reply += f"\n{name}: " + ", ".join([str(i) for i in count_data])
|
||||||
|
await sign_status.send(reply)
|
||||||
|
|
||||||
|
|
||||||
|
@sign_status.assign("chart")
|
||||||
|
async def _():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@scheduler.scheduled_job("interval", seconds=SIGN_COUNT_DURATION, next_run_time=datetime.datetime.now())
|
||||||
|
async def update_sign_count():
|
||||||
|
if not SIGN_COUNT_URLS:
|
||||||
|
return
|
||||||
|
data = await get_now_sign()
|
||||||
|
for name, count in data.items():
|
||||||
|
await save_sign_count(count[0], count[1], SIGN_COUNT_URLS[name])
|
||||||
|
|
||||||
|
|
||||||
|
async def get_now_sign() -> dict[str, tuple[float, int]]:
|
||||||
|
"""
|
||||||
|
Get the sign count and the time of the latest sign
|
||||||
|
Returns:
|
||||||
|
tuple[float, int] | None: (time, count)
|
||||||
|
"""
|
||||||
|
data = {}
|
||||||
|
now = time.time()
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
for name, url in SIGN_COUNT_URLS.items():
|
||||||
|
resp = await client.get(url)
|
||||||
|
count = resp.json()["count"]
|
||||||
|
data[name] = (now, count)
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
|
async def save_sign_count(timestamp: float, count: int, sid: str):
|
||||||
|
"""
|
||||||
|
Save the sign count to the database
|
||||||
|
Args:
|
||||||
|
sid: the sign id, use url as the id
|
||||||
|
count:
|
||||||
|
timestamp (float): the time of the sign count (int): the count of the sign
|
||||||
|
"""
|
||||||
|
sign_db.save(SignCount(time=timestamp, count=count, sid=sid))
|
||||||
|
|
||||||
|
|
||||||
|
async def generate_chart(duration: int = 60):
|
||||||
|
pass
|
Loading…
Reference in New Issue
Block a user