107 lines
4.4 KiB
Python
107 lines
4.4 KiB
Python
from pathlib import Path
|
||
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):
|
||
qbittorrent: Qbittorrent = None
|
||
|
||
def init_module(self) -> None:
|
||
self.qbittorrent = Qbittorrent()
|
||
|
||
def init_setting(self) -> Tuple[str, Union[str, bool]]:
|
||
return "DOWNLOADER", "qbittorrent"
|
||
|
||
def download(self, torrent_path: Path, cookie: str,
|
||
episodes: Set[int] = None) -> Optional[Tuple[Optional[str], str]]:
|
||
"""
|
||
根据种子文件，选择并添加下载任务
|
||
:param torrent_path: 种子文件地址
|
||
:param cookie: cookie
|
||
:param episodes: 需要下载的集数
|
||
:return: 种子Hash
|
||
"""
|
||
if not torrent_path.exists():
|
||
return None, f"种子文件不存在：{torrent_path}"
|
||
# 生成随机Tag
|
||
tag = StringUtils.generate_random_str(10)
|
||
if settings.TORRENT_TAG:
|
||
tag = [tag, settings.TORRENT_TAG]
|
||
# 如果要选择文件则先暂停
|
||
is_paused = True if episodes else False
|
||
# 添加任务
|
||
state = self.qbittorrent.add_torrent(content=torrent_path.read_bytes(),
|
||
download_dir=settings.DOWNLOAD_PATH,
|
||
is_paused=is_paused,
|
||
tag=tag,
|
||
cookie=cookie)
|
||
if not state:
|
||
return None, f"添加种子任务失败：{torrent_path}"
|
||
else:
|
||
# 获取种子Hash
|
||
torrent_hash = self.qbittorrent.get_torrent_id_by_tag(tag=tag)
|
||
if not torrent_hash:
|
||
return None, f"获取种子Hash失败：{torrent_path}"
|
||
else:
|
||
if is_paused:
|
||
# 种子文件
|
||
torrent_files = self.qbittorrent.get_files(torrent_hash)
|
||
if not torrent_files:
|
||
return torrent_hash, "获取种子文件失败，下载任务可能在暂停状态"
|
||
|
||
# 不需要的文件ID
|
||
file_ids = []
|
||
# 需要的集清单
|
||
sucess_epidised = []
|
||
|
||
for torrent_file in torrent_files:
|
||
file_id = torrent_file.get("id")
|
||
file_name = torrent_file.get("name")
|
||
meta_info = MetaInfo(file_name)
|
||
if not meta_info.get_episode_list() \
|
||
or not set(meta_info.get_episode_list()).issubset(episodes):
|
||
file_ids.append(file_id)
|
||
else:
|
||
sucess_epidised = list(set(sucess_epidised).union(set(meta_info.get_episode_list())))
|
||
if sucess_epidised and file_ids:
|
||
# 选择文件
|
||
self.qbittorrent.set_files(torrent_hash=torrent_hash, file_ids=file_ids, priority=0)
|
||
# 开始任务
|
||
self.qbittorrent.start_torrents(torrent_hash)
|
||
return torrent_hash, f"添加下载成功，已选择集数：{sucess_epidised}"
|
||
else:
|
||
return torrent_hash, "添加下载成功"
|
||
|
||
def transfer_completed(self, hashs: Union[str, list]) -> bool:
|
||
"""
|
||
转移完成后的处理
|
||
:param hashs: 种子Hash
|
||
:return: 处理状态
|
||
"""
|
||
return self.qbittorrent.set_torrents_tag(ids=hashs, tag=['已整理'])
|
||
|
||
def list_torrents(self, status: TorrentStatus) -> Optional[List[dict]]:
|
||
"""
|
||
获取下载器种子列表
|
||
:param status: 种子状态
|
||
:return: 下载器中符合状态的种子列表
|
||
"""
|
||
if status == TorrentStatus.TRANSFER:
|
||
torrents = self.qbittorrent.get_transfer_torrents(tag=settings.TORRENT_TAG or None)
|
||
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)
|