feat rss

app/api/endpoints/rss.py

@@ -0,0 +1,68 @@
+from typing import List, Any
+
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app import schemas
+from app.core.security import verify_token
+from app.db import get_db
+from app.db.models.rss import Rss
+
+router = APIRouter()
+
+
+@router.get("/", summary="所有自定义订阅", response_model=List[schemas.Rss])
+def read_rsses(
+        db: Session = Depends(get_db),
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    查询所有自定义订阅
+    """
+    return Rss.list(db)
+
+
+@router.post("/", summary="新增自定义订阅", response_model=schemas.Response)
+def create_rss(
+        *,
+        db: Session = Depends(get_db),
+        rss_in: schemas.Rss,
+        _: schemas.TokenPayload = Depends(verify_token)
+) -> Any:
+    """
+    新增自定义订阅
+    """
+    rss = Rss.get_by_tmdbid(db, tmdbid=rss_in.tmdbid, season=rss_in.season)
+    if rss:
+        return schemas.Response(success=False, message="自定义订阅已存在")
+    rss = Rss(**rss_in.dict())
+    rss.create(db)
+    return schemas.Response(success=True)
+
+
+@router.put("/", summary="更新自定义订阅", response_model=schemas.Response)
+def update_rss(
+        *,
+        rss_in: schemas.Rss,
+        db: Session = Depends(get_db),
+        _: schemas.TokenPayload = Depends(verify_token)
+) -> Any:
+    """
+    更新自定义订阅信息
+    """
+    rss = Rss.get(db, rss_in.id)
+    if not rss:
+        return schemas.Response(success=False, message="自定义订阅不存在")
+
+    rss.update(db, rss_in.dict())
+    return schemas.Response(success=True)
+
+
+@router.get("/{rssid}", summary="查询订阅详情", response_model=schemas.Rss)
+def read_rss(
+        rssid: int,
+        db: Session = Depends(get_db),
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据ID查询自定义订阅详情
+    """
+    return Rss.get(db, rssid)