chore: update func name in rss engine.

This commit is contained in:
EstrellaXD
2023-08-06 23:49:39 +08:00
parent a9268ffe2e
commit 2e46676fa9
3 changed files with 8 additions and 3 deletions

View File

@@ -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"])

View File

@@ -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)

View File

@@ -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)