11import asyncio
22import logging
33
4+ from sqlalchemy import Engine
5+
46from module .database import Database , engine
57from module .downloader import DownloadQueue , download_queue
68from module .models import Bangumi , RSSItem , Torrent
@@ -40,14 +42,15 @@ async def pull_rss(self) -> list[Torrent]:
4042
4143
4244class RSSRefresh (BaseRefresh ):
45+
4346 def __init__ (self , rss_item : RSSItem , _engine = engine ):
4447 super ().__init__ (_engine )
4548 self .rss_item : RSSItem = rss_item
4649 self .bangumis : list [Bangumi ] = []
4750 self .url : str = rss_item .url
4851 self .bangumi : Bangumi | None = None
49- self .analyser = RSSAnalyser ()
50- self .download_queue = download_queue
52+ self .analyser : RSSAnalyser = RSSAnalyser ()
53+ self .download_queue : DownloadQueue = download_queue
5154
5255 async def download_rss (self ) -> list [Torrent ]:
5356 """下载 rss_item 对应的 torrents"""
@@ -151,7 +154,7 @@ def __init__(self, bangumi: Bangumi, _engine=engine):
151154 self .bangumi : Bangumi = bangumi
152155 self .url : str = bangumi .rss_link
153156 self .analyser = RSSAnalyser ()
154- self .download_queue :DownloadQueue = download_queue
157+ self .download_queue : DownloadQueue = download_queue
155158
156159 async def refresh (self ) -> list [Torrent ]:
157160 """刷新 bangumi 的 rss"""
@@ -214,8 +217,8 @@ class RSSEngine:
214217 """
215218
216219 def __init__ (self , _engine = engine ) -> None :
217- self .engine = _engine
218- self .queue :DownloadQueue = download_queue
220+ self .engine : Engine = _engine
221+ self .queue : DownloadQueue = download_queue
219222
220223 def get_active_rss (self ) -> list [RSSItem ]:
221224 """获取所有活跃的rss"""
@@ -224,7 +227,7 @@ def get_active_rss(self) -> list[RSSItem]:
224227 logger .debug (f"[RSS] get { len (rss_items )} active rss items" )
225228 return rss_items
226229
227- async def refresh_rss (self , rss_item : RSSItem ) -> list [ Torrent ] :
230+ async def refresh_rss (self , rss_item : RSSItem ) -> None :
228231 rssrefresh = RSSRefresh (rss_item = rss_item )
229232 await rssrefresh .find_new_bangumi ()
230233 await rssrefresh .download_rss ()
0 commit comments