fix: rss.parser set to default value when subscribe Bangumi

This commit is contained in:
KotaHv
2024-01-24 21:09:25 +08:00
parent 6973773b53
commit 2149182a54
4 changed files with 14 additions and 7 deletions

View File

@@ -194,7 +194,7 @@ async def download_collection(data: Bangumi):
@router.post(
"/subscribe", response_model=APIResponse, dependencies=[Depends(get_current_user)]
)
async def subscribe(data: Bangumi):
async def subscribe(data: Bangumi, rss: RSSItem):
with SeasonCollector() as collector:
resp = collector.subscribe_season(data)
resp = collector.subscribe_season(data, parser=rss.parser)
return u_response(resp)

View File

@@ -46,12 +46,15 @@ class SeasonCollector(DownloadClient):
)
@staticmethod
def subscribe_season(data: Bangumi):
def subscribe_season(data: Bangumi, parser: str = "mikan"):
with RSSEngine() as engine:
data.added = True
data.eps_collect = True
engine.add_rss(
rss_link=data.rss_link, name=data.official_title, aggregate=False
rss_link=data.rss_link,
name=data.official_title,
aggregate=False,
parser=parser,
)
result = engine.download_bangumi(data)
engine.bangumi.add(data)