This commit is contained in:
2026-02-12 17:44:37 +08:00
parent 4b98e94942
commit be25807b6e

View File

@@ -109,12 +109,12 @@ def _build_output_url(abs_output_path: str) -> Optional[str]:
@mcp.tool() @mcp.tool()
async def list_docx_images(docx_path: str) -> List[Dict[str, Any]]: async def list_docx_images(docx_url: str) -> List[Dict[str, Any]]:
""" """
列出指定 DOCX 文件中的所有图片信息。 列出指定 DOCX 文件中的所有图片信息。
参数: 参数:
- docx_path: DOCX 文件的路径(相对或绝对),也可以是 HTTP/HTTPS URL。 - docx_url: 文件的HTTP/HTTPS URL。
返回: 返回:
- 图片信息列表,每一项包含: - 图片信息列表,每一项包含:
@@ -124,38 +124,11 @@ async def list_docx_images(docx_path: str) -> List[Dict[str, Any]]:
- docpr_name: Word 内部的图片名称 - docpr_name: Word 内部的图片名称
- width_cm / height_cm: 近似尺寸(厘米),可能为 None - width_cm / height_cm: 近似尺寸(厘米),可能为 None
""" """
tmp_file: Optional[str] = None imgs = get_images_info(_download_to_temp(docx_url, suffix=".docx"))
try: return imgs
local_path = docx_path
if _is_url(docx_path):
parsed = urllib.parse.urlparse(docx_path)
ext = os.path.splitext(parsed.path)[1] or ".docx"
tmp_file = _download_to_temp(docx_path, suffix=ext)
local_path = tmp_file
if not os.path.exists(local_path):
raise FileNotFoundError(f"DOCX 文件不存在: {docx_path}")
imgs = get_images_info(local_path)
# 为了避免泄露容器内部路径,屏蔽 abs_path 字段
for img in imgs:
img.pop("abs_path", None)
return imgs
finally:
if tmp_file and os.path.exists(tmp_file):
try:
os.remove(tmp_file)
except OSError:
pass
@mcp.tool() @mcp.tool()
async def edit_docx( async def edit_docx(input_docx_url: str, output_docx_url: str, replacements: Optional[List[Dict[str, str]]] = None, image_replacements: Optional[List[Dict[str, Any]]] = None) -> Dict[str, Any]:
input_docx: str,
output_docx: str,
replacements: Optional[List[Dict[str, str]]] = None,
image_replacements: Optional[List[Dict[str, Any]]] = None,
) -> Dict[str, Any]:
""" """
使用原始 mcp_docx 逻辑对 DOCX 文件进行编辑。 使用原始 mcp_docx 逻辑对 DOCX 文件进行编辑。
@@ -165,8 +138,8 @@ async def edit_docx(
- 替换指定序号的图片 - 替换指定序号的图片
参数: 参数:
- input_docx: 输入 DOCX 文件路径,或 HTTP/HTTPS URL - input_docx_url: 输入 DOCX 文件路径,或 HTTP/HTTPS URL
- output_docx: 输出 DOCX 文件路径 - output_docx_url: 输出 DOCX 文件路径
- replacements: 文本替换规则列表,例如: - replacements: 文本替换规则列表,例如:
[ [
{\"old\": \"旧标题\", \"new\": \"<span color='#FF0000'>新标题</span>\"}, {\"old\": \"旧标题\", \"new\": \"<span color='#FF0000'>新标题</span>\"},
@@ -189,15 +162,15 @@ async def edit_docx(
tmp_images: List[str] = [] tmp_images: List[str] = []
try: try:
local_input = input_docx local_input = input_docx_url
if _is_url(input_docx): if _is_url(input_docx_url):
parsed = urllib.parse.urlparse(input_docx) parsed = urllib.parse.urlparse(input_docx_url)
ext = os.path.splitext(parsed.path)[1] or ".docx" ext = os.path.splitext(parsed.path)[1] or ".docx"
tmp_input = _download_to_temp(input_docx, suffix=ext) tmp_input = _download_to_temp(input_docx_url, suffix=ext)
local_input = tmp_input local_input = tmp_input
if not os.path.exists(local_input): if not os.path.exists(local_input):
raise FileNotFoundError(f"输入 DOCX 文件不存在: {input_docx}") raise FileNotFoundError(f"输入 DOCX 文件不存在: {input_docx_url}")
if replacements is None: if replacements is None:
replacements = [] replacements = []
@@ -247,13 +220,13 @@ async def edit_docx(
# 复用原始处理函数 # 复用原始处理函数
process( process(
input_docx=local_input, input_docx=local_input,
output_docx=output_docx, output_docx=output_docx_url,
replacements=rep_pairs, replacements=rep_pairs,
image_replacements=img_pairs, image_replacements=img_pairs,
color_keywords=color_keywords, color_keywords=color_keywords,
) )
abs_out = os.path.abspath(output_docx) abs_out = os.path.abspath(output_docx_url)
return { return {
"output_path": abs_out, "output_path": abs_out,
"output_url": _build_output_url(abs_out), "output_url": _build_output_url(abs_out),