fix site userdata

This commit is contained in:
jxxghp
2024-09-12 15:44:31 +08:00
parent 53195457c7
commit 73407825f5
4 changed files with 47 additions and 11 deletions

View File

@@ -14,6 +14,7 @@ from app.db.models import User
from app.db.models.site import Site
from app.db.models.siteicon import SiteIcon
from app.db.models.sitestatistic import SiteStatistic
from app.db.models.siteuserdata import SiteUserData
from app.db.systemconfig_oper import SystemConfigOper
from app.db.user_oper import get_current_active_superuser
from app.helper.sites import SitesHelper
@@ -164,7 +165,7 @@ def update_cookie(
return schemas.Response(success=state, message=message)
@router.get("/userdata/{site_id}", summary="更新站点用户数据", response_model=schemas.Response)
@router.post("/userdata/{site_id}", summary="更新站点用户数据", response_model=schemas.Response)
def refresh_userdata(
site_id: int,
db: Session = Depends(get_db),
@@ -185,6 +186,27 @@ def refresh_userdata(
return schemas.Response(success=True, data=user_data)
@router.get("/userdata/{site_id}", summary="查询站点用户数据", response_model=schemas.Response)
def read_userdata(
site_id: int,
workdate: str = None,
db: Session = Depends(get_db),
_: schemas.TokenPayload = Depends(get_current_active_superuser)) -> Any:
"""
查询站点用户数据
"""
site = Site.get(db, site_id)
if not site:
raise HTTPException(
status_code=404,
detail=f"站点 {site_id} 不存在",
)
user_data = SiteUserData.get_by_domain(db, domain=site.domain, workdate=workdate)
if not user_data:
return schemas.Response(success=False, data=[])
return schemas.Response(success=True, data=user_data)
@router.get("/test/{site_id}", summary="连接测试", response_model=schemas.Response)
def test_site(site_id: int,
db: Session = Depends(get_db),

View File

@@ -317,7 +317,7 @@ async def seerr_subscribe(request: Request, background_tasks: BackgroundTasks,
@router.get("/history/{mtype}", summary="查询订阅历史", response_model=List[schemas.Subscribe])
def read_subscribe(
def subscribe_history(
mtype: str,
page: int = 1,
count: int = 30,