113 lines
3.4 KiB
Python
113 lines
3.4 KiB
Python
import json
|
||
import json
|
||
import time
|
||
from datetime import datetime
|
||
from typing import Union
|
||
|
||
from fastapi import APIRouter, HTTPException, Depends
|
||
from fastapi.responses import StreamingResponse
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app import schemas
|
||
from app.core.config import settings
|
||
from app.core.security import verify_token
|
||
from app.db import get_db
|
||
from app.db.systemconfig_oper import SystemConfigOper
|
||
from app.helper.message import MessageHelper
|
||
from app.helper.progress import ProgressHelper
|
||
from app.utils.http import RequestUtils
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
@router.get("/progress/{process_type}", summary="实时进度")
|
||
def get_progress(process_type: str, token: str):
|
||
"""
|
||
实时获取处理进度，返回格式为SSE
|
||
"""
|
||
if not token or not verify_token(token):
|
||
raise HTTPException(
|
||
status_code=403,
|
||
detail="认证失败！",
|
||
)
|
||
|
||
progress = ProgressHelper()
|
||
|
||
def event_generator():
|
||
while True:
|
||
detail = progress.get(process_type)
|
||
yield 'data: %s\n\n' % json.dumps(detail)
|
||
time.sleep(0.2)
|
||
|
||
return StreamingResponse(event_generator(), media_type="text/event-stream")
|
||
|
||
|
||
@router.get("/setting/{key}", summary="查询系统设置", response_model=schemas.Response)
|
||
def get_setting(key: str,
|
||
db: Session = Depends(get_db),
|
||
_: schemas.TokenPayload = Depends(verify_token)):
|
||
"""
|
||
查询系统设置
|
||
"""
|
||
return schemas.Response(success=True, data={
|
||
"value": SystemConfigOper(db).get(key)
|
||
})
|
||
|
||
|
||
@router.post("/setting/{key}", summary="更新系统设置", response_model=schemas.Response)
|
||
def set_setting(key: str, value: Union[list, dict, str, int],
|
||
db: Session = Depends(get_db),
|
||
_: schemas.TokenPayload = Depends(verify_token)):
|
||
"""
|
||
更新系统设置
|
||
"""
|
||
SystemConfigOper(db).set(key, value)
|
||
return schemas.Response(success=True)
|
||
|
||
|
||
@router.get("/message", summary="实时消息")
|
||
def get_progress(token: str):
|
||
"""
|
||
实时获取系统消息，返回格式为SSE
|
||
"""
|
||
if not token or not verify_token(token):
|
||
raise HTTPException(
|
||
status_code=403,
|
||
detail="认证失败！",
|
||
)
|
||
|
||
message = MessageHelper()
|
||
|
||
def event_generator():
|
||
while True:
|
||
detail = message.get()
|
||
yield 'data: %s\n\n' % (detail or '')
|
||
time.sleep(3)
|
||
|
||
return StreamingResponse(event_generator(), media_type="text/event-stream")
|
||
|
||
|
||
@router.get("/nettest", summary="测试网络连通性")
|
||
def nettest(url: str,
|
||
proxy: bool,
|
||
_: schemas.TokenPayload = Depends(verify_token)):
|
||
"""
|
||
测试网络连通性
|
||
"""
|
||
# 记录开始的毫秒数
|
||
start_time = datetime.now()
|
||
result = RequestUtils(proxies=settings.PROXY if proxy else None).get_res(url)
|
||
# 计时结束的毫秒数
|
||
end_time = datetime.now()
|
||
# 计算相关秒数
|
||
if result and result.status_code == 200:
|
||
return schemas.Response(success=True, data={
|
||
"time": round((end_time - start_time).microseconds / 1000)
|
||
})
|
||
elif result:
|
||
return schemas.Response(success=False, message=f"错误码：{result.status_code}", data={
|
||
"time": round((end_time - start_time).microseconds / 1000)
|
||
})
|
||
else:
|
||
return schemas.Response(success=False, message="网络连接失败！")
|