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

add 失败历史记录

Browse Source
This commit is contained in:
jxxghp 2023-06-26 12:30:06 +08:00
parent 178ac334a0
commit 66f081de98
3 changed files with 45 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
app/chain/transfer.py
View File

@ -95,7 +95,7 @@ class TransferChain(ChainBase):
# 识别元数据
meta: MetaBase = MetaInfo(title=title, subtitle=subtitle)
if not meta.name:
logger.warn(f'未识别到元数据，标题：{title}')
logger.error(f'未识别到元数据，标题：{title}')
continue
if not arg_mediainfo:
# 查询下载记录识别情况
@ -111,6 +111,16 @@ class TransferChain(ChainBase):
self.post_message(title=f"{torrent.title} 未识别到媒体信息，无法入库！\n"
f"回复：```\n/transfer {torrent.hash} [tmdbid]\n``` 手动识别转移。",
userid=userid)
# 新增转移失败历史记录
self.transferhis.add(
src=str(torrent.path),
mode=settings.TRANSFER_TYPE,
seasons=meta.season,
episodes=meta.episode,
download_hash=torrent.hash,
status=0,
errmsg="未识别到媒体信息"
)
continue
else:
mediainfo = arg_mediainfo
@ -127,9 +137,29 @@ class TransferChain(ChainBase):
text=f"原因：{transferinfo.message if transferinfo else '未知'}",
image=mediainfo.get_message_image(),
userid=userid
),
)
# 新增转移失败历史记录
self.transferhis.add(
src=str(torrent.path),
dest=str(transferinfo.target_path),
mode=settings.TRANSFER_TYPE,
type=mediainfo.type.value,
category=mediainfo.category,
title=mediainfo.title,
year=mediainfo.year,
tmdbid=mediainfo.tmdb_id,
imdbid=mediainfo.imdb_id,
tvdbid=mediainfo.tvdb_id,
doubanid=mediainfo.douban_id,
seasons=meta.season,
episodes=meta.episode,
image=mediainfo.get_poster_image(),
download_hash=torrent.hash,
status=0,
errmsg=transferinfo.message
)
continue
# 新增转移历史记录
# 新增转移成功历史记录
self.transferhis.add(
src=str(torrent.path),
dest=str(transferinfo.target_path),
@ -145,7 +175,8 @@ class TransferChain(ChainBase):
seasons=meta.season,
episodes=meta.episode,
image=mediainfo.get_poster_image(),
download_hash=torrent.hash
download_hash=torrent.hash,
status=1
)
# 转移完成
self.transfer_completed(hashs=torrent.hash, transinfo=transferinfo)

6
app/db/models/transferhistory.py
View File

@ -36,7 +36,7 @@ class TransferHistory(Base):
# 海报
image = Column(String)
# 下载器hash
download_hash = Column(String)
download_hash = Column(String, index=True)
# 转移成功状态
status = Column(Boolean(), default=True)
# 转移失败信息
@ -52,3 +52,7 @@ class TransferHistory(Base):
@staticmethod
def list_by_page(db: Session, page: int = 1, count: int = 30):
return db.query(TransferHistory).offset((page - 1) * count).limit(count).all()
@staticmethod
def get_by_hash(db: Session, download_hash: str):
return db.query(TransferHistory).filter(TransferHistory.download_hash == download_hash).first()

7
app/db/transferhistory_oper.py
View File

@ -20,5 +20,8 @@ class TransferHistoryOper(DbOper):
"""
新增转移历史
"""
transferhistory = TransferHistory(**kwargs)
return transferhistory.create(self._db)
if kwargs.get("download_hash"):
transferhistory = TransferHistory.get_by_hash(self._db, kwargs.get("download_hash"))
if transferhistory:
transferhistory.delete(self._db, transferhistory.id)
return TransferHistory(**kwargs).create(self._db)