feat site reset

app/api/endpoints/site.py

@@ -72,6 +72,18 @@ def cookie_cloud_sync(background_tasks: BackgroundTasks,
     return schemas.Response(success=True, message="CookieCloud同步任务已启动！")
 
 
+@router.get("/reset", summary="重置站点", response_model=schemas.Response)
+def cookie_cloud_sync(background_tasks: BackgroundTasks,
+                      db: Session = Depends(get_db),
+                      _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    清空所有站点数据并重新同步CookieCloud站点信息
+    """
+    Site.reset(db)
+    CookieCloudChain().process(manual=True)
+    return schemas.Response(success=True, message="站点已重置！")
+
+
 @router.get("/cookie/{site_id}", summary="更新站点Cookie&UA", response_model=schemas.Response)
 def update_cookie(
         site_id: int,

app/chain/cookiecloud.py

@@ -58,6 +58,8 @@ class CookieCloudChain(ChainBase):
         cookies, msg = self.cookiecloud.download()
         if not cookies:
             logger.error(f"CookieCloud同步失败：{msg}")
+            if manual:
+                self.message.put(f"CookieCloud同步失败： {msg}")
             return False, msg
         # 保存Cookie或新增站点
         _update_count = 0

app/db/models/site.py

@@ -57,3 +57,8 @@ class Site(Base):
     @staticmethod
     def list_order_by_pri(db: Session):
         return db.query(Site).order_by(Site.pri).all()
+
+    @staticmethod
+    def reset(db: Session):
+        db.query(Site).delete()
+        db.commit()