suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-11-04 13:34:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ 整理代码结构

Browse Source
This commit is contained in:
suyiiyii 2024-12-12 11:22:11 +08:00
parent a83cf3ca79
commit b7ba0ac661
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
3 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nonebot_bison/admin_page/__init__.py
View File

@ -12,7 +12,7 @@ from nonebot.adapters.onebot.v11.event import PrivateMessageEvent
from .api import router as api_router from .api import router as api_router
from ..plugin_config import plugin_config from ..plugin_config import plugin_config
from .token_manager import token_manager as tm from .token_manager import token_manager as tm
from .api import metrics_router as metrics_router from ..metrics import metrics_router as metrics_router
if TYPE_CHECKING: if TYPE_CHECKING:
from nonebot.drivers.fastapi import Driver from nonebot.drivers.fastapi import Driver

10
nonebot_bison/admin_page/api.py
View File

@ -3,13 +3,11 @@ from typing import cast
import nonebot import nonebot
from fastapi import status from fastapi import status
from fastapi.routing import APIRouter from fastapi.routing import APIRouter
from starlette.responses import Response
from fastapi.param_functions import Depends from fastapi.param_functions import Depends
from fastapi.exceptions import HTTPException from fastapi.exceptions import HTTPException
from nonebot_plugin_saa import TargetQQGroup from nonebot_plugin_saa import TargetQQGroup
from nonebot_plugin_saa.auto_select_bot import get_bot from nonebot_plugin_saa.auto_select_bot import get_bot
from fastapi.security.oauth2 import OAuth2PasswordBearer from fastapi.security.oauth2 import OAuth2PasswordBearer
from prometheus_client import CONTENT_TYPE_LATEST, generate_latest
from ..types import WeightConfig from ..types import WeightConfig
from ..apis import check_sub_target from ..apis import check_sub_target
@ -285,11 +283,3 @@ async def get_cookie_valid(site_name: str, content: str) -> StatusResp:
return StatusResp(ok=True, msg="") return StatusResp(ok=True, msg="")
else: else:
return StatusResp(ok=False, msg="") return StatusResp(ok=False, msg="")
metrics_router = APIRouter(prefix="/api/metrics", tags=["metrics"])
@metrics_router.get("")
async def metrics():
return Response(media_type=CONTENT_TYPE_LATEST, content=generate_latest())

10
nonebot_bison/metrics.py
View File

@ -1,4 +1,6 @@
from prometheus_client import Counter from fastapi import APIRouter
from starlette.responses import Response
from prometheus_client import CONTENT_TYPE_LATEST, Counter, generate_latest
# Request counter # Request counter
request_counter = Counter("bison_request_counter", "The number of requests") request_counter = Counter("bison_request_counter", "The number of requests")
@ -7,3 +9,9 @@ success_counter = Counter("bison_success_counter", "The number of successful req
# Sent counter # Sent counter
sent_counter = Counter("bison_sent_counter", "The number of sent messages") sent_counter = Counter("bison_sent_counter", "The number of sent messages")
metrics_router = APIRouter(prefix="/api/metrics", tags=["metrics"])
@metrics_router.get("")
async def metrics():
return Response(media_type=CONTENT_TYPE_LATEST, content=generate_latest())