import re from typing import List, Optional, Union from app.chain import ChainBase from app.core.config import settings from app.core.context import MediaInfo from app.core.meta import MetaBase from app.core.metainfo import MetaInfo from app.db.downloadhistory_oper import DownloadHistoryOper from app.db.models.downloadhistory import DownloadHistory from app.db.transferhistory_oper import TransferHistoryOper from app.helper.progress import ProgressHelper from app.log import logger from app.schemas import TransferInfo, TransferTorrent from app.schemas.types import TorrentStatus, EventType, MediaType, ProgressKey from app.utils.string import StringUtils class TransferChain(ChainBase): """ 文件转移处理链 """ def __init__(self): super().__init__() self.downloadhis = DownloadHistoryOper() self.transferhis = TransferHistoryOper() self.progress = ProgressHelper() def process(self, arg_str: str = None, userid: Union[str, int] = None) -> bool: """ 获取下载器中的种子列表，并执行转移 """ def extract_hash_and_number(string: str): """ 从字符串中提取种子hash和编号 """ pattern = r'([a-fA-F0-9]+) (\d+)' match = re.search(pattern, string) if match: hash_value = match.group(1) number = match.group(2) return hash_value, int(number) else: return None, None if arg_str: logger.info(f"开始转移下载器文件，参数：{arg_str}") # 解析中种子hash，TMDB ID torrent_hash, tmdbid = extract_hash_and_number(arg_str) if not hash or not tmdbid: logger.error(f"参数错误，参数：{arg_str}") return False # 获取种子 torrents: Optional[List[TransferTorrent]] = self.list_torrents(hashs=torrent_hash) if not torrents: logger.error(f"没有获取到种子，参数：{arg_str}") return False # 查询媒体信息 arg_mediainfo = self.recognize_media(tmdbid=tmdbid) else: arg_mediainfo = None logger.info("开始执行下载器文件转移 ...") # 从下载器获取种子列表 torrents: Optional[List[TransferTorrent]] = self.list_torrents(status=TorrentStatus.TRANSFER) if not torrents: logger.info("没有获取到已完成的下载任务") return False logger.info(f"获取到 {len(torrents)} 个已完成的下载任务") # 开始进度 self.progress.start(ProgressKey.FileTransfer) # 总数 total_num = len(torrents) # 已处理数量 processed_num = 0 self.progress.update(value=0, text=f"开始转移下载任务文件，共 {total_num} 个任务 ...", key=ProgressKey.FileTransfer) for torrent in torrents: # 更新进度 self.progress.update(value=processed_num / total_num * 100, text=f"正在转移 {torrent.title} ...", key=ProgressKey.FileTransfer) # 识别前预处理 result: Optional[tuple] = self.prepare_recognize(title=torrent.title) if result: title, subtitle = result else: title, subtitle = torrent.title, None # 识别元数据 meta: MetaBase = MetaInfo(title=title, subtitle=subtitle) if not meta.name: logger.warn(f'未识别到元数据，标题：{title}') continue if not arg_mediainfo: # 查询下载记录识别情况 downloadhis: DownloadHistory = self.downloadhis.get_by_hash(torrent.hash) if downloadhis: mtype = MediaType.TV if downloadhis.type == MediaType.TV.value else MediaType.MOVIE mediainfo = self.recognize_media(mtype=mtype, tmdbid=downloadhis.tmdbid) else: # 使用标题识别媒体信息 mediainfo: MediaInfo = self.recognize_media(meta=meta) if not mediainfo: logger.warn(f'未识别到媒体信息，标题：{torrent.title}') self.post_message(title=f"{torrent.title} 未识别到媒体信息，无法入库！\n" f"回复：```\n/transfer {torrent.hash} [tmdbid]\n``` 手动识别转移。", userid=userid) continue else: mediainfo = arg_mediainfo logger.info(f"{torrent.title} 识别为：{mediainfo.type.value} {mediainfo.title_year}") # 更新媒体图片 self.obtain_image(mediainfo=mediainfo) # 转移 transferinfo: TransferInfo = self.transfer(mediainfo=mediainfo, path=torrent.path) if not transferinfo or not transferinfo.target_path: # 转移失败 logger.warn(f"{torrent.title} 入库失败") self.post_message( title=f"{mediainfo.title_year}{meta.season_episode} 入库失败！", text=f"原因：{transferinfo.message if transferinfo else '未知'}", image=mediainfo.get_message_image(), userid=userid ), continue # 新增转移历史记录 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 ) # 转移完成 self.transfer_completed(hashs=torrent.hash, transinfo=transferinfo) # 刮剥 self.scrape_metadata(path=transferinfo.target_path, mediainfo=mediainfo) # 刷新媒体库 self.refresh_mediaserver(mediainfo=mediainfo, file_path=transferinfo.target_path) # 发送通知 self.__send_transfer_message(meta=meta, mediainfo=mediainfo, transferinfo=transferinfo) # 广播事件 self.eventmanager.send_event(EventType.TransferComplete, { 'meta': meta, 'mediainfo': mediainfo, 'transferinfo': transferinfo }) # 计数 processed_num += 1 # 更新进度 self.progress.update(value=processed_num / total_num * 100, text=f"{torrent.title} 转移完成", key=ProgressKey.FileTransfer) # 结束进度 self.progress.end(ProgressKey.FileTransfer) logger.info("下载器文件转移执行完成") return True def __send_transfer_message(self, meta: MetaBase, mediainfo: MediaInfo, transferinfo: TransferInfo): """ 发送入库成功的消息 """ msg_title = f"{mediainfo.title_year} 已入库" 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.resource_term: msg_str = f"{msg_str}，质量：{meta.resource_term}" msg_str = f"{msg_str}，共{transferinfo.file_count}个文件，" \ f"大小：{StringUtils.str_filesize(transferinfo.total_size)}" if transferinfo.message: msg_str = f"{msg_str}，以下文件处理失败：\n{transferinfo.message}" # 发送 self.post_message(title=msg_title, text=msg_str, image=mediainfo.get_message_image())