fix commands

app/api/endpoints/site.py

@@ -5,6 +5,7 @@ from sqlalchemy.orm import Session
 
 from app import schemas
 from app.chain.cookiecloud import CookieCloudChain
+from app.chain.site import SiteChain
 from app.db import get_db
 from app.db.models.site import Site
 from app.db.models.user import User
@@ -24,10 +25,10 @@ async def read_sites(db: Session = Depends(get_db),
 
 @router.put("/", response_model=schemas.Site)
 async def update_site(
-    *,
-    db: Session = Depends(get_db),
-    site_in: schemas.Site,
-    _: User = Depends(get_current_active_superuser),
+        *,
+        db: Session = Depends(get_db),
+        site_in: schemas.Site,
+        _: User = Depends(get_current_active_superuser),
 ) -> Any:
     """
     更新站点信息
@@ -42,6 +43,24 @@ async def update_site(
     return site
 
 
+@router.get("/{site_id}", response_model=schemas.Site)
+async def read_site(
+        site_id: int,
+        db: Session = Depends(get_db),
+        _: User = Depends(get_current_active_user),
+) -> Any:
+    """
+    获取站点信息
+    """
+    site = Site.get(db, site_id)
+    if not site:
+        raise HTTPException(
+            status_code=404,
+            detail=f"站点 {site_id} 不存在",
+        )
+    return site
+
+
 @router.get("/cookiecloud", response_model=schemas.Response)
 async def cookie_cloud_sync(_: User = Depends(get_current_active_user)) -> Any:
     """
@@ -51,3 +70,30 @@ async def cookie_cloud_sync(_: User = Depends(get_current_active_user)) -> Any:
     if not status:
         return {"success": False, "message": error_msg}
     return {"success": True, "message": error_msg}
+
+
+@router.get("/cookie", response_model=schemas.Response)
+async def update_cookie(
+        site_id: int,
+        username: str,
+        password: str,
+        db: Session = Depends(get_db),
+        _: User = Depends(get_current_active_user)) -> Any:
+    """
+    使用用户密码更新站点Cookie
+    """
+    # 查询站点
+    site_info = Site.get(db, site_id)
+    if not site_info:
+        raise HTTPException(
+            status_code=404,
+            detail=f"站点 {site_id} 不存在！",
+        )
+    # 更新Cookie
+    status, msg = SiteChain().update_cookie(site_info=site_info,
+                                            username=username,
+                                            password=password)
+    if not status:
+        return {"success": False, "message": msg}
+    else:
+        return {"success": True, "message": msg}