From 2e46676fa9fe24ed6f5bd8ca937627f0bda338ca Mon Sep 17 00:00:00 2001 From: EstrellaXD Date: Sun, 6 Aug 2023 23:49:39 +0800 Subject: [PATCH] chore: update func name in rss engine. --- backend/src/module/api/rss.py | 1 + backend/src/module/core/sub_thread.py | 2 +- backend/src/module/rss/engine.py | 8 ++++++-- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/backend/src/module/api/rss.py b/backend/src/module/api/rss.py index 667118b9..4e4b1102 100644 --- a/backend/src/module/api/rss.py +++ b/backend/src/module/api/rss.py @@ -4,6 +4,7 @@ from fastapi.responses import JSONResponse from module.models import RSSItem from module.rss import RSSEngine from module.security import get_current_user +from module.downloader import DownloadClient router = APIRouter(prefix="/rss", tags=["rss"]) diff --git a/backend/src/module/core/sub_thread.py b/backend/src/module/core/sub_thread.py index 15d9f781..649a6884 100644 --- a/backend/src/module/core/sub_thread.py +++ b/backend/src/module/core/sub_thread.py @@ -29,7 +29,7 @@ class RSSThread(ProgramStatus): for rss in rss_list: self.analyser.rss_to_data(rss_link=rss.url, engine=engine) # Run RSS Engine - engine.run(client) + engine.refresh_rss(client) if settings.bangumi_manage.eps_complete: eps_complete() self.stop_event.wait(settings.program.rss_time) diff --git a/backend/src/module/rss/engine.py b/backend/src/module/rss/engine.py index d90ec2d8..299bd9b8 100644 --- a/backend/src/module/rss/engine.py +++ b/backend/src/module/rss/engine.py @@ -56,9 +56,13 @@ class RSSEngine(Database): return matched return None - def run(self, client: DownloadClient): + def refresh_rss(self, client: DownloadClient, rss_id: Optional[int] = None): # Get All RSS Items - rss_items: list[RSSItem] = self.rss.search_active() + if not rss_id: + rss_items: list[RSSItem] = self.rss.search_active() + else: + rss_item = self.rss.search_id(rss_id) + rss_items = [rss_item] if rss_item else [] # From RSS Items, get all torrents for rss_item in rss_items: new_torrents = self.pull_rss(rss_item)