lightpdf-aipdf-mcp 0.1.52__py3-none-any.whl → 0.1.53__py3-none-any.whl

lightpdf_aipdf_mcp/editor.py

@@ -163,13 +163,12 @@ class Editor(BaseApiClient):
                     original_name=original_name
                 )
 
-    async def rotate_pdf(self, file_path: str, angle: int, pages: str = "", password: Optional[str] = None, original_name: Optional[str] = None) -> EditResult:
+    async def rotate_pdf(self, file_path: str, angle_params: Dict[str, str], password: Optional[str] = None, original_name: Optional[str] = None) -> EditResult:
         """旋转PDF文件的页面
         
         Args:
             file_path: 要旋转的PDF文件路径
-            angle: 旋转角度，可选值为90、180、270
-            pages: 指定要旋转的页面范围，例如 "1,3,5-7" 或 "" 表示所有页面
+            angle_params: 旋转角度和页面范围的映射，如 {"90": "1-3", "180": "4-6"} 
             password: 文档密码，如果文档受密码保护，则需要提供（可选）
             original_name: 原始文件名（可选）
             
@@ -181,19 +180,20 @@ class Editor(BaseApiClient):
             return EditResult(success=False, file_path=file_path, error_message="非PDF文件", original_name=original_name)
         
         # 验证旋转角度
-        valid_angles = {90, 180, 270}
-        if angle not in valid_angles:
-            await self.logger.error("无效的旋转角度。角度必须是: 90, 180, 270")
-            return EditResult(success=False, file_path=file_path, error_message="无效的旋转角度。角度必须是: 90, 180, 270", original_name=original_name)
+        valid_angles = {"90", "180", "270"}
+        for angle in angle_params.keys():
+            if angle not in valid_angles:
+                await self.logger.error("无效的旋转角度。角度必须是: 90, 180, 270")
+                return EditResult(success=False, file_path=file_path, error_message="无效的旋转角度。角度必须是: 90, 180, 270", original_name=original_name)
         
         # 构建API参数
         extra_params = {
-            "angle": json.dumps({str(angle): pages})
+            "angle": json.dumps(angle_params)
         }
         
         # 记录操作描述
-        angle_desc = f"{angle}°" + (f": {pages}" if pages else ": 所有页面")
-        await self._log_operation("旋转PDF文件", f"参数: {angle_desc}")
+        angle_details = ", ".join([f"{angle}°: {pages}" for angle, pages in angle_params.items()])
+        await self._log_operation("旋转PDF文件", f"参数: {angle_details}")
         
         # 调用edit_pdf方法处理API请求
         return await self.edit_pdf(file_path, EditType.ROTATE, extra_params, password, original_name)

lightpdf_aipdf_mcp/server.py

@@ -195,10 +195,29 @@ async def process_edit_file(
             original_name=original_name
         )
     elif edit_type == "rotate":
+        # 从extra_params获取旋转参数列表
+        rotation_arguments = extra_params.get("arguments", [])
+        
+        # 验证旋转参数列表
+        if not rotation_arguments:
+            return EditResult(
+                success=False, 
+                file_path=file_path, 
+                error_message="旋转操作需要至少提供一个旋转参数",
+                original_name=original_name
+            )
+        
+        # 构建angle_params字典: {"90": "2-4,6-8", "180": "all"}
+        angle_params = {}
+        for arg in rotation_arguments:
+            angle = str(arg.get("angle", 90))
+            pages = arg.get("pages", "all") or "all"  # 确保空字符串转为"all"
+            angle_params[angle] = pages
+        
+        # 直接调用rotate_pdf方法，传入角度参数字典
         return await editor.rotate_pdf(
-            file_path=file_path, 
-            angle=extra_params.get("angle", 90),
-            pages=extra_params.get("pages", ""),
+            file_path=file_path,
+            angle_params=angle_params,
             password=password,
             original_name=original_name
         )
@@ -718,18 +737,29 @@ async def handle_list_tools() -> list[types.Tool]:
                         },
                         "description": "需要旋转的PDF文件列表，每个文件包含路径和可选的密码"
                     },
-                    "angle": {
-                        "type": "integer",
-                        "description": "旋转角度，可选值为90、180、270",
-                        "enum": [90, 180, 270],
-                        "default": 90
-                    },
-                    "pages": {
-                        "type": "string",
-                        "description": "指定要旋转的页面范围，例如 '1,3,5-7' 或 ''（空字符串或不设置）表示所有页面"
+                    "arguments": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "angle": {
+                                    "type": "integer",
+                                    "description": "旋转角度，可选值为90、180、270",
+                                    "enum": [90, 180, 270],
+                                    "default": 90
+                                },
+                                "pages": {
+                                    "type": "string",
+                                    "description": "指定要旋转的页面范围，例如 '1,3,5-7' 或 'all' 表示所有页面",
+                                    "default": "all"
+                                }
+                            },
+                            "required": ["angle", "pages"]
+                        },
+                        "description": "参数列表，每个参数包含旋转角度和页面范围"
                     }
                 },
-                "required": ["files", "angle"]
+                "required": ["files", "arguments"]
             }
         )
     ]
@@ -792,7 +822,7 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
         "rotate_pdf": {
             "edit_type": "rotate",  # 编辑类型
             "is_edit_operation": True,  # 标记为编辑操作
-            "param_keys": ["angle", "pages"]  # 需要从arguments获取的参数
+            "param_keys": ["arguments"]  # 只需要arguments参数，移除对旧格式的支持
         }
     }
     
@@ -808,7 +838,7 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
         "split_type": "page",
         "merge_all": 1,
         "angle": 90,
-        "pages": ""
+        "pages": "all"  # 更新默认值为"all"
     }
     
     if name in TOOL_CONFIG:

{lightpdf_aipdf_mcp-0.1.52.dist-info → lightpdf_aipdf_mcp-0.1.53.dist-info}/METADATA

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

lightpdf_aipdf_mcp-0.1.53.dist-info/RECORD

@@ -0,0 +1,9 @@
+lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
+lightpdf_aipdf_mcp/common.py,sha256=-7LU6gm-As_F8Ly68ssy15Vc9Zt_eNSnvDLEtVZDwlI,6633
+lightpdf_aipdf_mcp/converter.py,sha256=QBwKcCg83FrGGDCy3IFyXjldIZg2hzej_zCH8nyOPak,14768
+lightpdf_aipdf_mcp/editor.py,sha256=RuHxTMlyvUp2keh0Plue2g4vwGe4Vzv5mauJuncyM8o,23551
+lightpdf_aipdf_mcp/server.py,sha256=9EQ6sodIjKC8fZMpVQbKN1FRttebIQ9o9LCH0_IPtyo,41465
+lightpdf_aipdf_mcp-0.1.53.dist-info/METADATA,sha256=tgBnVSIcOx5eLRV27B5gIlnqb3ysWuKaHJwepEGCy6o,7906
+lightpdf_aipdf_mcp-0.1.53.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.53.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.53.dist-info/RECORD,,

lightpdf_aipdf_mcp-0.1.52.dist-info/RECORD

@@ -1,9 +0,0 @@
-lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
-lightpdf_aipdf_mcp/common.py,sha256=-7LU6gm-As_F8Ly68ssy15Vc9Zt_eNSnvDLEtVZDwlI,6633
-lightpdf_aipdf_mcp/converter.py,sha256=QBwKcCg83FrGGDCy3IFyXjldIZg2hzej_zCH8nyOPak,14768
-lightpdf_aipdf_mcp/editor.py,sha256=PVsgpgqc89MTf7-D2iRoescXJW90Kv51mP1oXGeneaY,23534
-lightpdf_aipdf_mcp/server.py,sha256=25UY58YF3vIypWjUx3GdEP0PHxA2hvZMiblT_A4ispg,40063
-lightpdf_aipdf_mcp-0.1.52.dist-info/METADATA,sha256=tdV-B3SqBY78N7czBjSTRi0sJQkNiXwF6a0unYDUrnI,7906
-lightpdf_aipdf_mcp-0.1.52.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.52.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.52.dist-info/RECORD,,