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

fix tr

Browse Source
This commit is contained in:
jxxghp
2023-06-10 12:23:24 +08:00
parent e0730960d3
commit 396961ebad
2 changed files with 10 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,34 +47,25 @@ class TransmissionModule(_ModuleBase):
return None, f"添加种子任务失败：{torrent_path}" return None, f"添加种子任务失败：{torrent_path}"
else: else:
torrent_hash = torrent.hashString torrent_hash = torrent.hashString
torrent_id = torrent.id
if is_paused: if is_paused:
# 选择文件 # 选择文件
torrent_files = self.transmission.get_files(torrent_hash) torrent_files = self.transmission.get_files(torrent_hash)
if not torrent_files: if not torrent_files:
return torrent_hash, "获取种子文件失败，下载任务可能在暂停状态" return torrent_hash, "获取种子文件失败，下载任务可能在暂停状态"
# 需要的文件信息 # 需要的文件信息
files_info = {} file_ids = []
# 需要的集清单
sucess_epidised = []
for torrent_file in torrent_files: for torrent_file in torrent_files:
file_id = torrent_file.id file_id = torrent_file.id
file_name = torrent_file.name file_name = torrent_file.name
meta_info = MetaInfo(file_name) meta_info = MetaInfo(file_name)
if not meta_info.get_episode_list(): if not meta_info.get_episode_list():
selected = False continue
else: selected = set(meta_info.get_episode_list()).issubset(set(episodes))
selected = set(meta_info.get_episode_list()).issubset(set(episodes)) if not selected:
if selected: continue
sucess_epidised = list(set(sucess_epidised).union(set(meta_info.get_episode_list()))) file_ids.append(file_id)
if not files_info.get(torrent_id): # 选择文件
files_info[torrent_id] = {file_id: {'priority': 'normal', 'selected': selected}} self.transmission.set_files(torrent_hash, file_ids)
else:
files_info[torrent_id][file_id] = {'priority': 'normal', 'selected': selected}
if sucess_epidised and files_info:
self.transmission.set_files(file_info=files_info)
# 开始任务 # 开始任务
self.transmission.start_torrents(torrent_hash) self.transmission.start_torrents(torrent_hash)
else: else:

16
app/modules/transmission/transmission.py
View File

@ -240,24 +240,12 @@ class Transmission(metaclass=Singleton):
else: else:
return None return None
def set_files(self, **kwargs) -> bool: def set_files(self, tid: str, file_ids: list) -> bool:
""" """
设置下载文件的状态 设置下载文件的状态
{
<torrent id>: {
<file id>: {
'priority': <priority ('high'|'normal'|'low')>,
'selected': <selected for download (True|False)>
},
...
},
...
}
""" """
if not kwargs.get("file_info"):
return False
try: try:
self.trc.set_files(kwargs.get("file_info")) self.trc.change_torrent(ids=tid, files_wanted=file_ids)
return True return True
except Exception as err: except Exception as err:
logger.error(f"设置下载文件状态出错：{err}") logger.error(f"设置下载文件状态出错：{err}")