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

fix 下载文件记录只登记选中的文件

Browse Source
This commit is contained in:
jxxghp 2023-09-05 17:11:02 +08:00
parent 1a03981463
commit 36880a8b7d
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app/chain/download.py
View File

@ -8,6 +8,7 @@ from app.chain import ChainBase
from app.core.config import settings from app.core.config import settings
from app.core.context import MediaInfo, TorrentInfo, Context from app.core.context import MediaInfo, TorrentInfo, Context
from app.core.meta import MetaBase from app.core.meta import MetaBase
from app.core.metainfo import MetaInfo
from app.db.downloadhistory_oper import DownloadHistoryOper from app.db.downloadhistory_oper import DownloadHistoryOper
from app.db.mediaserver_oper import MediaServerOper from app.db.mediaserver_oper import MediaServerOper
from app.helper.torrent import TorrentHelper from app.helper.torrent import TorrentHelper
@ -164,6 +165,7 @@ class DownloadChain(ChainBase):
download_path = download_dir / _folder_name download_path = download_dir / _folder_name
else: else:
download_path = download_dir / _file_list[0] if _file_list else download_dir download_path = download_dir / _file_list[0] if _file_list else download_dir
# 登记下载记录 # 登记下载记录
self.downloadhis.add( self.downloadhis.add(
path=str(download_path), path=str(download_path),
@ -182,17 +184,27 @@ class DownloadChain(ChainBase):
torrent_description=_torrent.description, torrent_description=_torrent.description,
torrent_site=_torrent.site_name torrent_site=_torrent.site_name
) )
# 登记下载文件 # 登记下载文件
self.downloadhis.add_files([ files_to_add = []
{ for file in _file_list:
if episodes:
# 识别文件集
file_meta = MetaInfo(Path(file).stem)
if not file_meta.begin_episode \
or file_meta.begin_episode not in episodes:
continue
files_to_add.append({
"download_hash": _hash, "download_hash": _hash,
"downloader": settings.DOWNLOADER, "downloader": settings.DOWNLOADER,
"fullpath": str(download_dir / _folder_name / file), "fullpath": str(download_dir / _folder_name / file),
"savepath": str(download_dir / _folder_name), "savepath": str(download_dir / _folder_name),
"filepath": file, "filepath": file,
"torrentname": _meta.org_string, "torrentname": _meta.org_string,
} for file in _file_list if file })
]) if files_to_add:
self.downloadhis.add_files(files_to_add)
# 发送消息 # 发送消息
self.post_download_message(meta=_meta, mediainfo=_media, torrent=_torrent, channel=channel) self.post_download_message(meta=_meta, mediainfo=_media, torrent=_torrent, channel=channel)
# 下载成功后处理 # 下载成功后处理