From 762c1bec8d00ef0910b15c3f9f7b8d2e6abbce8a Mon Sep 17 00:00:00 2001 From: jxxghp Date: Wed, 14 Jun 2023 10:53:20 +0800 Subject: [PATCH] add sonarr api --- app/api/servarr.py | 119 +++++++++++++++++++++++++++++++++++-- app/db/models/subscribe.py | 9 +++ 2 files changed, 122 insertions(+), 6 deletions(-) diff --git a/app/api/servarr.py b/app/api/servarr.py index a86fea39..58669582 100644 --- a/app/api/servarr.py +++ b/app/api/servarr.py @@ -8,7 +8,7 @@ from app.chain.subscribe import SubscribeChain from app.core.config import settings from app.db import get_db from app.db.models.subscribe import Subscribe -from app.schemas import RadarrMovie +from app.schemas import RadarrMovie, SonarrSeries from app.utils.types import MediaType from version import APP_VERSION @@ -331,7 +331,7 @@ async def arr_remove_movie(apikey: str, mid: int, db: Session = Depends(get_db)) @arr_router.get("/series", response_model=List[schemas.SonarrSeries]) -async def arr_series(apikey: str) -> Any: +async def arr_series(apikey: str, db: Session = Depends(get_db)) -> Any: """ 查询Sonarr剧集 """ @@ -442,10 +442,35 @@ async def arr_series(apikey: str) -> Any: status_code=403, detail="认证失败！", ) + # 查询所有电视剧订阅 + result = [] + subscribes = Subscribe.list(db) + for subscribe in subscribes: + if subscribe.type != "电视剧": + continue + result.append(SonarrSeries( + id=subscribe.id, + title=subscribe.name, + seasonCount=1, + seasons=[{ + "seasonNumber": subscribe.season, + "monitored": True, + }], + year=subscribe.year, + isAvailable=True, + monitored=True, + tmdbId=subscribe.tmdbid, + profileId=1, + languageProfileId=1, + qualityProfileId=1, + added=True, + hasFile=False, + )) + return result @arr_router.get("/series/lookup") -async def arr_series_lookup(apikey: str, term: str) -> Any: +async def arr_series_lookup(apikey: str, term: str, db: Session = Depends(get_db)) -> Any: """ 查询Sonarr剧集 term: `tvdb:${id}` title """ @@ -454,10 +479,45 @@ async def arr_series_lookup(apikey: str, term: str) -> Any: status_code=403, detail="认证失败！", ) + if not term.startswith("tvdb:"): + title = term + subscribe = Subscribe.get_by_title(db, title) + else: + tmdbid = term.replace("tvdb:", "") + subscribe = Subscribe.get_by_tmdbid(db, int(tmdbid)) + if subscribe: + return [SonarrSeries( + id=subscribe.id, + title=subscribe.name, + seasonCount=1, + seasons=[{ + "seasonNumber": subscribe.season, + "monitored": True, + }], + year=subscribe.year, + isAvailable=True, + monitored=True, + tvdbId=subscribe.tvdbid, + profileId=1, + languageProfileId=1, + qualityProfileId=1, + added=True, + hasFile=False, + )] + else: + return [SonarrSeries( + isAvailable=False, + monitored=False, + profileId=1, + languageProfileId=1, + qualityProfileId=1, + added=False, + hasFile=False, + )] @arr_router.get("/series/{tid}") -async def arr_serie(apikey: str) -> Any: +async def arr_serie(apikey: str, tid: int, db: Session = Depends(get_db)) -> Any: """ 查询Sonarr剧集 """ @@ -466,10 +526,35 @@ async def arr_serie(apikey: str) -> Any: status_code=403, detail="认证失败！", ) + subscribe = Subscribe.get(db, tid) + if subscribe: + return SonarrSeries( + id=subscribe.id, + title=subscribe.name, + seasonCount=1, + seasons=[{ + "seasonNumber": subscribe.season, + "monitored": True, + }], + year=subscribe.year, + isAvailable=True, + monitored=True, + tvdbId=subscribe.tvdbid, + profileId=1, + languageProfileId=1, + qualityProfileId=1, + added=True, + hasFile=False, + ) + else: + raise HTTPException( + status_code=404, + detail="未找到该电视剧！" + ) @arr_router.post("/series") -async def arr_add_series(apikey: str, title: str, seasons: list, year: int) -> Any: +async def arr_add_series(apikey: str, tv: schemas.SonarrSeries) -> Any: """ 新增Sonarr剧集订阅 """ @@ -478,10 +563,23 @@ async def arr_add_series(apikey: str, title: str, seasons: list, year: int) -> A status_code=403, detail="认证失败！", ) + sid = SubscribeChain().process(title=tv.title, + year=str(tv.year) if tv.year else None, + mtype=MediaType.TV, + userid="Seerr") + if sid: + return { + "id": sid + } + else: + raise HTTPException( + status_code=500, + detail="添加订阅失败！" + ) @arr_router.delete("/series/{tid}") -async def arr_remove_series(apikey: str, tid: int) -> Any: +async def arr_remove_series(apikey: str, tid: int, db: Session = Depends(get_db)) -> Any: """ 删除Sonarr剧集订阅 """ @@ -490,3 +588,12 @@ async def arr_remove_series(apikey: str, tid: int) -> Any: status_code=403, detail="认证失败！", ) + subscribe = Subscribe.get(db, tid) + if subscribe: + subscribe.delete(db, tid) + return {"success": True} + else: + raise HTTPException( + status_code=404, + detail="未找到该电视剧！" + ) diff --git a/app/db/models/subscribe.py b/app/db/models/subscribe.py index f2beed32..ac16e6c5 100644 --- a/app/db/models/subscribe.py +++ b/app/db/models/subscribe.py @@ -14,6 +14,7 @@ class Subscribe(Base): type = Column(String) keyword = Column(String) tmdbid = Column(Integer, index=True) + tvdbid = Column(Integer, index=True) doubanid = Column(String) season = Column(Integer) image = Column(String) @@ -41,3 +42,11 @@ class Subscribe(Base): @staticmethod def get_by_tmdbid(db: Session, tmdbid: int): return db.query(Subscribe).filter(Subscribe.tmdbid == tmdbid).first() + + @staticmethod + def get_by_title(db: Session, title: str): + return db.query(Subscribe).filter(Subscribe.name == title).first() + + @staticmethod + def get_by_tvdbid(db: Session, tvdbid: int): + return db.query(Subscribe).filter(Subscribe.tvdbid == tvdbid).first()