codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix list_torrents

Browse Source
This commit is contained in:
jxxghp 2023-06-06 10:58:57 +08:00
parent a7251ecf0c
commit e8178a1f55
3 changed files with 51 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/modules/__init__.py
View File

@ -131,7 +131,7 @@ class _ModuleBase(metaclass=ABCMeta):
"""
pass
def list_torrents(self, status: TorrentStatus) -> List[dict]:
def list_torrents(self, status: TorrentStatus) -> Optional[List[dict]]:
"""
获取下载器种子列表
:param status: 种子状态
@ -139,10 +139,10 @@ class _ModuleBase(metaclass=ABCMeta):
"""
pass
def remove_torrents(self, status: Union[str, list]) -> bool:
def remove_torrents(self, hashs: Union[str, list]) -> bool:
"""
删除下载器种子
:param status: 种子状态
:param hashs: 种子Hash
:return: bool
"""
pass

25
app/modules/qbittorrent/__init__.py
View File

@ -1,10 +1,11 @@
from pathlib import Path
from typing import Set, Tuple, Optional, Union
from typing import Set, Tuple, Optional, Union, List
from app.core import settings, MetaInfo
from app.modules import _ModuleBase
from app.modules.qbittorrent.qbittorrent import Qbittorrent
from app.utils.string import StringUtils
from app.utils.types import TorrentStatus
class QbittorrentModule(_ModuleBase):
@ -73,3 +74,25 @@ class QbittorrentModule(_ModuleBase):
return torrent_hash, f"添加下载成功，已选择集数：{sucess_epidised}"
else:
return torrent_hash, "添加下载成功"
def list_torrents(self, status: TorrentStatus) -> Optional[List[dict]]:
"""
获取下载器种子列表
:param status: 种子状态
:return: 下载器中符合状态的种子列表
"""
if status == TorrentStatus.COMPLETE:
torrents = self.qbittorrent.get_completed_torrents()
elif status == TorrentStatus.DOWNLOADING:
torrents = self.qbittorrent.get_downloading_torrents()
else:
return None
return torrents
def remove_torrents(self, hashs: Union[str, list]) -> bool:
"""
删除下载器种子
:param hashs: 种子Hash
:return: bool
"""
return self.qbittorrent.delete_torrents(delete_file=True, ids=hashs)

25
app/modules/transmission/__init__.py
View File

@ -1,9 +1,10 @@
from pathlib import Path
from typing import Set, Tuple, Optional, Union
from typing import Set, Tuple, Optional, Union, List
from app.core import settings, MetaInfo
from app.modules import _ModuleBase
from app.modules.transmission.transmission import Transmission
from app.utils.types import TorrentStatus
class TransmissionModule(_ModuleBase):
@ -68,3 +69,25 @@ class TransmissionModule(_ModuleBase):
self.transmission.start_torrents(torrent_hash)
else:
return torrent_hash, "添加下载任务成功"
def list_torrents(self, status: TorrentStatus) -> Optional[List[dict]]:
"""
获取下载器种子列表
:param status: 种子状态
:return: 下载器中符合状态的种子列表
"""
if status == TorrentStatus.COMPLETE:
torrents = self.transmission.get_completed_torrents()
elif status == TorrentStatus.DOWNLOADING:
torrents = self.transmission.get_completed_torrents()
else:
return None
return torrents
def remove_torrents(self, hashs: Union[str, list]) -> bool:
"""
删除下载器种子
:param hashs: 种子Hash
:return: bool
"""
return self.transmission.delete_torrents(delete_file=True, ids=hashs)