init
This commit is contained in:
jxxghp
325
app/modules/qbittorrent/qbittorrent.py
Normal file
325
app/modules/qbittorrent/qbittorrent.py
Normal file
@ -0,0 +1,325 @@
|
||||
import time
|
||||
from pathlib import Path
|
||||
from typing import Optional, Union, Tuple
|
||||
|
||||
import qbittorrentapi
|
||||
from qbittorrentapi import TorrentFilesList
|
||||
from qbittorrentapi.client import Client
|
||||
|
||||
from app.core import settings
|
||||
from app.log import logger
|
||||
from app.utils.singleton import Singleton
|
||||
|
||||
|
||||
class Qbittorrent(metaclass=Singleton):
|
||||
|
||||
_host: str = None
|
||||
_port: int = None
|
||||
_username: str = None
|
||||
_passowrd: str = None
|
||||
|
||||
def __init__(self):
|
||||
host = settings.QB_HOST
|
||||
if host and host.find(":") != -1:
|
||||
self._host = settings.QB_HOST.split(":")[0]
|
||||
self._port = settings.QB_HOST.split(":")[1]
|
||||
self._username = settings.QB_USER
|
||||
self._password = settings.QB_PASSWORD
|
||||
if self._host and self._port and self._username and self._password:
|
||||
self.qbc = self.__login_qbittorrent()
|
||||
|
||||
def __login_qbittorrent(self) -> Optional[Client]:
|
||||
"""
|
||||
连接qbittorrent
|
||||
:return: qbittorrent对象
|
||||
"""
|
||||
try:
|
||||
# 登录
|
||||
qbt = qbittorrentapi.Client(host=self._host,
|
||||
port=self._port,
|
||||
username=self._username,
|
||||
password=self._password,
|
||||
VERIFY_WEBUI_CERTIFICATE=False,
|
||||
REQUESTS_ARGS={'timeout': (15, 60)})
|
||||
try:
|
||||
qbt.auth_log_in()
|
||||
except qbittorrentapi.LoginFailed as e:
|
||||
print(str(e))
|
||||
return qbt
|
||||
except Exception as err:
|
||||
logger.error(f"qbittorrent 连接出错：{err}")
|
||||
return None
|
||||
|
||||
def get_torrents(self, ids: Union[str, list] = None,
|
||||
status: Union[str, list] = None, tag: Union[str, list] = None) -> Tuple[list, bool]:
|
||||
"""
|
||||
获取种子列表
|
||||
return: 种子列表, 是否发生异常
|
||||
"""
|
||||
if not self.qbc:
|
||||
return [], True
|
||||
try:
|
||||
torrents = self.qbc.torrents_info(torrent_hashes=ids,
|
||||
status_filter=status)
|
||||
if tag:
|
||||
results = []
|
||||
if not isinstance(tag, list):
|
||||
tag = [tag]
|
||||
for torrent in torrents:
|
||||
include_flag = True
|
||||
for t in tag:
|
||||
if t and t not in torrent.get("tags"):
|
||||
include_flag = False
|
||||
break
|
||||
if include_flag:
|
||||
results.append(torrent)
|
||||
return results or [], False
|
||||
return torrents or [], False
|
||||
except Exception as err:
|
||||
logger.error(f"获取种子列表出错：{err}")
|
||||
return [], True
|
||||
|
||||
def get_completed_torrents(self, ids: Union[str, list] = None,
|
||||
tag: Union[str, list] = None) -> Optional[list]:
|
||||
"""
|
||||
获取已完成的种子
|
||||
return: 种子列表, 如发生异常则返回None
|
||||
"""
|
||||
if not self.qbc:
|
||||
return None
|
||||
torrents, error = self.get_torrents(status=["completed"], ids=ids, tag=tag)
|
||||
return None if error else torrents or []
|
||||
|
||||
def get_downloading_torrents(self, ids: Union[str, list] = None,
|
||||
tag: Union[str, list] = None) -> Optional[list]:
|
||||
"""
|
||||
获取正在下载的种子
|
||||
return: 种子列表, 如发生异常则返回None
|
||||
"""
|
||||
if not self.qbc:
|
||||
return None
|
||||
torrents, error = self.get_torrents(ids=ids,
|
||||
status=["downloading"],
|
||||
tag=tag)
|
||||
return None if error else torrents or []
|
||||
|
||||
def remove_torrents_tag(self, ids: Union[str, list], tag: Union[str, list]) -> bool:
|
||||
"""
|
||||
移除种子Tag
|
||||
:param ids: 种子Hash列表
|
||||
:param tag: 标签内容
|
||||
"""
|
||||
try:
|
||||
self.qbc.torrents_delete_tags(torrent_hashes=ids, tags=tag)
|
||||
return True
|
||||
except Exception as err:
|
||||
logger.error(f"移除种子Tag出错：{err}")
|
||||
return False
|
||||
|
||||
def set_torrents_status(self, ids: Union[str, list]):
|
||||
"""
|
||||
设置种子状态为已整理，以及是否强制做种
|
||||
"""
|
||||
if not self.qbc:
|
||||
return
|
||||
try:
|
||||
# 打标签
|
||||
self.qbc.torrents_add_tags(tags="已整理", torrent_hashes=ids)
|
||||
except Exception as err:
|
||||
logger.error(f"设置种子Tag出错：{err}")
|
||||
|
||||
def torrents_set_force_start(self, ids: Union[str, list]):
|
||||
"""
|
||||
设置强制作种
|
||||
"""
|
||||
try:
|
||||
self.qbc.torrents_set_force_start(enable=True, torrent_hashes=ids)
|
||||
except Exception as err:
|
||||
logger.error(f"设置强制作种出错：{err}")
|
||||
|
||||
def get_transfer_task(self, tag: Union[str, list] = None) -> Optional[list]:
|
||||
"""
|
||||
获取下载文件转移任务种子
|
||||
"""
|
||||
# 处理下载完成的任务
|
||||
torrents = self.get_completed_torrents() or []
|
||||
trans_tasks = []
|
||||
for torrent in torrents:
|
||||
torrent_tags = torrent.get("tags") or ""
|
||||
# 含"已整理"tag的不处理
|
||||
if "已整理" in torrent_tags:
|
||||
continue
|
||||
# 开启标签隔离，未包含指定标签的不处理
|
||||
if tag and tag not in torrent_tags:
|
||||
logger.debug(f"{torrent.get('name')} 未包含指定标签：{tag}")
|
||||
continue
|
||||
path = torrent.get("save_path")
|
||||
# 无法获取下载路径的不处理
|
||||
if not path:
|
||||
logger.warn(f"未获取到 {torrent.get('name')} 下载保存路径")
|
||||
continue
|
||||
content_path = torrent.get("content_path")
|
||||
if content_path:
|
||||
trans_name = content_path.replace(path, "").replace("\\", "/")
|
||||
if trans_name.startswith('/'):
|
||||
trans_name = trans_name[1:]
|
||||
else:
|
||||
trans_name = torrent.get('name')
|
||||
trans_tasks.append({
|
||||
'path': Path(settings.DOWNLOAD_PATH) / trans_name,
|
||||
'id': torrent.get('hash')
|
||||
})
|
||||
return trans_tasks
|
||||
|
||||
def __get_last_add_torrentid_by_tag(self, tag: Union[str, list],
|
||||
status: Union[str, list] = None) -> Optional[str]:
|
||||
"""
|
||||
根据种子的下载链接获取下载中或暂停的钟子的ID
|
||||
:return: 种子ID
|
||||
"""
|
||||
try:
|
||||
torrents, _ = self.get_torrents(status=status, tag=tag)
|
||||
except Exception as err:
|
||||
logger.error(f"获取种子列表出错：{err}")
|
||||
return None
|
||||
if torrents:
|
||||
return torrents[0].get("hash")
|
||||
else:
|
||||
return None
|
||||
|
||||
def get_torrent_id_by_tag(self, tag: Union[str, list],
|
||||
status: Union[str, list] = None) -> Optional[str]:
|
||||
"""
|
||||
通过标签多次尝试获取刚添加的种子ID，并移除标签
|
||||
"""
|
||||
torrent_id = None
|
||||
# QB添加下载后需要时间，重试5次每次等待5秒
|
||||
for i in range(1, 6):
|
||||
time.sleep(5)
|
||||
torrent_id = self.__get_last_add_torrentid_by_tag(tag=tag,
|
||||
status=status)
|
||||
if torrent_id is None:
|
||||
continue
|
||||
else:
|
||||
self.remove_torrents_tag(torrent_id, tag)
|
||||
break
|
||||
return torrent_id
|
||||
|
||||
def add_torrent(self,
|
||||
content: Union[str, bytes],
|
||||
is_paused: bool = False,
|
||||
download_dir: str = None,
|
||||
tag: Union[str, list] = None,
|
||||
cookie=None
|
||||
) -> bool:
|
||||
"""
|
||||
添加种子
|
||||
:param content: 种子urls或文件内容
|
||||
:param is_paused: 添加后暂停
|
||||
:param tag: 标签
|
||||
:param download_dir: 下载路径
|
||||
:param cookie: 站点Cookie用于辅助下载种子
|
||||
:return: bool
|
||||
"""
|
||||
if not self.qbc or not content:
|
||||
return False
|
||||
|
||||
if isinstance(content, str):
|
||||
urls = content
|
||||
torrent_files = None
|
||||
else:
|
||||
urls = None
|
||||
torrent_files = content
|
||||
|
||||
if download_dir:
|
||||
save_path = download_dir
|
||||
is_auto = False
|
||||
else:
|
||||
save_path = None
|
||||
is_auto = None
|
||||
|
||||
if tag:
|
||||
tags = tag
|
||||
else:
|
||||
tags = None
|
||||
|
||||
try:
|
||||
|
||||
# 添加下载
|
||||
qbc_ret = self.qbc.torrents_add(urls=urls,
|
||||
torrent_files=torrent_files,
|
||||
save_path=save_path,
|
||||
is_paused=is_paused,
|
||||
tags=tags,
|
||||
use_auto_torrent_management=is_auto,
|
||||
cookie=cookie)
|
||||
return True if qbc_ret and str(qbc_ret).find("Ok") != -1 else False
|
||||
except Exception as err:
|
||||
logger.error(f"添加种子出错：{err}")
|
||||
return False
|
||||
|
||||
def start_torrents(self, ids: Union[str, list]) -> bool:
|
||||
"""
|
||||
启动种子
|
||||
"""
|
||||
if not self.qbc:
|
||||
return False
|
||||
try:
|
||||
self.qbc.torrents_resume(torrent_hashes=ids)
|
||||
return True
|
||||
except Exception as err:
|
||||
logger.error(f"启动种子出错：{err}")
|
||||
return False
|
||||
|
||||
def stop_torrents(self, ids: Union[str, list]) -> bool:
|
||||
"""
|
||||
暂停种子
|
||||
"""
|
||||
if not self.qbc:
|
||||
return False
|
||||
try:
|
||||
self.qbc.torrents_pause(torrent_hashes=ids)
|
||||
return True
|
||||
except Exception as err:
|
||||
logger.error(f"暂停种子出错：{err}")
|
||||
return False
|
||||
|
||||
def delete_torrents(self, delete_file: bool, ids: Union[str, list]) -> bool:
|
||||
"""
|
||||
删除种子
|
||||
"""
|
||||
if not self.qbc:
|
||||
return False
|
||||
if not ids:
|
||||
return False
|
||||
try:
|
||||
self.qbc.torrents_delete(delete_files=delete_file, torrent_hashes=ids)
|
||||
return True
|
||||
except Exception as err:
|
||||
logger.error(f"删除种子出错：{err}")
|
||||
return False
|
||||
|
||||
def get_files(self, tid: str) -> Optional[TorrentFilesList]:
|
||||
"""
|
||||
获取种子文件清单
|
||||
"""
|
||||
try:
|
||||
return self.qbc.torrents_files(torrent_hash=tid)
|
||||
except Exception as err:
|
||||
logger.error(f"获取种子文件列表出错：{err}")
|
||||
return None
|
||||
|
||||
def set_files(self, **kwargs) -> bool:
|
||||
"""
|
||||
设置下载文件的状态，priority为0为不下载，priority为1为下载
|
||||
"""
|
||||
if not kwargs.get("torrent_hash") or not kwargs.get("file_ids"):
|
||||
return False
|
||||
try:
|
||||
self.qbc.torrents_file_priority(torrent_hash=kwargs.get("torrent_hash"),
|
||||
file_ids=kwargs.get("file_ids"),
|
||||
priority=kwargs.get("priority"))
|
||||
return True
|
||||
except Exception as err:
|
||||
logger.error(f"设置种子文件状态出错：{err}")
|
||||
return False
|
||||
Reference in New Issue
Block a user