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

add transfer notification

Browse Source
This commit is contained in:
jxxghp
2023-06-09 15:52:53 +08:00
parent 017fce9e6f
commit 31fe552af4
6 changed files with 102 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
app/chain/transfer.py
View File

@ -1,10 +1,12 @@
from pathlib import Path
from typing import List, Optional
from app.chain import ChainBase
from app.core import MetaInfo, MediaInfo, settings
from app.core.meta import MetaBase
from app.log import logger
from app.utils.types import TorrentStatus
from app.utils.string import StringUtils
from app.utils.system import SystemUtils
from app.utils.types import TorrentStatus, MediaType
class TransferChain(ChainBase):
@ -26,7 +28,7 @@ class TransferChain(ChainBase):
# 识别
for torrent in torrents:
# 识别元数据
meta = MetaInfo(torrent.get("title"))
meta: MetaBase = MetaInfo(torrent.get("title"))
if not meta.get_name():
logger.warn(f'未识别到元数据，标题：{torrent.get("title")}')
continue
@ -34,24 +36,70 @@ class TransferChain(ChainBase):
mediainfo: MediaInfo = self.recognize_media(meta=meta)
if not mediainfo:
logger.warn(f'未识别到媒体信息，标题：{torrent.get("title")}')
self.post_message(title=f"{torrent.get('title')} 未识别到媒体信息，无法入库！")
continue
logger.info(f"{torrent.get('title')} 识别为：{mediainfo.type.value} {mediainfo.get_title_string()}")
# 更新媒体图片
self.obtain_image(mediainfo=mediainfo)
# 转移
dest_path: Path = self.transfer(mediainfo=mediainfo, path=torrent.get("path"))
if not dest_path:
logger.warn(f"{torrent.get('title')} 转移失败")
transferinfo: dict = self.transfer(mediainfo=mediainfo, path=torrent.get("path"))
if not transferinfo or not transferinfo.get("target_path"):
logger.warn(f"{torrent.get('title')} 入库失败")
self.post_message(
title=f"{mediainfo.get_title_string()}{meta.get_season_episode_string()} 入库失败！",
text=f"原因：{transferinfo.get('message') if transferinfo else '未知'}\n"
f"路径：{torrent.get('path')}",
image=mediainfo.get_message_image()
),
continue
# 转移完成
self.transfer_completed(hashs=torrent.get("hash"))
# 刮剥
self.scrape_metadata(path=dest_path, mediainfo=mediainfo)
self.scrape_metadata(path=transferinfo.get('target_path'), mediainfo=mediainfo)
# 移动模式删除种子
if settings.TRANSFER_TYPE == "move":
result = self.remove_torrents(hashs=torrent.get("hash"))
if result:
logger.info(f"移动模式删除种子成功：{torrent.get('title')} ")
# 发送通知
self.__send_transfer_message(meta=meta, mediainfo=mediainfo, transferinfo=transferinfo)
logger.info("下载器文件转移执行完成")
return True
def __send_transfer_message(self, meta: MetaBase, mediainfo: MediaInfo, transferinfo: dict):
"""
发送入库成功的消息
"""
# 文件大小
file_size = StringUtils.str_filesize(
SystemUtils.get_directory_size(
transferinfo.get('target_path')
)
)
if mediainfo.type == MediaType.TV:
# 电影
msg_title = f"{mediainfo.get_title_string()} {meta.get_season_episode_string()} 已入库"
if mediainfo.vote_average:
msg_str = f"评分：{mediainfo.vote_average}，类型：{mediainfo.type.value}"
else:
msg_str = f"类型：{mediainfo.type.value}"
if mediainfo.category:
msg_str = f"{msg_str}，类别：{mediainfo.category}"
if meta.total_seasons <= 1 and meta.total_episodes == 1:
msg_str = f"{msg_str}，大小：{file_size}"
else:
msg_str = f"{msg_str}，共{meta.total_seasons}季{meta.total_episodes}集，总大小：{file_size}"
else:
msg_title = f"{mediainfo.get_title_string()} 已入库"
if mediainfo.vote_average:
msg_str = f"评分：{mediainfo.vote_average}，类型：电影"
else:
msg_str = "类型：电影"
if mediainfo.category:
msg_str = f"{msg_str}，类别：{mediainfo.category}"
if meta.get_resource_type_string():
msg_str = f"{msg_str}，质量：{meta.get_resource_type_string()}"
msg_str = f"{msg_str}， 大小：{file_size}"
# 发送
self.post_message(title=msg_title, text=msg_str, image=mediainfo.get_message_image())