mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-19 19:46:55 +08:00
ea646149c0
- Introduced DeleteDownloadTool, QueryDirectoriesTool, ListDirectoryTool, QueryTransferHistoryTool, and TransferFileTool to the toolset for improved download management. - Updated __all__ exports in init.py and factory.py to include the new tools. - Enhanced QueryDownloadsTool to support querying downloads by hash and title, providing more flexible search options and detailed results.
117 lines
5.2 KiB
Python
117 lines
5.2 KiB
Python
"""查询整理历史记录工具"""
|
||
|
||
import json
|
||
from typing import Optional, Type
|
||
|
||
import jieba
|
||
from pydantic import BaseModel, Field
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.agent.tools.base import MoviePilotTool
|
||
from app.db import AsyncSessionFactory
|
||
from app.db.models.transferhistory import TransferHistory
|
||
from app.log import logger
|
||
|
||
|
||
class QueryTransferHistoryInput(BaseModel):
|
||
"""查询整理历史记录工具的输入参数模型"""
|
||
explanation: str = Field(..., description="Clear explanation of why this tool is being used in the current context")
|
||
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"
|
||
description: str = "Query file transfer history records. Shows transfer status, source and destination paths, media information, and transfer details. Supports filtering by title and status."
|
||
args_schema: Type[BaseModel] = QueryTransferHistoryInput
|
||
|
||
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
|
||
|
||
# 每页记录数
|
||
count = 30
|
||
|
||
# 获取数据库会话
|
||
async with AsyncSessionFactory() as db:
|
||
# 处理标题搜索
|
||
if title:
|
||
# 使用 jieba 分词处理标题
|
||
words = jieba.cut(title, HMM=False)
|
||
title_search = "%".join(words)
|
||
# 查询记录
|
||
result = await TransferHistory.async_list_by_title(
|
||
db, title=title_search, page=page, count=count, status=status_bool
|
||
)
|
||
total = await TransferHistory.async_count_by_title(
|
||
db, title=title_search, status=status_bool
|
||
)
|
||
else:
|
||
# 查询所有记录
|
||
result = await TransferHistory.async_list_by_page(
|
||
db, page=page, count=count, status=status_bool
|
||
)
|
||
total = await TransferHistory.async_count(db, status=status_bool)
|
||
|
||
if not result:
|
||
return "未找到相关整理历史记录"
|
||
|
||
# 转换为字典格式，只保留关键信息
|
||
simplified_records = []
|
||
for record in result:
|
||
simplified = {
|
||
"id": record.id,
|
||
"title": record.title,
|
||
"year": record.year,
|
||
"type": 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
|
||
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)}"
|
||
|