fix rss api

app/api/endpoints/rss.py

@@ -2,17 +2,26 @@ from typing import List, Any
 
 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
+from starlette.background import BackgroundTasks
 
 from app import schemas
 from app.chain.rss import RssChain
 from app.core.security import verify_token
 from app.db import get_db
 from app.db.models.rss import Rss
+from app.helper.rss import RssHelper
 from app.schemas import MediaType
 
 router = APIRouter()
 
 
+def start_rss_refresh(rssid: int = None):
+    """
+    启动自定义订阅刷新
+    """
+    RssChain().refresh(rssid=rssid, manual=True)
+
+
 @router.get("/", summary="所有自定义订阅", response_model=List[schemas.Rss])
 def read_rsses(
         db: Session = Depends(get_db),
@@ -65,6 +74,41 @@ def update_rss(
     return schemas.Response(success=True)
 
 
+@router.get("/preview/{rssid}", summary="预览自定义订阅", response_model=List[schemas.TorrentInfo])
+def preview_rss(
+        rssid: int,
+        db: Session = Depends(get_db),
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据ID查询自定义订阅RSS报文
+    """
+    rssinfo: Rss = Rss.get(db, rssid)
+    if not rssinfo:
+        return []
+    torrents = RssHelper.parse(rssinfo.url, proxy=True if rssinfo.proxy else False) or []
+    return [schemas.TorrentInfo(
+        title=t.get("title"),
+        description=t.get("description"),
+        enclosure=t.get("enclosure"),
+        size=t.get("size"),
+        page_url=t.get("link"),
+        pubdate=t["pubdate"].strftime("%Y-%m-%d %H:%M:%S") if t.get("pubdate") else None,
+    ) for t in torrents]
+
+
+@router.get("/refresh/{rssid}", summary="刷新自定义订阅", response_model=schemas.Response)
+def refresh_rss(
+        rssid: int,
+        background_tasks: BackgroundTasks,
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据ID刷新自定义订阅
+    """
+    background_tasks.add_task(start_rss_refresh,
+                              rssid=rssid)
+    return schemas.Response(success=True)
+
+
 @router.get("/{rssid}", summary="查询自定义订阅详情", response_model=schemas.Rss)
 def read_rss(
         rssid: int,