From 7042d94213532191a0c72d34d7c85193184c079f Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Wed, 4 May 2022 16:07:18 +0800 Subject: =?UTF-8?q?=E2=9C=A8=20=E4=B8=BA=E5=89=8D=E7=AB=AF=E9=A1=B5?= =?UTF-8?q?=E9=9D=A2=E6=89=93=E5=9F=BA=E7=A1=80,=20=E6=9B=B4=E6=96=B0?= =?UTF-8?q?=E5=91=BD=E4=BB=A4=E5=8F=8AAPI?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/status/__init__.py | 216 ++++++++++++++++++++++++++++++++- ATRI/plugins/status/data_source.py | 106 +++++++++++++--- ATRI/plugins/status/driver/__init__.py | 41 +++++++ ATRI/plugins/status/driver/view.py | 39 ++++++ ATRI/plugins/status/listener.py | 49 ++++++++ ATRI/plugins/status/models.py | 32 +++++ 6 files changed, 463 insertions(+), 20 deletions(-) create mode 100644 ATRI/plugins/status/driver/__init__.py create mode 100644 ATRI/plugins/status/driver/view.py create mode 100644 ATRI/plugins/status/listener.py create mode 100644 ATRI/plugins/status/models.py (limited to 'ATRI/plugins') diff --git a/ATRI/plugins/status/__init__.py b/ATRI/plugins/status/__init__.py index 05ece79..9aa5364 100644 --- a/ATRI/plugins/status/__init__.py +++ b/ATRI/plugins/status/__init__.py @@ -1,21 +1,34 @@ +import json + +from nonebot.params import ArgPlainText +from nonebot.permission import SUPERUSER +from nonebot.adapters.onebot.v11 import GroupMessageEvent, PrivateMessageEvent + +from ATRI.log import logger as log +from ATRI.config import BotSelfConfig +from ATRI.exceptions import ReadFileError, WriteError from ATRI.utils.apscheduler import scheduler -from .data_source import IsSurvive +from .data_source import Status, STATUS_DIR +from .models import ForAuthData +from .driver import init +init() -ping = IsSurvive().on_command("/ping", "检测bot简单信息处理速度") + +ping = Status().on_command("/ping", "检测bot简单信息处理速度") @ping.handle() async def _ping(): - await ping.finish(IsSurvive.ping()) + await ping.finish(Status.ping()) -status = IsSurvive().on_command("/status", "查看运行资源占用") +status = Status().on_command("/status", "查看运行资源占用") @status.handle() async def _status(): - msg, _ = IsSurvive.get_status() + msg, _ = Status.get_status() await status.finish(msg) @@ -24,6 +37,197 @@ info_msg = "アトリは高性能ですから！" @scheduler.scheduled_job("interval", name="状态检查", minutes=10, misfire_grace_time=15) # type: ignore async def _check_runtime(): - msg, stat = IsSurvive().get_status() + log.info("开始检查资源消耗...") + msg, stat = Status().get_status() if not stat: await status.finish(msg) + + +get_console_key = Status().on_command("/auth", "获取进入网页后台的凭证", permission=SUPERUSER) + + +@get_console_key.got("is_pub_n", "咱的运行环境是否有公网(y/n)") +async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): + if is_pub_n != "y": + ip = str(await Status().get_host_ip(False)) + await get_console_key.send("没有公网吗...嗯知道了") + else: + ip = str(await Status().get_host_ip(True)) + + p = BotSelfConfig.port + rs = Status().get_random_str(20) + + df = STATUS_DIR / "data.json" + try: + if not df.is_file(): + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + + d = json.loads(df.read_bytes()) + + ca = d.get("data", None) + if ca: + # 此处原本想用 matcher.finish 但这是在 try 里啊！ + await get_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") + return + + d["data"] = ForAuthData(ip=ip, port=str(p), token=rs).dict() + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(d)) + except WriteError: + msg = f""" + 哦吼！写入文件失败了...还请自行记下哦... + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await get_console_key.send(msg) + + raise WriteError("Writing file: " + str(df) + " failed!") + + msg = f""" + 该信息已保存！可通过 /gauth 获取~ + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await get_console_key.finish(msg) + + +@get_console_key.handle() +async def _(event: GroupMessageEvent): + await get_console_key.finish("请私戳咱获取（") + + +load_console_key = Status().on_command("/gauth", "获取已生成的后台凭证", permission=SUPERUSER) + + +@load_console_key.handle() +async def _(event: PrivateMessageEvent): + df = STATUS_DIR / "data.json" + if not df.is_file(): + await load_console_key.finish("你还没有问咱索要奥！/auth 以获取") + + try: + d = json.loads(df.read_bytes()) + except ReadFileError: + await load_console_key.send("获取数据失败了...请自行打开文件查看吧:\n" + str(df)) + raise ReadFileError("Reading file: " + str(df) + " failed!") + + data = d["data"] + msg = f""" + 诶嘿嘿嘿——凭证信息来咯！ + IP: {data['ip']} + PORT: {data['port']} + TOKEN: {data['token']} + 切记！不要告诉他人！！ + """.strip() + await load_console_key.finish(msg) + + +@load_console_key.handle() +async def _(event: GroupMessageEvent): + await load_console_key.finish("请私戳咱获取（") + + +del_console_key = Status().on_command("/deauth", "销毁进入网页后台的凭证", permission=SUPERUSER) + + +@del_console_key.got("is_sure_d", "...你确定吗(y/n)") +async def _(is_sure: str = ArgPlainText("is_sure_d")): + if is_sure != "y": + await del_console_key.finish("反悔了呢...") + + df = STATUS_DIR / "data.json" + if not df.is_file(): + await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") + + try: + data: dict = json.loads(df.read_bytes()) + + del data["data"] + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(data)) + except WriteError: + await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) + raise WriteError("Writing / Reading file: " + str(df) + " failed!") + + await del_console_key.finish("销毁成功！如需再次获取: /auth") + + +res_console_key = Status().on_command("/reauth", "重置进入网页后台的凭证", permission=SUPERUSER) + + +@res_console_key.got("is_sure_r", "...你确定吗(y/n)") +async def _(is_sure: str = ArgPlainText("is_sure_r")): + if is_sure != "y": + await res_console_key.finish("反悔了呢...") + + df = STATUS_DIR / "data.json" + if not df.is_file(): + await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") + + try: + data: dict = json.loads(df.read_bytes()) + + del data["data"] + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(data)) + except WriteError: + await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) + raise WriteError("Writing / Reading file: " + str(df) + " failed!") + + +@res_console_key.got("is_pub_r_n", "咱的运行环境是否有公网(y/n)") +async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): + if is_pub_n != "y": + ip = str(await Status().get_host_ip(False)) + await res_console_key.send("没有公网吗...嗯知道了") + else: + ip = str(await Status().get_host_ip(True)) + + p = BotSelfConfig.port + rs = Status().get_random_str(20) + + df = STATUS_DIR / "data.json" + try: + if not df.is_file(): + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + + d = json.loads(df.read_bytes()) + + ca = d.get("data", None) + if ca: + await res_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") + return + + d["data"] = ForAuthData(ip=ip, port=str(p), token=rs).dict() + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(d)) + except WriteError: + msg = f""" + 哦吼！写入文件失败了...还请自行记下哦... + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await res_console_key.send(msg) + + raise WriteError("Writing file: " + str(df) + " failed!") + + msg = f""" + 该信息已保存！可通过 /gauth 获取~ + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await res_console_key.finish(msg) diff --git a/ATRI/plugins/status/data_source.py b/ATRI/plugins/status/data_source.py index 3595c0c..1658468 100644 --- a/ATRI/plugins/status/data_source.py +++ b/ATRI/plugins/status/data_source.py @@ -1,29 +1,41 @@ +import os import time +import json import psutil +import socket +import string +from pathlib import Path +from random import sample from datetime import datetime from ATRI.service import Service from ATRI.log import logger as log from ATRI.rule import is_in_service -from ATRI.exceptions import GetStatusError +from ATRI.utils import request +from ATRI.exceptions import GetStatusError, WriteError +from .models import PlatfromRuntimeInfo, BotRuntimeInfo +STATUS_DIR = Path(".") / "data" / "database" / "status" +STATUS_DIR.mkdir(exist_ok=True) + _status_msg = """ > Status Overview -[CPU: {cpu}%] -[Memory: {mem}%] -[Disk usage: {disk}%] +[CPU: {p_cpu}% of {b_cpu}] +[Memory: {p_mem}% of {b_mem}] +[Disk usage: {p_disk}%] [Net sent: {inteSENT}MB] [Net recv: {inteRECV}MB] -[Runtime: {up_time}] +[Bot runtime: {bot_time}] +[Platform runtime: {boot_time}] {msg} """.strip() -class IsSurvive(Service): +class Status(Service): def __init__(self): Service.__init__(self, "状态", "检查自身状态", rule=is_in_service("状态")) @@ -32,8 +44,9 @@ class IsSurvive(Service): return "I'm fine." @staticmethod - def get_status(): - log.info("开始检查资源消耗...") + def get_status(is_for_fn: bool = False) -> tuple: + data_p = Path(".") / "data" + try: cpu = psutil.cpu_percent(interval=1) mem = psutil.virtual_memory().percent @@ -41,12 +54,21 @@ class IsSurvive(Service): inteSENT = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore inteRECV = psutil.net_io_counters().bytes_recv / 1000000 # type: ignore + process = psutil.Process(os.getpid()) + b_cpu = process.cpu_percent(interval=1) + b_mem = process.memory_percent(memtype="rss") + now = time.time() boot = psutil.boot_time() - up_time = str( + b = process.create_time() + boot_time = str( datetime.utcfromtimestamp(now).replace(microsecond=0) - datetime.utcfromtimestamp(boot).replace(microsecond=0) ) + bot_time = str( + datetime.utcfromtimestamp(now).replace(microsecond=0) + - datetime.utcfromtimestamp(b).replace(microsecond=0) + ) except GetStatusError: raise GetStatusError("Failed to get status.") @@ -64,17 +86,73 @@ class IsSurvive(Service): msg = "咱感觉身体要被塞满了..." is_ok = False else: - log.info("资源占用正常") is_ok = True + if is_for_fn: + return ( + PlatfromRuntimeInfo( + stat_msg=msg, + cpu_percent=str(cpu), + mem_percent=mem, + disk_percent=str(disk), + inte_send=str(inteSENT), + inte_recv=str(inteRECV), + boot_time=boot_time, + ).dict(), + BotRuntimeInfo( + cpu_percent=str(b_cpu), + mem_percent=str(b_mem), + bot_run_time=bot_time, + ).dict(), + ) + msg0 = _status_msg.format( - cpu=cpu, - mem=mem, - disk=disk, + p_cpu=cpu, + p_mem=mem, + p_disk=disk, + b_cpu=f"{b_cpu}%", + b_mem="%.1f%%" % b_mem, inteSENT=inteSENT, inteRECV=inteRECV, - up_time=up_time, + bot_time=bot_time, + boot_time=boot_time, msg=msg, ) return msg0, is_ok + + @staticmethod + async def get_host_ip(is_pub: bool): + if is_pub: + data = await request.get("https://ifconfig.me/ip") + return data.text + + s = None + try: + s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + s.connect(('8.8.8.8', 80)) + ip = s.getsockname()[0] + return ip + finally: + if s: + s.close() + + @staticmethod + def get_random_str(k: int) -> str: + return "".join(sample(string.ascii_letters + string.digits, k)) + + @staticmethod + def get_auth_info() -> dict: + df = STATUS_DIR / "data.json" + if not df.is_file(): + try: + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + except WriteError: + raise WriteError("Writing file: " + str(df) + " failed!") + + base_data: dict = json.loads(df.read_bytes()) + data = base_data.get("data", None) + if not data: + return {"data": None} + return data diff --git a/ATRI/plugins/status/driver/__init__.py b/ATRI/plugins/status/driver/__init__.py new file mode 100644 index 0000000..199fb1e --- /dev/null +++ b/ATRI/plugins/status/driver/__init__.py @@ -0,0 +1,41 @@ +from nonebot import get_driver +from nonebot.drivers.fastapi import Driver + +from fastapi.middleware.cors import CORSMiddleware + +from .view import ( + handle_auther, + handle_base_uri, + handle_runtime_info, + handle_message_deal_info, +) + + +CONSOLE_API_URI = "/capi" # base point +CONSOLE_API_AUTH_URI = "/capi/auth" # 验证后台许可 +CONSOLE_API_RUNTIME_URI = "/capi/runtime" # 获取运行占用信息 +CONSOLE_API_MESSAGE_URI = "/capi/message" +# CONSOLE_API_AUTH_COOKIES_URI = "/capi/auth/cookies" # 验证cookies + + +def register_routes(driver: Driver): + app = driver.server_app + + origins = ["*"] + app.add_middleware( + CORSMiddleware, + allow_origins=origins, + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], + ) + + app.get(CONSOLE_API_URI)(handle_base_uri) + app.get(CONSOLE_API_RUNTIME_URI)(handle_runtime_info) + app.get(CONSOLE_API_MESSAGE_URI)(handle_message_deal_info) + app.get(CONSOLE_API_AUTH_URI)(handle_auther) + + +def init(): + driver = get_driver() + register_routes(driver) # type: ignore diff --git a/ATRI/plugins/status/driver/view.py b/ATRI/plugins/status/driver/view.py new file mode 100644 index 0000000..5d8c56f --- /dev/null +++ b/ATRI/plugins/status/driver/view.py @@ -0,0 +1,39 @@ +from ..data_source import Status +from ..listener import get_message_deal_info + + +def handle_base_uri(): + return {"status": 204, "msg": "This path just for console load."} + + +def handle_runtime_info(token: str): + auth, data = auth_token(token) + plat, bot = Status().get_status(True) + if auth: + return {"status": 200, "data": {"platform": plat, "bot": bot}} + else: + return data + + +def handle_message_deal_info(token: str): + auth, data = auth_token(token) + if auth: + return {"status": 200, "data": get_message_deal_info()} + else: + return data + + +def handle_auther(token: str): + auth, data = auth_token(token) + return data if auth else data + + +def auth_token(token: str) -> tuple: + auth_data: dict = Status().get_auth_info() + if not auth_data.get("token", None): + return False, {"status": 500, "msg": "This bot is not create auth data yet."} + _token = auth_data["token"] + if token != _token: + return False, {"status": 403, "msg": "Token error, please check again."} + else: + return True, {"status": 200, "msg": "OK"} diff --git a/ATRI/plugins/status/listener.py b/ATRI/plugins/status/listener.py new file mode 100644 index 0000000..0ae5c25 --- /dev/null +++ b/ATRI/plugins/status/listener.py @@ -0,0 +1,49 @@ +from typing import Optional + +from nonebot.message import run_postprocessor + +from ATRI.utils.apscheduler import scheduler +from .models import MessageDealerInfo + + +recv_msg = int() +deal_msg = int() +failed_deal_msg = int() + +total_r_m = int() +total_d_m = int() +total_f_m = int() + + +@run_postprocessor +async def _(exception: Optional[Exception]): + global recv_msg, deal_msg, failed_deal_msg, total_r_m, total_d_m, total_f_m + + if exception: + failed_deal_msg += 1 + total_f_m += 1 + else: + deal_msg += 1 + total_d_m += 1 + + recv_msg += 1 + total_r_m += 1 + + +def get_message_deal_info() -> dict: + return MessageDealerInfo( + recv_msg=str(recv_msg), + deal_msg=str(deal_msg), + failed_deal_msg=str(failed_deal_msg), + total_r_m=str(total_r_m), + total_d_m=str(total_d_m), + total_f_m=str(total_f_m), + ).dict() + + +@scheduler.scheduled_job("interval", name="信息数据重置", seconds=15, misfire_grace_time=1) # type: ignore +async def _(): + global recv_msg, deal_msg, failed_deal_msg + recv_msg = int() + deal_msg = int() + failed_deal_msg = int() diff --git a/ATRI/plugins/status/models.py b/ATRI/plugins/status/models.py new file mode 100644 index 0000000..9026b2d --- /dev/null +++ b/ATRI/plugins/status/models.py @@ -0,0 +1,32 @@ +from pydantic import BaseModel + + +class ForAuthData(BaseModel): + ip: str + port: str + token: str + + +class PlatfromRuntimeInfo(BaseModel): + stat_msg: str + cpu_percent: str + mem_percent: str + disk_percent: str + inte_send: str + inte_recv: str + boot_time: str + + +class BotRuntimeInfo(BaseModel): + cpu_percent: str + mem_percent: str + bot_run_time: str + + +class MessageDealerInfo(BaseModel): + recv_msg: str + deal_msg: str + failed_deal_msg: str + total_r_m: str + total_d_m: str + total_f_m: str -- cgit v1.2.3 From 817b55b84ab65e0404b15195bed97a46d977f3b9 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Fri, 13 May 2022 23:59:19 +0800 Subject: =?UTF-8?q?=F0=9F=9A=9A=20=E7=8B=AC=E7=AB=8B=E6=8E=A7=E5=88=B6?= =?UTF-8?q?=E5=8F=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/status/__init__.py | 210 +-------------------------------- ATRI/plugins/status/data_source.py | 79 +------------ ATRI/plugins/status/driver/__init__.py | 41 ------- ATRI/plugins/status/driver/view.py | 39 ------ ATRI/plugins/status/listener.py | 49 -------- ATRI/plugins/status/models.py | 32 ----- 6 files changed, 10 insertions(+), 440 deletions(-) delete mode 100644 ATRI/plugins/status/driver/__init__.py delete mode 100644 ATRI/plugins/status/driver/view.py delete mode 100644 ATRI/plugins/status/listener.py delete mode 100644 ATRI/plugins/status/models.py (limited to 'ATRI/plugins') diff --git a/ATRI/plugins/status/__init__.py b/ATRI/plugins/status/__init__.py index 9aa5364..ae7702a 100644 --- a/ATRI/plugins/status/__init__.py +++ b/ATRI/plugins/status/__init__.py @@ -1,25 +1,13 @@ -import json - -from nonebot.params import ArgPlainText -from nonebot.permission import SUPERUSER -from nonebot.adapters.onebot.v11 import GroupMessageEvent, PrivateMessageEvent - from ATRI.log import logger as log -from ATRI.config import BotSelfConfig -from ATRI.exceptions import ReadFileError, WriteError from ATRI.utils.apscheduler import scheduler -from .data_source import Status, STATUS_DIR -from .models import ForAuthData -from .driver import init - -init() +from .data_source import Status ping = Status().on_command("/ping", "检测bot简单信息处理速度") @ping.handle() -async def _ping(): +async def _(): await ping.finish(Status.ping()) @@ -27,7 +15,7 @@ status = Status().on_command("/status", "查看运行资源占用") @status.handle() -async def _status(): +async def _(): msg, _ = Status.get_status() await status.finish(msg) @@ -36,198 +24,8 @@ info_msg = "アトリは高性能ですから！" @scheduler.scheduled_job("interval", name="状态检查", minutes=10, misfire_grace_time=15) # type: ignore -async def _check_runtime(): +async def _(): log.info("开始检查资源消耗...") msg, stat = Status().get_status() if not stat: await status.finish(msg) - - -get_console_key = Status().on_command("/auth", "获取进入网页后台的凭证", permission=SUPERUSER) - - -@get_console_key.got("is_pub_n", "咱的运行环境是否有公网(y/n)") -async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): - if is_pub_n != "y": - ip = str(await Status().get_host_ip(False)) - await get_console_key.send("没有公网吗...嗯知道了") - else: - ip = str(await Status().get_host_ip(True)) - - p = BotSelfConfig.port - rs = Status().get_random_str(20) - - df = STATUS_DIR / "data.json" - try: - if not df.is_file(): - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps({})) - - d = json.loads(df.read_bytes()) - - ca = d.get("data", None) - if ca: - # 此处原本想用 matcher.finish 但这是在 try 里啊！ - await get_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") - return - - d["data"] = ForAuthData(ip=ip, port=str(p), token=rs).dict() - - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps(d)) - except WriteError: - msg = f""" - 哦吼！写入文件失败了...还请自行记下哦... - IP: {ip} - PORT: {p} - TOKEN: {rs} - 一定要保管好哦！切勿告诉他人哦！ - """.strip() - await get_console_key.send(msg) - - raise WriteError("Writing file: " + str(df) + " failed!") - - msg = f""" - 该信息已保存！可通过 /gauth 获取~ - IP: {ip} - PORT: {p} - TOKEN: {rs} - 一定要保管好哦！切勿告诉他人哦！ - """.strip() - await get_console_key.finish(msg) - - -@get_console_key.handle() -async def _(event: GroupMessageEvent): - await get_console_key.finish("请私戳咱获取（") - - -load_console_key = Status().on_command("/gauth", "获取已生成的后台凭证", permission=SUPERUSER) - - -@load_console_key.handle() -async def _(event: PrivateMessageEvent): - df = STATUS_DIR / "data.json" - if not df.is_file(): - await load_console_key.finish("你还没有问咱索要奥！/auth 以获取") - - try: - d = json.loads(df.read_bytes()) - except ReadFileError: - await load_console_key.send("获取数据失败了...请自行打开文件查看吧:\n" + str(df)) - raise ReadFileError("Reading file: " + str(df) + " failed!") - - data = d["data"] - msg = f""" - 诶嘿嘿嘿——凭证信息来咯！ - IP: {data['ip']} - PORT: {data['port']} - TOKEN: {data['token']} - 切记！不要告诉他人！！ - """.strip() - await load_console_key.finish(msg) - - -@load_console_key.handle() -async def _(event: GroupMessageEvent): - await load_console_key.finish("请私戳咱获取（") - - -del_console_key = Status().on_command("/deauth", "销毁进入网页后台的凭证", permission=SUPERUSER) - - -@del_console_key.got("is_sure_d", "...你确定吗(y/n)") -async def _(is_sure: str = ArgPlainText("is_sure_d")): - if is_sure != "y": - await del_console_key.finish("反悔了呢...") - - df = STATUS_DIR / "data.json" - if not df.is_file(): - await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") - - try: - data: dict = json.loads(df.read_bytes()) - - del data["data"] - - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps(data)) - except WriteError: - await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) - raise WriteError("Writing / Reading file: " + str(df) + " failed!") - - await del_console_key.finish("销毁成功！如需再次获取: /auth") - - -res_console_key = Status().on_command("/reauth", "重置进入网页后台的凭证", permission=SUPERUSER) - - -@res_console_key.got("is_sure_r", "...你确定吗(y/n)") -async def _(is_sure: str = ArgPlainText("is_sure_r")): - if is_sure != "y": - await res_console_key.finish("反悔了呢...") - - df = STATUS_DIR / "data.json" - if not df.is_file(): - await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") - - try: - data: dict = json.loads(df.read_bytes()) - - del data["data"] - - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps(data)) - except WriteError: - await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) - raise WriteError("Writing / Reading file: " + str(df) + " failed!") - - -@res_console_key.got("is_pub_r_n", "咱的运行环境是否有公网(y/n)") -async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): - if is_pub_n != "y": - ip = str(await Status().get_host_ip(False)) - await res_console_key.send("没有公网吗...嗯知道了") - else: - ip = str(await Status().get_host_ip(True)) - - p = BotSelfConfig.port - rs = Status().get_random_str(20) - - df = STATUS_DIR / "data.json" - try: - if not df.is_file(): - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps({})) - - d = json.loads(df.read_bytes()) - - ca = d.get("data", None) - if ca: - await res_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") - return - - d["data"] = ForAuthData(ip=ip, port=str(p), token=rs).dict() - - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps(d)) - except WriteError: - msg = f""" - 哦吼！写入文件失败了...还请自行记下哦... - IP: {ip} - PORT: {p} - TOKEN: {rs} - 一定要保管好哦！切勿告诉他人哦！ - """.strip() - await res_console_key.send(msg) - - raise WriteError("Writing file: " + str(df) + " failed!") - - msg = f""" - 该信息已保存！可通过 /gauth 获取~ - IP: {ip} - PORT: {p} - TOKEN: {rs} - 一定要保管好哦！切勿告诉他人哦！ - """.strip() - await res_console_key.finish(msg) diff --git a/ATRI/plugins/status/data_source.py b/ATRI/plugins/status/data_source.py index 1658468..ddd3568 100644 --- a/ATRI/plugins/status/data_source.py +++ b/ATRI/plugins/status/data_source.py @@ -1,24 +1,13 @@ import os import time -import json import psutil -import socket -import string -from pathlib import Path -from random import sample from datetime import datetime from ATRI.service import Service -from ATRI.log import logger as log from ATRI.rule import is_in_service -from ATRI.utils import request -from ATRI.exceptions import GetStatusError, WriteError -from .models import PlatfromRuntimeInfo, BotRuntimeInfo +from ATRI.exceptions import GetStatusError -STATUS_DIR = Path(".") / "data" / "database" / "status" -STATUS_DIR.mkdir(exist_ok=True) - _status_msg = """ > Status Overview @@ -44,15 +33,13 @@ class Status(Service): return "I'm fine." @staticmethod - def get_status(is_for_fn: bool = False) -> tuple: - data_p = Path(".") / "data" - + def get_status() -> tuple: try: cpu = psutil.cpu_percent(interval=1) mem = psutil.virtual_memory().percent disk = psutil.disk_usage("/").percent - inteSENT = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore - inteRECV = psutil.net_io_counters().bytes_recv / 1000000 # type: ignore + inte_send = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore + inte_recv = psutil.net_io_counters().bytes_recv / 1000000 # type: ignore process = psutil.Process(os.getpid()) b_cpu = process.cpu_percent(interval=1) @@ -88,71 +75,17 @@ class Status(Service): else: is_ok = True - if is_for_fn: - return ( - PlatfromRuntimeInfo( - stat_msg=msg, - cpu_percent=str(cpu), - mem_percent=mem, - disk_percent=str(disk), - inte_send=str(inteSENT), - inte_recv=str(inteRECV), - boot_time=boot_time, - ).dict(), - BotRuntimeInfo( - cpu_percent=str(b_cpu), - mem_percent=str(b_mem), - bot_run_time=bot_time, - ).dict(), - ) - msg0 = _status_msg.format( p_cpu=cpu, p_mem=mem, p_disk=disk, b_cpu=f"{b_cpu}%", b_mem="%.1f%%" % b_mem, - inteSENT=inteSENT, - inteRECV=inteRECV, + inteSENT=inte_send, + inteRECV=inte_recv, bot_time=bot_time, boot_time=boot_time, msg=msg, ) return msg0, is_ok - - @staticmethod - async def get_host_ip(is_pub: bool): - if is_pub: - data = await request.get("https://ifconfig.me/ip") - return data.text - - s = None - try: - s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) - s.connect(('8.8.8.8', 80)) - ip = s.getsockname()[0] - return ip - finally: - if s: - s.close() - - @staticmethod - def get_random_str(k: int) -> str: - return "".join(sample(string.ascii_letters + string.digits, k)) - - @staticmethod - def get_auth_info() -> dict: - df = STATUS_DIR / "data.json" - if not df.is_file(): - try: - with open(df, "w", encoding="utf-8") as w: - w.write(json.dumps({})) - except WriteError: - raise WriteError("Writing file: " + str(df) + " failed!") - - base_data: dict = json.loads(df.read_bytes()) - data = base_data.get("data", None) - if not data: - return {"data": None} - return data diff --git a/ATRI/plugins/status/driver/__init__.py b/ATRI/plugins/status/driver/__init__.py deleted file mode 100644 index 199fb1e..0000000 --- a/ATRI/plugins/status/driver/__init__.py +++ /dev/null @@ -1,41 +0,0 @@ -from nonebot import get_driver -from nonebot.drivers.fastapi import Driver - -from fastapi.middleware.cors import CORSMiddleware - -from .view import ( - handle_auther, - handle_base_uri, - handle_runtime_info, - handle_message_deal_info, -) - - -CONSOLE_API_URI = "/capi" # base point -CONSOLE_API_AUTH_URI = "/capi/auth" # 验证后台许可 -CONSOLE_API_RUNTIME_URI = "/capi/runtime" # 获取运行占用信息 -CONSOLE_API_MESSAGE_URI = "/capi/message" -# CONSOLE_API_AUTH_COOKIES_URI = "/capi/auth/cookies" # 验证cookies - - -def register_routes(driver: Driver): - app = driver.server_app - - origins = ["*"] - app.add_middleware( - CORSMiddleware, - allow_origins=origins, - allow_credentials=True, - allow_methods=["*"], - allow_headers=["*"], - ) - - app.get(CONSOLE_API_URI)(handle_base_uri) - app.get(CONSOLE_API_RUNTIME_URI)(handle_runtime_info) - app.get(CONSOLE_API_MESSAGE_URI)(handle_message_deal_info) - app.get(CONSOLE_API_AUTH_URI)(handle_auther) - - -def init(): - driver = get_driver() - register_routes(driver) # type: ignore diff --git a/ATRI/plugins/status/driver/view.py b/ATRI/plugins/status/driver/view.py deleted file mode 100644 index 5d8c56f..0000000 --- a/ATRI/plugins/status/driver/view.py +++ /dev/null @@ -1,39 +0,0 @@ -from ..data_source import Status -from ..listener import get_message_deal_info - - -def handle_base_uri(): - return {"status": 204, "msg": "This path just for console load."} - - -def handle_runtime_info(token: str): - auth, data = auth_token(token) - plat, bot = Status().get_status(True) - if auth: - return {"status": 200, "data": {"platform": plat, "bot": bot}} - else: - return data - - -def handle_message_deal_info(token: str): - auth, data = auth_token(token) - if auth: - return {"status": 200, "data": get_message_deal_info()} - else: - return data - - -def handle_auther(token: str): - auth, data = auth_token(token) - return data if auth else data - - -def auth_token(token: str) -> tuple: - auth_data: dict = Status().get_auth_info() - if not auth_data.get("token", None): - return False, {"status": 500, "msg": "This bot is not create auth data yet."} - _token = auth_data["token"] - if token != _token: - return False, {"status": 403, "msg": "Token error, please check again."} - else: - return True, {"status": 200, "msg": "OK"} diff --git a/ATRI/plugins/status/listener.py b/ATRI/plugins/status/listener.py deleted file mode 100644 index 0ae5c25..0000000 --- a/ATRI/plugins/status/listener.py +++ /dev/null @@ -1,49 +0,0 @@ -from typing import Optional - -from nonebot.message import run_postprocessor - -from ATRI.utils.apscheduler import scheduler -from .models import MessageDealerInfo - - -recv_msg = int() -deal_msg = int() -failed_deal_msg = int() - -total_r_m = int() -total_d_m = int() -total_f_m = int() - - -@run_postprocessor -async def _(exception: Optional[Exception]): - global recv_msg, deal_msg, failed_deal_msg, total_r_m, total_d_m, total_f_m - - if exception: - failed_deal_msg += 1 - total_f_m += 1 - else: - deal_msg += 1 - total_d_m += 1 - - recv_msg += 1 - total_r_m += 1 - - -def get_message_deal_info() -> dict: - return MessageDealerInfo( - recv_msg=str(recv_msg), - deal_msg=str(deal_msg), - failed_deal_msg=str(failed_deal_msg), - total_r_m=str(total_r_m), - total_d_m=str(total_d_m), - total_f_m=str(total_f_m), - ).dict() - - -@scheduler.scheduled_job("interval", name="信息数据重置", seconds=15, misfire_grace_time=1) # type: ignore -async def _(): - global recv_msg, deal_msg, failed_deal_msg - recv_msg = int() - deal_msg = int() - failed_deal_msg = int() diff --git a/ATRI/plugins/status/models.py b/ATRI/plugins/status/models.py deleted file mode 100644 index 9026b2d..0000000 --- a/ATRI/plugins/status/models.py +++ /dev/null @@ -1,32 +0,0 @@ -from pydantic import BaseModel - - -class ForAuthData(BaseModel): - ip: str - port: str - token: str - - -class PlatfromRuntimeInfo(BaseModel): - stat_msg: str - cpu_percent: str - mem_percent: str - disk_percent: str - inte_send: str - inte_recv: str - boot_time: str - - -class BotRuntimeInfo(BaseModel): - cpu_percent: str - mem_percent: str - bot_run_time: str - - -class MessageDealerInfo(BaseModel): - recv_msg: str - deal_msg: str - failed_deal_msg: str - total_r_m: str - total_d_m: str - total_f_m: str -- cgit v1.2.3 From 5c911c71a18e325863b0ef7af78ba04cab78ade2 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 14 May 2022 00:00:26 +0800 Subject: =?UTF-8?q?=E2=99=BF=EF=B8=8F=20=E4=BC=98=E5=8C=96=E6=8A=A5?= =?UTF-8?q?=E9=94=99=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/kimo/data_source.py | 6 +++--- ATRI/plugins/polaroid/data_source.py | 6 +++--- ATRI/plugins/setu/tf_dealer.py | 10 +++++----- 3 files changed, 11 insertions(+), 11 deletions(-) (limited to 'ATRI/plugins') diff --git a/ATRI/plugins/kimo/data_source.py b/ATRI/plugins/kimo/data_source.py index 6dd448a..fb3f3f3 100644 --- a/ATRI/plugins/kimo/data_source.py +++ b/ATRI/plugins/kimo/data_source.py @@ -8,7 +8,7 @@ from ATRI.service import Service from ATRI.rule import to_bot, is_in_service from ATRI.log import logger as log from ATRI.utils import request -from ATRI.exceptions import ReadFileError, WriteError +from ATRI.exceptions import ReadFileError, WriteFileError CHAT_PATH = Path(".") / "data" / "database" / "kimo" @@ -39,8 +39,8 @@ class Kimo(Service): with open(path, "w", encoding="utf-8") as w: w.write(json.dumps(data, indent=4)) log.info("生成完成") - except WriteError: - raise WriteError("Writing kimo words failed!") + except WriteFileError: + raise WriteFileError("Writing kimo words failed!") @classmethod async def _load_data(cls) -> dict: diff --git a/ATRI/plugins/polaroid/data_source.py b/ATRI/plugins/polaroid/data_source.py index 82e0337..d4df4fe 100644 --- a/ATRI/plugins/polaroid/data_source.py +++ b/ATRI/plugins/polaroid/data_source.py @@ -4,7 +4,7 @@ from ATRI.service import Service from ATRI.rule import is_in_service from ATRI.utils import request from ATRI.log import logger as log -from ATRI.exceptions import RequestError, WriteError +from ATRI.exceptions import RequestError, WriteFileError from .image_dealer import image_dealer @@ -52,8 +52,8 @@ async def init_source(): with open(path, "wb") as w: w.write(data.read()) log.info("所需资源装载完成") - except WriteError: - raise WriteError("装载资源失败") + except WriteFileError: + raise WriteFileError("装载资源失败") loop = asyncio.get_event_loop() diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py index f966636..58b4337 100644 --- a/ATRI/plugins/setu/tf_dealer.py +++ b/ATRI/plugins/setu/tf_dealer.py @@ -13,7 +13,7 @@ import tensorflow as tf from ATRI.log import logger as log from ATRI.utils import request -from ATRI.exceptions import RequestError, WriteError +from ATRI.exceptions import RequestError, WriteFileError SETU_PATH = Path(".") / "data" / "database" / "setu" @@ -60,8 +60,8 @@ async def detect_image(url: str, file_size: int) -> list: path = TEMP_PATH / f"{file_name}.jpg" with open(path, "wb") as f: f.write(req.read()) - except WriteError: - raise WriteError("Writing file failed!") + except WriteFileError: + raise WriteFileError("Writing file failed!") await init_module() model_path = str((SETU_PATH / "nsfw.tflite").absolute()) @@ -107,8 +107,8 @@ async def init_module(): with open(path, "wb") as w: w.write(data.read()) log.info("模型装载完成") - except WriteError: - raise WriteError("NSFW TF module init failed!") + except WriteFileError: + raise WriteFileError("NSFW TF module init failed!") loop = asyncio.get_event_loop() -- cgit v1.2.3 From 2abccb5960cc57e6d8ef8823df5340cb357a6b38 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 14 May 2022 00:01:14 +0800 Subject: =?UTF-8?q?=F0=9F=9A=91=EF=B8=8F=20=E9=98=BB=E6=AD=A2=E5=8F=91?= =?UTF-8?q?=E9=80=81=E7=A9=BA=E6=B6=88=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/kimo/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'ATRI/plugins') diff --git a/ATRI/plugins/kimo/__init__.py b/ATRI/plugins/kimo/__init__.py index b1c5698..31ecab1 100644 --- a/ATRI/plugins/kimo/__init__.py +++ b/ATRI/plugins/kimo/__init__.py @@ -21,7 +21,8 @@ async def _chat(event: MessageEvent): msg = str(event.message) repo = await Kimo().deal(msg, user_id) try: - await kimo.finish(repo) + if repo: + await kimo.send(repo) except Exception: return -- cgit v1.2.3 From d8dde99fb3a6f742488aee09c20bec7b9f4a3a1b Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 14 May 2022 00:02:20 +0800 Subject: =?UTF-8?q?=E2=9C=A8=20=E6=96=B0=E5=A2=9E=E6=8F=92=E4=BB=B6?= =?UTF-8?q?=E4=BB=A5=E6=94=AF=E6=8C=81=E6=8E=A7=E5=88=B6=E5=8F=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/console/__init__.py | 202 ++++++++++++++++++++++++++++++++ ATRI/plugins/console/data_source.py | 58 +++++++++ ATRI/plugins/console/driver/__init__.py | 56 +++++++++ ATRI/plugins/console/driver/api.py | 193 ++++++++++++++++++++++++++++++ ATRI/plugins/console/driver/view.py | 101 ++++++++++++++++ ATRI/plugins/console/listener.py | 49 ++++++++ ATRI/plugins/console/models.py | 40 +++++++ 7 files changed, 699 insertions(+) create mode 100644 ATRI/plugins/console/__init__.py create mode 100644 ATRI/plugins/console/data_source.py create mode 100644 ATRI/plugins/console/driver/__init__.py create mode 100644 ATRI/plugins/console/driver/api.py create mode 100644 ATRI/plugins/console/driver/view.py create mode 100644 ATRI/plugins/console/listener.py create mode 100644 ATRI/plugins/console/models.py (limited to 'ATRI/plugins') diff --git a/ATRI/plugins/console/__init__.py b/ATRI/plugins/console/__init__.py new file mode 100644 index 0000000..c6fd847 --- /dev/null +++ b/ATRI/plugins/console/__init__.py @@ -0,0 +1,202 @@ +import json + +from nonebot.params import ArgPlainText +from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent + +from ATRI.config import BotSelfConfig +from ATRI.exceptions import WriteFileError, ReadFileError +from .data_source import Console, CONSOLE_DIR +from .models import AuthData +from .driver import init + +init() + + +gen_console_key = Console().cmd_as_group("auth", "获取进入网页后台的凭证") + + +@gen_console_key.got("is_pub_n", "咱的运行环境是否有公网(y/n)") +async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): + if is_pub_n != "y": + ip = str(await Console().get_host_ip(False)) + await gen_console_key.send("没有公网吗...嗯知道了") + else: + ip = str(await Console().get_host_ip(True)) + + p = BotSelfConfig.port + rs = Console().get_random_str(20) + + df = CONSOLE_DIR / "data.json" + try: + if not df.is_file(): + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + + d = json.loads(df.read_bytes()) + + ca = d.get("data", None) + if ca: + # 此处原本想用 matcher.finish 但这是在 try 里啊！ + await gen_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") + return + + d["data"] = AuthData(ip=ip, port=str(p), token=rs).dict() + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(d)) + except WriteFileError: + msg = f""" + 哦吼！写入文件失败了...还请自行记下哦... + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await gen_console_key.send(msg) + + raise WriteFileError("Writing file: " + str(df) + " failed!") + + msg = f""" + 该信息已保存！可通过 /gauth 获取~ + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await gen_console_key.finish(msg) + + +@gen_console_key.handle() +async def _(event: GroupMessageEvent): + await gen_console_key.finish("请私戳咱获取（") + + +load_console_key = Console().cmd_as_group("load", "获取已生成的后台凭证") + + +@load_console_key.handle() +async def _(event: PrivateMessageEvent): + df = CONSOLE_DIR / "data.json" + if not df.is_file(): + await load_console_key.finish("你还没有问咱索要奥！/auth 以获取") + + try: + d = json.loads(df.read_bytes()) + except ReadFileError: + await load_console_key.send("获取数据失败了...请自行打开文件查看吧:\n" + str(df)) + raise ReadFileError("Reading file: " + str(df) + " failed!") + + data = d["data"] + msg = f""" + 诶嘿嘿嘿——凭证信息来咯！ + IP: {data['ip']} + PORT: {data['port']} + TOKEN: {data['token']} + 切记！不要告诉他人！！ + """.strip() + await load_console_key.finish(msg) + + +@load_console_key.handle() +async def _(event: GroupMessageEvent): + await load_console_key.finish("请私戳咱获取（") + + +del_console_key = Console().cmd_as_group("del", "销毁进入网页后台的凭证") + + +@del_console_key.got("is_sure_d", "...你确定吗(y/n)") +async def _(is_sure: str = ArgPlainText("is_sure_d")): + if is_sure != "y": + await del_console_key.finish("反悔了呢...") + + df = CONSOLE_DIR / "data.json" + if not df.is_file(): + await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") + + try: + data: dict = json.loads(df.read_bytes()) + + del data["data"] + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(data)) + except WriteFileError: + await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) + raise WriteFileError("Writing / Reading file: " + str(df) + " failed!") + + await del_console_key.finish("销毁成功！如需再次获取: /auth") + + +res_console_key = Console().cmd_as_group("reauth", "重置进入网页后台的凭证") + + +@res_console_key.got("is_sure_r", "...你确定吗(y/n)") +async def _(is_sure: str = ArgPlainText("is_sure_r")): + if is_sure != "y": + await res_console_key.finish("反悔了呢...") + + df = CONSOLE_DIR / "data.json" + if not df.is_file(): + await del_console_key.finish("你还没向咱索取凭证呢.../auth 以获取") + + try: + data: dict = json.loads(df.read_bytes()) + + del data["data"] + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(data)) + except WriteFileError: + await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df)) + raise WriteFileError("Writing / Reading file: " + str(df) + " failed!") + + +@res_console_key.got("is_pub_r_n", "咱的运行环境是否有公网(y/n)") +async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")): + if is_pub_n != "y": + ip = str(await Console().get_host_ip(False)) + await res_console_key.send("没有公网吗...嗯知道了") + else: + ip = str(await Console().get_host_ip(True)) + + p = BotSelfConfig.port + rs = Console().get_random_str(20) + + df = CONSOLE_DIR / "data.json" + try: + if not df.is_file(): + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + + d = json.loads(df.read_bytes()) + + ca = d.get("data", None) + if ca: + await res_console_key.send("咱已经告诉你了嗷！啊！忘了.../gauth 获取吧") + return + + d["data"] = AuthData(ip=ip, port=str(p), token=rs).dict() + + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps(d)) + except WriteFileError: + msg = f""" + 哦吼！写入文件失败了...还请自行记下哦... + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await res_console_key.send(msg) + + raise WriteFileError("Writing file: " + str(df) + " failed!") + + msg = f""" + 该信息已保存！可通过 /gauth 获取~ + IP: {ip} + PORT: {p} + TOKEN: {rs} + 一定要保管好哦！切勿告诉他人哦！ + """.strip() + await res_console_key.finish(msg) diff --git a/ATRI/plugins/console/data_source.py b/ATRI/plugins/console/data_source.py new file mode 100644 index 0000000..86a3813 --- /dev/null +++ b/ATRI/plugins/console/data_source.py @@ -0,0 +1,58 @@ +import json +import socket +import string +from random import sample +from pathlib import Path + +from nonebot.permission import SUPERUSER + +from ATRI.service import Service +from ATRI.utils import request +from ATRI.exceptions import WriteFileError + + +CONSOLE_DIR = Path(".") / "data" / "database" / "console" +CONSOLE_DIR.mkdir(exist_ok=True) + + +class Console(Service): + def __init__(self): + Service.__init__( + self, "控制台", "前端管理页面", True, main_cmd="/con", permission=SUPERUSER + ) + + @staticmethod + async def get_host_ip(is_pub: bool): + if is_pub: + data = await request.get("https://ifconfig.me/ip") + return data.text + + s = None + try: + s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + s.connect(('8.8.8.8', 80)) + ip = s.getsockname()[0] + return ip + finally: + if s: + s.close() + + @staticmethod + def get_random_str(k: int) -> str: + return "".join(sample(string.ascii_letters + string.digits, k)) + + @staticmethod + def get_auth_info() -> dict: + df = CONSOLE_DIR / "data.json" + if not df.is_file(): + try: + with open(df, "w", encoding="utf-8") as w: + w.write(json.dumps({})) + except WriteFileError: + raise WriteFileError("Writing file: " + str(df) + " failed!") + + base_data: dict = json.loads(df.read_bytes()) + data = base_data.get("data", None) + if not data: + return {"data": None} + return data diff --git a/ATRI/plugins/console/driver/__init__.py b/ATRI/plugins/console/driver/__init__.py new file mode 100644 index 0000000..fe4398a --- /dev/null +++ b/ATRI/plugins/console/driver/__init__.py @@ -0,0 +1,56 @@ +from nonebot import get_driver +from nonebot.drivers.fastapi import Driver + +from fastapi.middleware.cors import CORSMiddleware + +from .view import ( + handle_auther, + handle_base_uri, + handle_control_service, + handle_edit_block, + handle_get_block_list, + handle_get_service_list, + handle_runtime_info, + handle_message_deal_info, +) + + +CONSOLE_API_URI = "/capi" # base point +CONSOLE_API_AUTH_URI = "/capi/auth" # 验证后台许可 +CONSOLE_API_RUNTIME_URI = "/capi/runtime" # 获取运行占用信息 +CONSOLE_API_MESSAGE_URI = "/capi/message" # 获取信息处理信息 + +CONSOLE_API_SERVICE_LIST_URI = "/capi/service/list" # 获取服务列表 +CONSOLE_API_SERVICE_CONTROL_URI = "/capi/service/control" # 对服务作出修改 + +CONSOLE_API_BLOCK_LIST_URI = "/capi/block/list" # 获取封禁列表 +CONSOLE_API_BLOCK_EDIT_URI = "/capi/block/edit" # 编辑封禁列表 + + +def register_routes(driver: Driver): + app = driver.server_app + + origins = ["*"] + app.add_middleware( + CORSMiddleware, + allow_origins=origins, + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], + ) + + app.get(CONSOLE_API_URI)(handle_base_uri) + app.get(CONSOLE_API_AUTH_URI)(handle_auther) + app.get(CONSOLE_API_RUNTIME_URI)(handle_runtime_info) + app.get(CONSOLE_API_MESSAGE_URI)(handle_message_deal_info) + + app.get(CONSOLE_API_SERVICE_LIST_URI)(handle_get_service_list) + app.get(CONSOLE_API_SERVICE_CONTROL_URI)(handle_control_service) + + app.get(CONSOLE_API_BLOCK_LIST_URI)(handle_get_block_list) + app.get(CONSOLE_API_BLOCK_EDIT_URI)(handle_edit_block) + + +def init(): + driver = get_driver() + register_routes(driver) # type: ignore diff --git a/ATRI/plugins/console/driver/api.py b/ATRI/plugins/console/driver/api.py new file mode 100644 index 0000000..ba5405a --- /dev/null +++ b/ATRI/plugins/console/driver/api.py @@ -0,0 +1,193 @@ +import os +import json +import time +import psutil +from pathlib import Path +from datetime import datetime + +from ATRI.service import ServiceTools, SERVICES_DIR +from ATRI.exceptions import GetStatusError, ReadFileError, WriteFileError +from ..models import PlatformRuntimeInfo, BotRuntimeInfo, ServiceInfo + + +def get_processing_data() -> tuple: + try: + p_cpu = psutil.cpu_percent(interval=1) + p_mem = psutil.virtual_memory().percent + disk = psutil.disk_usage("/").percent + inte_send = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore + inte_recv = psutil.net_io_counters().bytes_recv / 1000000 # type: ignore + + process = psutil.Process(os.getpid()) + b_cpu = process.cpu_percent(interval=1) + b_mem = process.memory_percent(memtype="rss") + + now = time.time() + boot = psutil.boot_time() + b = process.create_time() + boot_time = str( + datetime.utcfromtimestamp(now).replace(microsecond=0) + - datetime.utcfromtimestamp(boot).replace(microsecond=0) + ) + bot_time = str( + datetime.utcfromtimestamp(now).replace(microsecond=0) + - datetime.utcfromtimestamp(b).replace(microsecond=0) + ) + except GetStatusError: + raise GetStatusError("Getting runtime failed.") + + if p_cpu > 90: # type: ignore + msg = "咱感觉有些头晕..." + if p_mem > 90: + msg = "咱感觉有点头晕并且有点累..." + elif p_mem > 90: + msg = "咱感觉有点累..." + elif disk > 90: + msg = "咱感觉身体要被塞满了..." + else: + msg = "アトリは、高性能ですから！" + + return ( + PlatformRuntimeInfo( + stat_msg=msg, + cpu_percent=str(p_cpu), + mem_percent=p_mem, + disk_percent=str(disk), + inte_send=str(inte_send), + inte_recv=str(inte_recv), + boot_time=boot_time, + ).dict(), + BotRuntimeInfo( + cpu_percent=str(b_cpu), mem_percent=str(b_mem), bot_run_time=bot_time + ).dict(), + ) + + +def get_service_list() -> dict: + result = dict() + + files = os.listdir(SERVICES_DIR) + for f in files: + # Thank you, MacOS + if f == ".DS_Store": + continue + + serv_path = SERVICES_DIR / f + data = json.loads(serv_path.read_bytes()) + + serv_name = data["service"] + serv_docs = data["docs"] + serv_is_enabled = data["enabled"] + serv_disable_user = data["disable_user"] + serv_disable_group = data["disable_group"] + + result[serv_name] = ServiceInfo( + service_name=serv_name, + service_docs=serv_docs, + is_enabled=serv_is_enabled, + disable_user=serv_disable_user, + disable_group=serv_disable_group, + ).dict() + + return result + + +def control_service( + serv_name: str, is_enab: int, enab_u: str, enab_g: str, disab_u: str, disab_g: str +) -> tuple: + try: + serv_data = ServiceTools().load_service(serv_name) + except ReadFileError: + return False, dict() + + if is_enab != 1: + if is_enab == 0: + serv_data["enabled"] = False + else: + serv_data["enabled"] = True + + if enab_u: + if enab_u not in serv_data["disable_user"]: + return False, {"msg": "Target not in list"} + serv_data["disable_user"].remove(enab_u) + + if enab_g: + if enab_g not in serv_data["disable_user"]: + return False, {"msg": "Target not in list"} + serv_data["disable_group"].remove(enab_g) + + if disab_u: + if disab_u in serv_data["disable_user"]: + return False, {"msg": "Target already exists in list"} + serv_data["disable_user"].append(disab_u) + + if disab_g: + if disab_g in serv_data["disable_user"]: + return False, {"msg": "Target already exists in list"} + serv_data["disable_group"].append(disab_g) + + try: + ServiceTools().save_service(serv_data, serv_name) + except WriteFileError: + return False, dict() + + return True, serv_data + + +MANEGE_DIR = Path(".") / "data" / "database" / "manege" + + +def get_block_list() -> dict: + u_f = "block_user.json" + path = MANEGE_DIR / u_f + u_data = json.loads(path.read_bytes()) + + g_f = "block_group.json" + path = MANEGE_DIR / g_f + g_data = json.loads(path.read_bytes()) + + return {"user": u_data, "group": g_data} + + +def edit_block_list(is_enab: int, user_id: str, group_id: str) -> tuple: + d = get_block_list() + u_d = d["user"] + g_d = d["group"] + + now_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S") + + if is_enab: + if user_id: + if user_id in u_d: + return False, {"msg": "Target already exists in list"} + u_d[user_id] = now_time + + if group_id: + if group_id in g_d: + return False, {"msg": "Target already exists in list"} + g_d[group_id] = now_time + else: + if user_id: + if user_id not in u_d: + return False, {"msg": "Target not in list"} + del u_d[user_id] + + if group_id: + if group_id not in g_d: + return False, {"msg": "Target not in list"} + del g_d[group_id] + + try: + u_f = "block_user.json" + path = MANEGE_DIR / u_f + with open(path, "w", encoding="utf-8") as w: + w.write(json.dumps(u_d)) + + g_f = "block_group.json" + path = MANEGE_DIR / g_f + with open(path, "w", encoding="utf-8") as w: + w.write(json.dumps(g_d)) + except WriteFileError: + return False, dict() + + return True, {"user": u_d, "group": g_d} diff --git a/ATRI/plugins/console/driver/view.py b/ATRI/plugins/console/driver/view.py new file mode 100644 index 0000000..624e41f --- /dev/null +++ b/ATRI/plugins/console/driver/view.py @@ -0,0 +1,101 @@ +from ..data_source import Console +from ..listener import get_message_deal_info +from .api import ( + control_service, + edit_block_list, + get_block_list, + get_processing_data, + get_service_list, +) + + +def auth_token(token: str) -> tuple: + auth_data: dict = Console().get_auth_info() + if not auth_data.get("token", None): + return False, {"status": 500, "msg": "This bot is not create auth data yet."} + _token = auth_data["token"] + if token != _token: + return False, {"status": 403, "msg": "Token error, please check again."} + else: + return True, {"status": 200, "msg": "OK"} + + +def handle_base_uri(): + return {"status": 204, "msg": "This path just for console load."} + + +def handle_auther(token: str): + auth, data = auth_token(token) + return data if auth else data + + +def handle_runtime_info(token: str): + auth, data = auth_token(token) + if not auth: + return data + + plat, bot = get_processing_data() + return {"status": 200, "data": {"platform": plat, "bot": bot}} + + +def handle_message_deal_info(token: str): + auth, data = auth_token(token) + if not auth: + return data + + return {"status": 200, "data": get_message_deal_info()} + + +def handle_get_service_list(token: str): + auth, data = auth_token(token) + if not auth: + return data + + return {"status": 200, "data": get_service_list()} + + +def handle_control_service( + token: str, + service: str, + is_enabled: int = 1, + enabled_user: str = str(), + enabled_group: str = str(), + disable_user: str = str(), + disable_group: str = str(), +): + auth, data = auth_token(token) + if not auth: + return data + + is_ok, data = control_service( + service, is_enabled, enabled_user, enabled_group, disable_user, disable_group + ) + if not is_ok: + return {"status": 422, "msg": "Dealing service data failed"} + + return {"status": 200, "data": data} + + +def handle_get_block_list(token: str): + auth, data = auth_token(token) + if not auth: + return data + + return {"status": 200, "data": get_block_list()} + + +def handle_edit_block( + token: str, + is_enabled: bool, + user_id: str = str(), + group_id: str = str(), +): + auth, data = auth_token(token) + if not auth: + return data + + is_ok, data = edit_block_list(is_enabled, user_id, group_id) + if not is_ok: + return {"status": 422, "msg": "Dealing block data failed"} + + return {"status": 200, "data": data} diff --git a/ATRI/plugins/console/listener.py b/ATRI/plugins/console/listener.py new file mode 100644 index 0000000..0ae5c25 --- /dev/null +++ b/ATRI/plugins/console/listener.py @@ -0,0 +1,49 @@ +from typing import Optional + +from nonebot.message import run_postprocessor + +from ATRI.utils.apscheduler import scheduler +from .models import MessageDealerInfo + + +recv_msg = int() +deal_msg = int() +failed_deal_msg = int() + +total_r_m = int() +total_d_m = int() +total_f_m = int() + + +@run_postprocessor +async def _(exception: Optional[Exception]): + global recv_msg, deal_msg, failed_deal_msg, total_r_m, total_d_m, total_f_m + + if exception: + failed_deal_msg += 1 + total_f_m += 1 + else: + deal_msg += 1 + total_d_m += 1 + + recv_msg += 1 + total_r_m += 1 + + +def get_message_deal_info() -> dict: + return MessageDealerInfo( + recv_msg=str(recv_msg), + deal_msg=str(deal_msg), + failed_deal_msg=str(failed_deal_msg), + total_r_m=str(total_r_m), + total_d_m=str(total_d_m), + total_f_m=str(total_f_m), + ).dict() + + +@scheduler.scheduled_job("interval", name="信息数据重置", seconds=15, misfire_grace_time=1) # type: ignore +async def _(): + global recv_msg, deal_msg, failed_deal_msg + recv_msg = int() + deal_msg = int() + failed_deal_msg = int() diff --git a/ATRI/plugins/console/models.py b/ATRI/plugins/console/models.py new file mode 100644 index 0000000..973d05d --- /dev/null +++ b/ATRI/plugins/console/models.py @@ -0,0 +1,40 @@ +from pydantic import BaseModel + + +class AuthData(BaseModel): + ip: str + port: str + token: str + + +class PlatformRuntimeInfo(BaseModel): + stat_msg: str + cpu_percent: str + mem_percent: str + disk_percent: str + inte_send: str + inte_recv: str + boot_time: str + + +class BotRuntimeInfo(BaseModel): + cpu_percent: str + mem_percent: str + bot_run_time: str + + +class MessageDealerInfo(BaseModel): + recv_msg: str + deal_msg: str + failed_deal_msg: str + total_r_m: str + total_d_m: str + total_f_m: str + + +class ServiceInfo(BaseModel): + service_name: str + service_docs: str + is_enabled: bool + disable_user: list + disable_group: list -- cgit v1.2.3