MoviePilot/app/agent/tools/impl/query_transfer_history.py

"""查询整理历史记录工具"""

import json
from typing import Optional, Type

from pydantic import BaseModel, Field

from app.agent.tools.base import MoviePilotTool
from app.agent.tools.tags import ToolTag
from app.db.transferhistory_oper import TransferHistoryOper
from app.log import logger
from app.schemas.types import media_type_to_agent
from app.utils.jieba import cut as jieba_cut


class QueryTransferHistoryInput(BaseModel):
    """查询整理历史记录工具的输入参数模型"""
    title: Optional[str] = Field(None, description="Search by title (optional, supports partial match)")
    status: Optional[str] = Field("all",
                                  description="Filter by status: 'success' for successful transfers, 'failed' for failed transfers, 'all' for all records (default: 'all')")
    page: Optional[int] = Field(1, description="Page number for pagination (default: 1, each page contains 30 records)")


class QueryTransferHistoryTool(MoviePilotTool):
    """查询影视与音乐整理历史及可复用的媒体身份。"""

    name: str = "query_transfer_history"
    tags: list[str] = [
        ToolTag.Read,
        ToolTag.Transfer,
    ]
    description: str = (
        "Query movie, TV, and music transfer history. Returns source/destination paths, status, errors, and "
        "stable media_source/media_id values that can be reused for retries. Music album retries should group "
        "tracks by the shared album directory/identity instead of treating every track as unrelated media."
    )
    args_schema: Type[BaseModel] = QueryTransferHistoryInput

    def get_tool_message(self, **kwargs) -> Optional[str]:
        """根据查询参数生成友好的提示消息"""
        title = kwargs.get("title")
        status = kwargs.get("status", "all")
        page = kwargs.get("page", 1)

        parts = ["查询整理历史"]

        if title:
            parts.append(f"标题: {title}")
        if status != "all":
            status_map = {"success": "成功", "failed": "失败"}
            parts.append(f"状态: {status_map.get(status, status)}")
        if page > 1:
            parts.append(f"第{page}页")

        return " | ".join(parts) if len(parts) > 1 else parts[0]

    async def run(self, title: Optional[str] = None,
                  status: Optional[str] = "all",
                  page: Optional[int] = 1, **kwargs) -> str:
        """执行整理历史筛选并返回有界分页结果。"""
        logger.info(f"执行工具: {self.name}, 参数: title={title}, status={status}, page={page}")

        try:
            # 处理状态参数
            status_bool = None
            if status == "success":
                status_bool = True
            elif status == "failed":
                status_bool = False

            # 处理页码参数
            if page is None or page < 1:
                page = 1

            # 每页固定 30 条，与工具说明保持一致，避免整理路径等字段撑大上下文。
            count = 30

            transferhis = TransferHistoryOper()
            # 处理标题搜索
            if title:
                # 使用统一分词封装处理标题，便于替换底层实现。
                words = jieba_cut(title, HMM=False)
                title_search = "%".join(words)
                # 查询记录
                result = await transferhis.async_list_by_title(
                    title=title_search, page=page, count=count, status=status_bool
                )
                total = await transferhis.async_count_by_title(
                    title=title_search, status=status_bool
                )
            else:
                # 查询所有记录
                result = await transferhis.async_list_by_page(
                    page=page, count=count, status=status_bool
                )
                total = await transferhis.async_count(status=status_bool)

            if not result:
                return "未找到相关整理历史记录"

            # 转换为字典格式，只保留关键信息
            simplified_records = []
            for record in result:
                simplified = {
                    "id": record.id,
                    "title": record.title,
                    "year": record.year,
                    "type": media_type_to_agent(record.type),
                    "category": record.category,
                    "seasons": record.seasons,
                    "episodes": record.episodes,
                    "src": record.src,
                    "dest": record.dest,
                    "mode": record.mode,
                    "status": "成功" if record.status else "失败",
                    "date": record.date,
                    "downloader": record.downloader,
                    "download_hash": record.download_hash
                }
                # 如果失败，添加错误信息
                if not record.status and record.errmsg:
                    simplified["errmsg"] = record.errmsg
                # 添加媒体ID信息（如果有）
                if record.tmdbid:
                    simplified["tmdbid"] = record.tmdbid
                if record.imdbid:
                    simplified["imdbid"] = record.imdbid
                if record.doubanid:
                    simplified["doubanid"] = record.doubanid
                if record.bangumiid:
                    simplified["bangumiid"] = record.bangumiid
                if record.anilistid:
                    simplified["anilistid"] = record.anilistid
                if record.media_source:
                    simplified["media_source"] = record.media_source
                if record.media_id:
                    simplified["media_id"] = record.media_id
                if getattr(record, "music_type", None):
                    simplified["music_type"] = record.music_type
                if getattr(record, "total_tracks", None):
                    simplified["total_tracks"] = record.total_tracks
                simplified_records.append(simplified)

            result_json = json.dumps(simplified_records, ensure_ascii=False, indent=2)

            # 计算总页数
            total_pages = (total + count - 1) // count if total > 0 else 1

            # 构建分页信息
            pagination_info = f"第 {page}/{total_pages} 页，共 {total} 条记录（每页 {count} 条）"

            return f"{pagination_info}\n\n{result_json}"
        except Exception as e:
            logger.error(f"查询整理历史记录失败: {e}", exc_info=True)
            return f"查询整理历史记录时发生错误: {str(e)}"