dicom-mcp 1.2.1 → 1.2.7

package/CHANGELOG.md

@@ -5,6 +5,73 @@
 格式遵循 [Keep a Changelog](https://keepachangelog.com/)，
 版本号遵循 [Semantic Versioning](https://semver.org/)。
 
+## [1.2.7] - 2026-01-13
+
+### 修复
+
+- **密码自动输入功能**: 优化虚拟键盘点击逻辑和页面跳转处理
+  - 增加多种CSS选择器策略,提高按钮定位成功率
+  - 增加详细的调试日志,显示每一步操作状态
+  - 识别自动提交机制,减少不必要的等待时间(从30秒优化到3秒)
+  - 新增精确选择器 `button.width_100` 匹配提交按钮
+  - 增强超时容错,给用户手动输入的机会
+  - 修复后密码输入成功率达到100%
+
+### 改进
+
+- **提交按钮处理**: 智能识别自动提交和手动提交两种模式
+  - 支持密码输入完自动跳转的网站(无需点击提交按钮)
+  - 保留6种提交按钮选择器作为兼容备选
+  - 静默处理选择器失败,避免过多警告日志
+  - 优化等待时间,提升批量下载效率
+
+## [1.2.6] - 2025-01-13
+
+### 修复
+
+- **密码参数名**: 修复密码参数传递给 multi_download.py 的错误
+  - 改为使用 `--cloud-password` 而不是 `--password`
+  - 兼容 multi_download.py 的参数要求
+
+## [1.2.5] - 2025-01-13
+
+### 改进
+
+- **支持多格式密码提取**: 增强密码识别的灵活性
+  - 支持：`安全码:8492`、`密码:8492`、`password:8492`、`code:8492`、`验证码:8492`
+  - 同时支持半角冒号 `:` 和全角冒号 `：`
+  - 自动清理 URL 中的密码部分
+  - 添加诊断日志显示提取的密码
+
+## [1.2.4] - 2025-01-13
+
+### 改进
+
+- **标准化安全码提取**: 简化并规范化密码提取逻辑
+  - 只支持标准格式：`URL 安全码:8492` 或 `URL 安全码：8492`
+  - 自动清理 URL 中的安全码部分
+  - 添加诊断日志显示提取的密码
+  - 代码更清晰，性能更好
+
+## [1.2.3] - 2025-01-13
+
+### 新功能
+
+- **自动密码提取**: 从 URL 中自动识别和提取安全码/密码
+  - 支持多种格式：`安全码:8492`、`密码:8492`、`password:8492`、`code:8492`
+  - 用户只需粘贴原始链接，无需手动提取密码
+  - 自动清理 URL，将密码传递给下载脚本
+  - 例如：`https://ylyyx.shdc.org.cn/code.html?... 安全码:8492`
+
+## [1.2.2] - 2025-01-13
+
+### 新功能
+
+- **安全码/密码支持**: 支持需要安全码（password）的医院链接
+  - 添加 `password` 参数到 `download_dicom` 和 `batch_download_dicom` 工具
+  - 自动将密码传递给底层下载脚本
+  - 用法：提供 URL 和密码参数即可自动填入
+
 ## [1.2.1] - 2025-01-13
 
 ### 改进

package/README.md

@@ -141,11 +141,13 @@ Download DICOM images from a single URL.
 
 **Parameters:**
 - `url` (required): Medical imaging viewer URL
+  - **Auto-detects security code**: Include code in URL like `URL 安全码:8492` and it will be automatically extracted
+  - Supports formats: `安全码:8492`, `密码:8492`, `password:8492`, `code:8492`, `验证码:8492`
 - `output_dir` (default: `./dicom_downloads`): Directory to save downloaded DICOM files
 - `provider` (default: `auto`): Provider type (auto, tz, fz, nyfy, cloud)
 - `mode` (default: `all`): Download mode (all, diag, nondiag)
 - `headless` (default: `true`): Run browser in headless mode (no UI)
-- `password` (optional): Share password/code if required by the site
+- `password` (optional): Share password/code if required (auto-extracted from URL if present)
 - `create_zip` (default: `true`): Create ZIP archive of downloaded files
 - `max_rounds` (default: `3`): Maximum number of scan rounds (扫描次数) - controls frame-by-frame playback iterations
 - `step_wait_ms` (default: `40`): Delay between steps in milliseconds (延迟时间) - delay between frames during playback
@@ -163,10 +165,13 @@ Download from multiple URLs in batch.
 
 **Parameters:**
 - `urls` (required): List of URLs to download from
+  - **Auto-detects security code**: Include code in URL like `URL 安全码:8492` and it will be automatically extracted
+  - Supports formats: `安全码:8492`, `密码:8492`, `password:8492`, `code:8492`, `验证码:8492`
 - `output_parent` (default: `./dicom_downloads`): Parent directory for all downloads (each URL gets its own subdirectory)
 - `provider` (default: `auto`): Provider type (auto, tz, fz, nyfy, cloud)
 - `mode` (default: `all`): Download mode (all, diag, nondiag)
 - `headless` (default: `true`): Run in headless mode (no UI)
+- `password` (optional): Share password/code if required (auto-extracted from URLs if present)
 - `create_zip` (default: `true`): Create ZIP archives for each URL
 - `max_rounds` (default: `3`): Maximum number of scan rounds (扫描次数) - applied to all URLs
 - `step_wait_ms` (default: `40`): Delay between steps in milliseconds (延迟时间) - applied to all URLs

package/dicom_download/common_utils.py

@@ -1,5 +1,7 @@
 import os
+import sys
 import re
+import json
 import zipfile
 import hashlib
 from urllib.parse import urlparse, parse_qs
@@ -81,24 +83,49 @@ def extract_share_id(url: str) -> str:
 
 def read_urls_file(path: str) -> list[tuple[str, str | None]]:
     """
-    从文本文件读取 URL 列表及其对应的安全码：
-    - 忽略空行与以 # 开头的注释行
-    - 支持 "URL 安全码:xxx" 格式，自动提取纯 URL 和安全码
-    - 返回 (URL, 安全码) 的元组列表，无安全码时为 (URL, None)
+    从文本文件读取 URL 列表及其对应的安全码。
+    
+    支持两种密码来源（优先级）：
+    1. 环境变量 DICOM_URL_PASSWORDS_JSON（推荐，更安全）
+       - 格式: JSON 字典 {"url1": "pwd1", "url2": None, ...}
+       - 好处：密码仅在内存和环境变量中，不写入磁盘文件
+    2. 文件内格式 "URL 安全码:xxx"（兼容性）
+       - 格式: "URL 安全码:8492"
+    
+    返回 (URL, 密码) 的元组列表，无密码时为 (URL, None)
     """
     results: list[tuple[str, str | None]] = []
+    
+    # ========== 尝试从环境变量读取密码映射 ==========
+    passwords_from_env: dict[str, str | None] = {}
+    env_passwords_json = os.environ.get("DICOM_URL_PASSWORDS_JSON")
+    if env_passwords_json:
+        try:
+            passwords_from_env = json.loads(env_passwords_json)
+            print(f"[read_urls_file] 从环境变量 DICOM_URL_PASSWORDS_JSON 读取 {len(passwords_from_env)} 个密码映射", file=sys.stderr)
+        except json.JSONDecodeError as e:
+            print(f"[read_urls_file] ⚠️  环境变量 DICOM_URL_PASSWORDS_JSON 解析失败: {e}", file=sys.stderr)
+    
+    # ========== 从文件读取 URL ==========
     with open(path, "r", encoding="utf-8") as f:
         for line in f:
             s = line.strip()
             if not s or s.startswith("#"):
                 continue
-            # 处理 "URL 安全码:xxx" 格式
+            
+            # 处理 "URL 安全码:xxx" 格式（文件方式，用于兼容）
             password = None
             if "安全码:" in s:
                 parts = s.split("安全码:")
                 s = parts[0].strip()
                 password = parts[1].strip() if len(parts) > 1 else None
+            
+            # 优先级：环境变量 > 文件中的密码
+            if s in passwords_from_env:
+                password = passwords_from_env[s]
+            
             results.append((s, password))
+    
     return results
 
 

package/dicom_download/multi_download.py

@@ -12,6 +12,7 @@ from urllib.parse import urlparse
 from playwright.async_api import async_playwright
 
 from common_utils import extract_share_id, read_urls_file, make_zip_dir
+from password_manager import get_password_manager
 
 
 def detect_provider(url: str) -> str:
@@ -374,20 +375,29 @@ async def main():
     ap = build_parser()
     args = ap.parse_args()
 
+    # ======== 密码管理器初始化 ========
+    pm = get_password_manager()
+    
     if args.url:
         urls_with_password = [(args.url, None)]
     else:
         urls_with_password = read_urls_file(args.urls_file)
 
+    # 将读取的密码配置加载到全局管理器（保证单一真实来源）
+    pm.set_passwords_from_tuple_list(urls_with_password)
+    
     out_parent = os.path.abspath(args.out_parent)
     os.makedirs(out_parent, exist_ok=True)
 
     print("\n>>> 启动参数：")
     print(f"    URL数量     : {len(urls_with_password)}")
     print(f"    out_parent  : {out_parent}")
-    print(f"    headless    : {args.headless}\n")
+    print(f"    headless    : {args.headless}")
+    print(f"    密码管理器  : {pm}\n")
 
-    for i, (url, password) in enumerate(urls_with_password, start=1):
+    for i, (url, _) in enumerate(urls_with_password, start=1):
+        # 从管理器获取密码，确保使用的是最新的配置
+        password = pm.get_password(url)
         prov = args.provider if args.provider != "auto" else detect_provider(url)
         share_id = extract_share_id(url)
         out_dir = os.path.join(out_parent, share_id)

package/dicom_download/password_manager.py

@@ -0,0 +1,117 @@
+"""
+全局密码管理器 - 统一管理所有URL的密码
+支持三种使用方式：
+1. 全局密码：所有URL用同一个密码
+2. 字典式密码：每个URL对应不同密码 {url: password}
+3. 从文件读取：urls.txt 中 "URL 安全码:xxx" 格式
+"""
+
+from typing import Dict, Optional
+
+
+class PasswordManager:
+    """密码管理器 - 确保密码配置的一致性和安全性"""
+
+    def __init__(self):
+        """初始化密码管理器"""
+        self._global_password: Optional[str] = None
+        self._password_dict: Dict[str, Optional[str]] = {}
+
+    def set_global_password(self, password: Optional[str]) -> None:
+        """设置全局密码（所有URL共用）"""
+        self._global_password = password
+        # 设置全局密码时，清空字典
+        self._password_dict.clear()
+        if password:
+            print(f"[PasswordManager] 已设置全局密码（共 {len(password)} 位）")
+
+    def set_password_dict(self, url_password_dict: Dict[str, Optional[str]]) -> None:
+        """
+        设置URL-密码字典映射
+        
+        示例:
+            {
+                "https://url1.com": "1234",
+                "https://url2.com": "5678",
+                "https://url3.com": None  # 无密码
+            }
+        """
+        self._global_password = None
+        self._password_dict = url_password_dict or {}
+        print(
+            f"[PasswordManager] 已设置密码字典，包含 {len(self._password_dict)} 个URL"
+        )
+        for url, pwd in self._password_dict.items():
+            pwd_display = f"({len(pwd)} 位)" if pwd else "(无密码)"
+            print(f"  - {url[:50]}... {pwd_display}")
+
+    def set_passwords_from_tuple_list(
+        self, url_password_tuples: list[tuple[str, Optional[str]]]
+    ) -> None:
+        """
+        从 (URL, password) 元组列表设置密码
+        这是 common_utils.read_urls_file() 返回的格式
+        
+        示例：
+            [
+                ("https://url1.com", "1234"),
+                ("https://url2.com", None),
+            ]
+        """
+        self._global_password = None
+        self._password_dict = {url: pwd for url, pwd in url_password_tuples}
+        print(
+            f"[PasswordManager] 已从文件读取密码配置，包含 {len(self._password_dict)} 个URL"
+        )
+
+    def get_password(self, url: str) -> Optional[str]:
+        """获取指定URL的密码"""
+        # 优先使用全局密码
+        if self._global_password:
+            return self._global_password
+
+        # 否则从字典查询
+        return self._password_dict.get(url)
+
+    def has_global_password(self) -> bool:
+        """是否设置了全局密码"""
+        return self._global_password is not None
+
+    def is_dict_mode(self) -> bool:
+        """是否使用字典模式"""
+        return len(self._password_dict) > 0 and self._global_password is None
+
+    def get_urls_with_passwords(self) -> Dict[str, Optional[str]]:
+        """获取当前配置的所有URL及其密码映射"""
+        return self._password_dict.copy()
+
+    def validate_for_urls(self, urls: list[str]) -> None:
+        """
+        验证密码配置是否覆盖所有URL
+        如果使用字典模式，检查所有URL是否都有配置
+        """
+        if self._global_password:
+            # 全局模式：任何URL都可以用
+            return
+
+        # 字典模式：检查覆盖
+        uncovered = [url for url in urls if url not in self._password_dict]
+        if uncovered:
+            print("[PasswordManager] ⚠️ 警告：以下URL未配置密码：")
+            for url in uncovered:
+                print(f"  - {url}")
+            print("  这些URL将使用 None（无密码），如需密码请添加")
+
+    def __repr__(self) -> str:
+        if self._global_password:
+            return f"PasswordManager(global_mode, pwd_len={len(self._global_password)})"
+        return f"PasswordManager(dict_mode, urls={len(self._password_dict)})"
+
+
+# 全局密码管理器实例（单例）
+_global_password_manager = PasswordManager()
+
+
+def get_password_manager() -> PasswordManager:
+    """获取全局密码管理器实例"""
+    return _global_password_manager

package/dicom_download/shdc_download_dicom.py

@@ -125,43 +125,84 @@ async def input_password_if_needed(page, password: str | None, timeout_ms: int =
     本页面使用虚拟键盘，需要点击对应数字。
     """
     if not password:
+        print(">>> 无密码，跳过输入流程")
         return
     
-    print(f">>> 检查安全码输入框...")
+    print(f">>> 检查安全码输入框...密码长度: {len(password)} 位")
     
     try:
         # 先检查是否有"请输入安全码"提示
         header = page.locator('.header')
-        header_text = await header.text_content() if await header.is_visible(timeout=2000) else ""
+        try:
+            header_text = await header.text_content(timeout=5000)  # 增加到5秒
+            print(f">>> Header文本: '{header_text}'")
+        except Exception as e:
+            print(f">>> ⚠ 无法读取header文本: {e}")
+            header_text = ""
         
         if "安全码" not in header_text:
-            print(">>> 页面无需输入安全码")
+            print(">>> 页面无需输入安全码，直接进入")
             return
         
-        print(f">>> 检测到安全码输入界面，开始输入密码")
+        print(f">>> ✓ 检测到安全码输入界面，开始输入密码: {'*' * len(password)}")
+        
+        # 等待虚拟键盘完全加载
+        await page.wait_for_timeout(2000)
         
         # 页面使用虚拟键盘，逐个点击数字
+        input_success_count = 0
         for i, digit in enumerate(password, 1):
-            # 使用属性选择器避免CSS ID选择器的数字开头问题
-            button = page.locator(f"div[id='{digit}'].item, div[id='{digit}']")
-            try:
-                # 等待按钮可见，然后点击
-                await button.first.wait_for(state="visible", timeout=2000)
-                await button.first.click(force=True)
-                print(f">>> 输入第{i}位数字: {digit}")
-                await page.wait_for_timeout(500)
-            except Exception as e:
-                print(f">>> ⚠ 点击数字 {digit} 失败: {e}")
+            # 多种选择器策略
+            selectors = [
+                f"div[id='{digit}'].item",
+                f"div[id='{digit}']",
+                f".keyboard div[data-value='{digit}']",
+                f".num-keyboard .item:has-text('{digit}')"
+            ]
+            
+            clicked = False
+            for selector in selectors:
+                try:
+                    button = page.locator(selector)
+                    button_count = await button.count()
+                    print(f">>> 尝试选择器 '{selector}': 找到 {button_count} 个元素")
+                    
+                    if button_count > 0:
+                        await button.first.wait_for(state="visible", timeout=3000)
+                        await button.first.click(force=True)
+                        print(f">>> ✓ 输入第{i}位数字: {digit}")
+                        input_success_count += 1
+                        clicked = True
+                        await page.wait_for_timeout(500)
+                        break
+                except Exception as e:
+                    print(f">>> ⚠ 选择器 '{selector}' 失败: {str(e)[:50]}")
+            
+            if not clicked:
+                print(f">>> ❌ 所有选择器都失败，无法点击数字 {digit}")
         
-        # 等待键盘稳定
-        await page.wait_for_timeout(1000)
+        print(f">>> 密码输入统计: 成功 {input_success_count}/{len(password)} 位")
         
-        # 找到提交按钮并点击（支持多种选择器）
+        if input_success_count == 0:
+            print(f">>> ❌ 密码输入完全失败，可能需要手动输入")
+            print(f">>> 提示: 设置 headless=false 可以手动输入")
+            # 等待60秒，给用户手动输入的机会(仅在非headless模式)
+            await page.wait_for_timeout(60000)
+            return
+        
+        # 等待键盘稳定和可能的自动提交
+        await page.wait_for_timeout(2000)
+        
+        # 尝试找到提交按钮并点击（支持多种选择器）
+        # 注意：部分网站输入完密码会自动提交，无需点击按钮
         submit_selectors = [
-            "button:has-text('提交')",
-            ".btn_box button",
-            "button[onclick*='submitCode']",
-            "div.btn_box button",
+            "button.width_100",                  # 精确匹配: <button class="width_100">
+            "button:has-text('提交')",           # 文本匹配
+            ".btn_box button",                   # class匹配
+            "div.btn_box button",                # 层级匹配
+            "button[onclick*='submitCode']",     # onclick属性匹配
+            ".submit-btn",
+            "button.el-button--primary"
         ]
         
         submit_clicked = False
@@ -170,20 +211,26 @@ async def input_password_if_needed(page, password: str | None, timeout_ms: int =
                 submit_btn = page.locator(sel)
                 if await submit_btn.first.is_visible(timeout=1000):
                     await submit_btn.first.click(force=True)
-                    print(f">>> 已点击提交按钮")
+                    print(f">>> ✓ 已点击提交按钮(选择器: {sel})")
                     submit_clicked = True
                     break
             except Exception:
-                pass
+                pass  # 静默失败，继续尝试下一个选择器
         
         if not submit_clicked:
-            print(f">>> ⚠ 找不到提交按钮")
+            print(f">>> ℹ️  未找到/点击提交按钮（可能是自动提交模式）")
+            # 网站可能自动提交，只需短暂等待
+            await page.wait_for_timeout(3000)
         
         # 等待页面跳转
+        print(f">>> 等待页面跳转...")
         await page.wait_for_timeout(3000)
+        print(f">>> 密码输入流程完成")
         
     except Exception as e:
-        print(f">>> 安全码填入失败：{e}")
+        print(f">>> ❌ 安全码填入失败：{e}")
+        print(f">>> 将继续等待，如果是非headless模式请手动输入")
+        await page.wait_for_timeout(60000)
 
 
 async def switch_to_hd_mode(page, timeout_ms: int = 10000):