lightpdf-aipdf-mcp 0.1.84__py3-none-any.whl → 0.1.85__py3-none-any.whl

lightpdf_aipdf_mcp/server.py

@@ -265,11 +265,11 @@ async def process_tool_call(
         types.TextContent: 包含处理结果的文本内容
     """
     file_handler = FileHandler(logger)
+    editor = Editor(logger, file_handler)
     
     # 根据操作类型选择不同的处理逻辑
     if operation_config.get("is_edit_operation"):
         # 编辑操作
-        editor = Editor(logger, file_handler)
         edit_type = operation_config.get("edit_type", "")
         extra_params = operation_config.get("extra_params")
         
@@ -305,35 +305,67 @@ async def process_tool_call(
         converter = Converter(logger, file_handler)
         format = operation_config.get("format", "")
         extra_params = operation_config.get("extra_params")
-        
-        # 获取操作描述
-        if format == "doc-repair":
-            operation_desc = "去除水印"
-            task_type = "水印去除"
-        elif format == "number-pdf":
-            operation_desc = "添加页码"
-            task_type = "添加页码"
-        elif format == "flatten-pdf":
-            operation_desc = "展平PDF"
-            task_type = "展平PDF"
+
+        # 新增：特殊处理PDF转Markdown和TXT转PDF
+        if format == "md":
+            # PDF转Markdown，调用Editor.edit_pdf，传递oss://pdf2md
+            results = await process_batch_files(
+                file_objects,
+                logger,
+                lambda file_path, password, original_name: editor.edit_pdf(
+                    file_path,
+                    edit_type="edit",
+                    extra_params={"pages": [{"url": "oss://pdf2md", "oss_file": ""}]},
+                    password=password,
+                    original_name=original_name
+                ),
+                "PDF转Markdown"
+            )
+            report_msg = generate_result_report(results)
+        elif format == "pdf":
+            # 只调用一次process_batch_files，在lambda里分流
+            results = await process_batch_files(
+                file_objects,
+                logger,
+                lambda file_path, password, original_name: (
+                    editor.edit_pdf(
+                        file_path,
+                        edit_type="edit",
+                        extra_params={"pages": [{"url": "oss://txt2pdf", "oss_file": ""}]},
+                        password=password,
+                        original_name=original_name
+                    ) if file_handler.get_file_extension(file_path) == ".txt" else process_conversion_file(
+                        file_path, format, converter, extra_params, password, original_name
+                    )
+                ),
+                f"转换为 {format} 格式"
+            )
+            report_msg = generate_result_report(results)
         else:
-            operation_desc = f"转换为 {format} 格式"
-            task_type = "转换"
-        
-        # 处理文件
-        results = await process_batch_files(
-            file_objects,
-            logger,
-            lambda file_path, password, original_name: process_conversion_file(
-                file_path, format, converter, extra_params, password, original_name
-            ),
-            operation_desc
-        )
-        
-        # 生成报告
-        report_msg = generate_result_report(
-            results
-        )
+            # 获取操作描述
+            if format == "doc-repair":
+                operation_desc = "去除水印"
+            elif format == "number-pdf":
+                operation_desc = "添加页码"
+            elif format == "flatten-pdf":
+                operation_desc = "展平PDF"
+            else:
+                operation_desc = f"转换为 {format} 格式"
+            
+            # 处理文件
+            results = await process_batch_files(
+                file_objects,
+                logger,
+                lambda file_path, password, original_name: process_conversion_file(
+                    file_path, format, converter, extra_params, password, original_name
+                ),
+                operation_desc
+            )
+            
+            # 生成报告
+            report_msg = generate_result_report(
+                results
+            )
     
     # 如果全部失败，记录错误
     if not any(r.success for r in results):
@@ -381,8 +413,8 @@ async def handle_list_tools() -> list[types.Tool]:
                     },
                     "format": {
                         "type": "string",
-                        "description": "Target format",
-                        "enum": ["pdf", "docx", "xlsx", "pptx", "jpg", "jpeg", "png", "html", "txt", "csv"]
+                        "description": "Target format. PDF can be converted to DOCX, XLSX, PPTX, images (including long images), HTML, TXT (text extraction), CSV, or Markdown (md). Other formats (DOCX, XLSX, PPTX, images, TXT) can be converted to PDF.",
+                        "enum": ["pdf", "docx", "xlsx", "pptx", "jpg", "jpeg", "png", "html", "txt", "csv", "md"]
                     },
                     "merge_all": {
                         "type": "integer",
@@ -889,6 +921,38 @@ async def handle_list_tools() -> list[types.Tool]:
                 },
                 "required": ["files"]
             }
+        ),
+        types.Tool(
+            name="restrict_printing",
+            description="Restrict PDF printing permission. This tool will set the PDF so that printing is not allowed.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "files": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "path": {
+                                    "type": "string",
+                                    "description": "PDF file URL to restrict printing, must include protocol, supports http/https/oss"
+                                },
+                                "password": {
+                                    "type": "string",
+                                    "description": "PDF document password, required if the document is password-protected"
+                                },
+                                "name": {
+                                    "type": "string",
+                                    "description": "Original filename of the document"
+                                }
+                            },
+                            "required": ["path"]
+                        },
+                        "description": "List of PDF files to restrict printing, each containing path and optional password"
+                    }
+                },
+                "required": ["files"]
+            }
         )
     ]
 
@@ -958,7 +1022,12 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             "edit_type": "extract_image",  # 编辑类型
             "is_edit_operation": True,  # 标记为编辑操作
             "param_keys": ["format"]  # 需要从arguments获取的参数
-        }
+        },
+        "restrict_printing": {
+            "edit_type": "encrypt",  # 或protect，和protect_pdf一致
+            "is_edit_operation": True,
+            "param_keys": []  # 不暴露provider
+        },
     }
     
     DEFAULTS = {
@@ -1019,6 +1088,10 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             # 对于protect_pdf工具，需要处理新密码
             if name == "protect_pdf" and "password" in arguments:
                 operation_config["extra_params"]["password"] = arguments.get("password")
+            
+            # restrict_printing工具自动加provider参数
+            if name == "restrict_printing":
+                operation_config["extra_params"]["provider"] = "printpermission"
         
         # 特殊处理merge_pdfs工具
         if name == "merge_pdfs":
@@ -1063,10 +1136,10 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
         # 调用通用处理函数
         result = await process_tool_call(logger, file_objects, operation_config)
         return [result]
-    
-    error_msg = f"未知工具: {name}"
-    await logger.error(error_msg, ValueError)
-    return [types.TextContent(type="text", text=error_msg)]
+    else:
+        error_msg = f"未知工具: {name}"
+        await logger.error(error_msg, ValueError)
+        return [types.TextContent(type="text", text=error_msg)]
 
 async def main():
     """应用主入口"""

{lightpdf_aipdf_mcp-0.1.84.dist-info → lightpdf_aipdf_mcp-0.1.85.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightpdf-aipdf-mcp
-Version: 0.1.84
+Version: 0.1.85
 Summary: MCP Server for LightPDF AI-PDF
 Author: LightPDF Team
 License: Proprietary

{lightpdf_aipdf_mcp-0.1.84.dist-info → lightpdf_aipdf_mcp-0.1.85.dist-info}/RECORD

@@ -2,8 +2,8 @@ lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6
 lightpdf_aipdf_mcp/common.py,sha256=_UO1f6S9Qr_3k6u5iBpdVDpvTK5U-tHEpu9KsDGqV8Y,6635
 lightpdf_aipdf_mcp/converter.py,sha256=f9YuDtOmXBGlMmS3O4Xn3rdWljY9XcNxu0CjftH4s0o,14726
 lightpdf_aipdf_mcp/editor.py,sha256=O7wF_HWs5l-IiXLbZYLNYjj1ygo2v4yGJEYMJtn7jpo,26916
-lightpdf_aipdf_mcp/server.py,sha256=cWekqqhTfJNCyNA6UPgM-XSVhQl2QNireEhUH5uBBgQ,48454
-lightpdf_aipdf_mcp-0.1.84.dist-info/METADATA,sha256=8TyXJQottgizev-7CVOM8dolO8q2cqVTHEQ4u3Wx2ZM,8119
-lightpdf_aipdf_mcp-0.1.84.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.84.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.84.dist-info/RECORD,,
+lightpdf_aipdf_mcp/server.py,sha256=sHFc2c7gLM6qh5sqbZREynoT53QZDvoXKCNEzWfnC6o,52200
+lightpdf_aipdf_mcp-0.1.85.dist-info/METADATA,sha256=Ib-0va5128UPeUT7QLNekFkGmTnylT8GXk5j1cVlmZg,8119
+lightpdf_aipdf_mcp-0.1.85.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.85.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.85.dist-info/RECORD,,