ascript 1.3.27__py3-none-win_amd64.whl

ascript/windows/tools/colors_tool.py

@@ -0,0 +1,391 @@
+import sys
+import eel
+
+from ..system import R
+from . import eel_api, config,plug_manager
+from ..window import Window
+from ..screen import Screen
+import  time
+import ctypes
+import threading
+import bottle
+import socket
+
+from ..ui.web_window import WebWindow
+
+
+_colors_web_window = None
+_screenshot_lock = threading.Lock()
+
+
+@eel.expose
+def colors_tool_screenshot(hwnd):
+    """截取屏幕或指定窗口的截图，返回 base64 编码的图像数据。
+
+    如果提供了窗口句柄，则截取该窗口；否则截取全屏。
+    图色助手窗口由 Windows WDA_EXCLUDEFROMCAPTURE 排除，不需要最小化、
+    隐藏或重新激活窗口，因此不会打断 WebView2 的焦点和合成状态。
+
+    Args:
+        hwnd: 目标窗口句柄。为 None 或空值时截取全屏。
+
+    Returns:
+        dict: 包含以下字段的字典：
+            - status (str): "success" 或 "error"
+            - data (str): base64 编码的图像数据
+            - save_info (dict): 保存信息，含 status 和 path
+            - message (str): 错误信息（如有）
+    """
+    if not _screenshot_lock.acquire(blocking=False):
+        return {
+            "status": "error",
+            "data": "",
+            "save_info": {"status": "error", "path": None},
+            "message": "已有截图任务正在执行，请稍后重试",
+        }
+
+    started_at = time.perf_counter()
+    my_file_temp = os.path.join(
+        config.file_capture_dir, str(int(time.time() * 1000)) + ".jpeg"
+    )
+    try:
+        print(
+            f"[colors] screenshot start hwnd={hwnd!r} path={my_file_temp}",
+            flush=True,
+        )
+        if hwnd:
+            result = eel_api.get_screenshot(
+                hwnd, my_file_temp, include_data=False
+            )
+        else:
+            image = Screen.capture(save_path=my_file_temp, format="pillow")
+            if image is None:
+                result = {
+                    "status": "error",
+                    "data": "",
+                    "save_info": {"status": "error", "path": my_file_temp},
+                    "message": "全屏截图失败",
+                }
+            else:
+                result = {
+                    "status": "success",
+                    "data": "",
+                    "save_info": {"status": "success", "path": my_file_temp},
+                    "message": "",
+                }
+        print(
+            f"[colors] screenshot done status={result.get('status')} "
+            f"elapsed={time.perf_counter() - started_at:.3f}s",
+            flush=True,
+        )
+        return result
+    except Exception as error:
+        print(f"[colors] screenshot error: {error!r}", flush=True)
+        return {
+            "status": "error",
+            "data": "",
+            "save_info": {"status": "error", "path": my_file_temp},
+            "message": f"截图失败: {error}",
+        }
+    finally:
+        _screenshot_lock.release()
+
+
+import os
+import base64
+import io
+from PIL import Image
+
+
+@eel.expose
+def ascript_get_image(file_path: str, max_side_size: int = 0):
+    """读取指定路径的图片，可选缩放后以 base64 格式返回。
+
+    支持相对路径和绝对路径。相对路径会自动拼接截图目录。
+    图片统一转为 RGB 模式输出 JPEG 格式。
+
+    Args:
+        file_path: 图片完整路径或文件名。若为相对路径，则从截图目录读取。
+        max_side_size: 缩放限制（像素）。0 表示不缩放；大于 0 则将长边限制在该像素内。
+
+    Returns:
+        dict: 包含以下字段的字典：
+            - status (str): "success" 或 "error"
+            - message (str): 操作结果描述
+            - data (str): base64 编码的图像数据（data URI 格式），失败时为空字符串
+    """
+    try:
+        # 1. 路径兼容性处理
+        if not os.path.isabs(file_path):
+            file_path = os.path.join(config.file_capture_dir, os.path.basename(file_path))
+
+        # 2. 检查文件是否存在
+        if not os.path.exists(file_path):
+            return {"status": "error", "message": f"找不到文件: {file_path}", "data": ""}
+
+        # 3. 处理图片
+        with Image.open(file_path) as img:
+            # 统一转为 RGB 模式以兼容 JPEG 格式
+            if img.mode != 'RGB':
+                img = img.convert('RGB')
+
+            img_byte_arr = io.BytesIO()
+
+            # 判断是否需要缩放
+            if max_side_size > 0:
+                # 使用 thumbnail 会保持宽高比，且仅在原图大于目标尺寸时缩小
+                img.thumbnail((max_side_size, max_side_size), Image.Resampling.LANCZOS)
+                # 缩略图可以使用较低的质量来节省空间
+                img.save(img_byte_arr, format='JPEG', quality=70, optimize=True)
+            else:
+                # 不缩放时，尽可能保持高质量输出
+                img.save(img_byte_arr, format='JPEG', quality=95)
+
+            img_data = img_byte_arr.getvalue()
+            img_base64 = base64.b64encode(img_data).decode()
+
+        return {
+            "status": "success",
+            "message": "读取成功" + ("并缩放" if max_side_size > 0 else ""),
+            "data": f"data:image/jpeg;base64,{img_base64}"
+        }
+
+    except Exception as e:
+        print(f"读取图片异常: {e}")
+        return {"status": "error", "message": str(e), "data": ""}
+
+
+
+
+
+
+
+@eel.expose
+def colors_tool_screenshot_files():
+    """获取截图目录下所有图片文件的绝对路径列表。
+
+    扫描截图目录中的图片文件（支持 png、jpg、jpeg、bmp、gif 格式），
+    按修改时间从新到旧排序返回。
+
+    Returns:
+        dict: 包含以下字段的字典：
+            - status (str): "success" 或 "error"
+            - message (str): 操作结果描述
+            - data (list): 图片文件绝对路径列表
+            - path (str): 截图目录路径（仅成功时包含）
+    """
+    try:
+        # 1. 获取配置中的目录路径
+        target_dir = config.file_capture_dir
+
+        # 2. 检查目录是否存在
+        if not os.path.exists(target_dir):
+            return {
+                "status": "success",
+                "message": "目录不存在",
+                "data": []
+            }
+
+        # 3. 定义支持的图片后缀名
+        valid_extensions = ('.png', '.jpg', '.jpeg', '.bmp', '.gif')
+
+        # 4. 获取目录下所有符合条件的文件，并转换为绝对路径
+        file_paths = []
+        for f in os.listdir(target_dir):
+            if f.lower().endswith(valid_extensions):
+                # 拼接成绝对路径
+                full_path = os.path.abspath(os.path.join(target_dir, f))
+                file_paths.append(full_path)
+
+        # 5. 排序：按文件修改时间从新到旧排列
+        # 注意：这里使用 os.path.getmtime 直接对绝对路径进行操作
+        file_paths.sort(key=lambda x: os.path.getmtime(x), reverse=True)
+
+        return {
+            "status": "success",
+            "message": f"成功获取 {len(file_paths)} 个文件路径",
+            "data": file_paths,  # 现在的 data 里面是完整的绝对路径列表
+            "path": target_dir
+        }
+
+    except Exception as e:
+        print(f"获取图片路径列表失败: {e}")
+        return {
+            "status": "error",
+            "message": str(e),
+            "data": []
+        }
+
+@eel.expose
+def save_dropped_image(base64_str, filename):
+    """将前端拖拽上传的 base64 图片保存到截图目录。
+
+    自动在文件名中添加时间戳以避免重名冲突。
+
+    Args:
+        base64_str: base64 编码的图片数据，支持 data URI 格式（如 "data:image/png;base64,..."）。
+        filename: 原始文件名，用于保留文件扩展名。
+
+    Returns:
+        dict: 包含以下字段的字典：
+            - status (str): "success" 或 "error"
+            - path (str): 保存后的文件绝对路径（仅成功时包含）
+            - message (str): 错误信息（仅失败时包含）
+    """
+    try:
+        # 提取 base64 数据部分
+        # 格式通常为: "data:image/png;base64,iVBOR..."
+        if "," in base64_str:
+            base64_str = base64_str.split(",")[1]
+
+        img_data = base64.b64decode(base64_str)
+
+        # 为了防止文件名重复，加上时间戳
+        name, ext = os.path.splitext(filename)
+        save_name = f"{name}_{int(time.time())}{ext}"
+        save_path = os.path.abspath(os.path.join(config.file_capture_dir, save_name))
+
+        with open(save_path, "wb") as f:
+            f.write(img_data)
+
+        return {"status": "success", "path": save_path}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+
+
+# 1. 定义一个触发函数
+def listen_win32_hotkey():
+    """监听 Win32 全局热键（Shift+S）并触发前端截图。
+
+    注册 Shift+S 全局热键，当热键被按下时调用前端 JS 函数
+    triggerScreenshotUI() 触发截图操作。该函数应在独立线程中运行。
+    """
+    user32 = ctypes.windll.user32
+
+    HOTKEY_ID = 101
+
+    # 修改这里：使用 MOD_SHIFT (0x0004)
+    MOD_SHIFT = 0x0004
+    # 如果你想用 Ctrl + Shift + S，就写：MOD_CTRL_SHIFT = 0x0002 | 0x0004
+
+    VK_S = 0x53  # 'S' 键的虚拟键码
+
+    # 注册热键
+    if not user32.RegisterHotKey(None, HOTKEY_ID, MOD_SHIFT, VK_S):
+        print("❌ 无法注册 Shift + S，可能被占用")
+        return
+
+    # print("✅ 成功注册热键: Shift + S")
+
+    try:
+        msg = ctypes.wintypes.MSG()
+        while user32.GetMessageW(ctypes.byref(msg), None, 0, 0) != 0:
+            if msg.message == 0x0312:  # WM_HOTKEY
+                # 触发 JS 截图逻辑
+                eel.triggerScreenshotUI()
+            user32.TranslateMessage(ctypes.byref(msg))
+            user32.DispatchMessageW(ctypes.byref(msg))
+    finally:
+        user32.UnregisterHotKey(None, HOTKEY_ID)
+
+
+# 启动线程
+from urllib.parse import unquote # 必须导入这个
+@eel.btl.route('/external/<base64_path>/<filename:path>')
+def server_external(base64_path, filename):
+    """提供外部文件的静态资源服务路由。
+
+    通过 base64 编码的物理路径定位文件根目录，支持中文路径和 URL 编码。
+    用于在 Eel Web 应用中访问本地文件系统中的图片等资源。
+
+    Args:
+        base64_path: base64 编码的物理目录路径（可能经过 URL 编码）。
+        filename: 相对于根目录的文件名路径。
+
+    Returns:
+        bottle.Response: 静态文件响应，或 500 错误响应。
+    """
+    try:
+        # 1. 先把 URL 里的 %2F 等符号转回正常字符
+        clean_base64 = unquote(base64_path)
+
+        # 2. Base64 解码并处理可能的中文编码
+        # 注意：这里增加 try-except 或直接解码
+        raw_path = base64.b64decode(clean_base64).decode('utf-8')
+
+        # 3. 再次 unquote 物理路径（因为 JS 端用了 encodeURIComponent）
+        actual_root = unquote(raw_path)
+
+        print(f"请求文件: {filename} | 物理根目录: {actual_root}")  # 调试用
+
+        return bottle.static_file(filename, root=actual_root)
+    except Exception as e:
+        print(f"路径解析失败: {e}")
+        return bottle.HTTPError(500, f"Path Error: {str(e)}")
+
+def start_colors(hwnd:str = None,window_title:str = None):
+    """启动图形处理（颜色工具）Web 应用。
+
+    初始化 Eel 框架并打开颜色工具页面，支持通过窗口句柄或窗口标题
+    指定目标窗口。自动检测可用浏览器（Chrome/Edge）。
+
+    Args:
+        hwnd: 目标窗口句柄字符串。优先级高于 window_title。
+        window_title: 目标窗口标题，用于自动查找窗口句柄。
+    """
+
+    if window_title:
+        temp_window = Window.find(window_title)
+        if temp_window:
+            hwnd = temp_window.hwnd
+
+    web_root = R.internal("windows", "tools", "web")
+    eel.init(web_root)
+
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as port_socket:
+        port_socket.bind(("127.0.0.1", 0))
+        port = port_socket.getsockname()[1]
+
+    server_error = []
+
+    def run_server():
+        try:
+            eel.start(
+                "colors.html",
+                mode=False,
+                host="127.0.0.1",
+                port=port,
+                block=True,
+                suppress_error=True,
+            )
+        except Exception as exc:
+            server_error.append(exc)
+
+    threading.Thread(target=run_server, daemon=True).start()
+
+    for _ in range(100):
+        if server_error:
+            raise RuntimeError(f"图色工具服务启动失败: {server_error[0]}")
+        try:
+            with socket.create_connection(("127.0.0.1", port), timeout=0.1):
+                break
+        except OSError:
+            time.sleep(0.05)
+    else:
+        raise RuntimeError("图色工具服务启动超时")
+
+    threading.Thread(target=listen_win32_hotkey, daemon=True).start()
+    global _colors_web_window
+    _colors_web_window = WebWindow(f"http://127.0.0.1:{port}/colors.html")
+    try:
+        return _colors_web_window.show(
+            title="图形处理 AScript",
+            width=1280,
+            height=820,
+            min_size=(980, 640),
+            background_color="#171a1c",
+            bring_to_front=True,
+            exclude_from_capture=True,
+        )
+    finally:
+        _colors_web_window = None

ascript/windows/tools/config.py

@@ -0,0 +1,26 @@
+"""工具模块配置文件。
+
+定义工具运行所需的临时目录路径（截图目录、临时数据目录），
+并在模块加载时自动创建所需的目录结构。
+
+模块属性:
+    file_root_path (str): 临时文件根目录，位于系统临时目录下的 ascript 子目录。
+    file_capture_dir (str): 截图保存目录。
+    file_data_temp (str): 临时数据目录。
+"""
+
+import os
+import tempfile
+
+# 1. 定义路径
+file_root_path = os.path.join(tempfile.gettempdir(), "ascript")
+file_capture_dir = os.path.join(file_root_path, "capture")
+file_data_temp = os.path.join(file_root_path, "temp")
+
+# 2. 判断并创建文件夹
+# 创建根目录及其子目录
+# os.makedirs 会自动创建中间路径（递归创建）
+os.makedirs(file_capture_dir, exist_ok=True)
+os.makedirs(file_data_temp, exist_ok=True)
+
+# print(f"路径已就绪: {file_capture_dir}")

ascript/windows/tools/dumptool.py

@@ -0,0 +1,18 @@
+from .tool_window import start_inspector
+
+def start_vtree(hwnd: str = None, window_title: str = None):
+    """启动 UI 元素树（VTree）查看器 Web 应用。
+
+    初始化 Eel 框架并打开 VTree 页面，用于查看和分析目标窗口的 UI 元素层级结构。
+    支持通过窗口句柄或窗口标题指定目标窗口。自动检测可用浏览器（Chrome/Edge）。
+
+    Args:
+        hwnd: 目标窗口句柄字符串。优先级高于 window_title。
+        window_title: 目标窗口标题，用于自动查找窗口句柄。
+    """
+    if window_title:
+        from ..window import Window
+        temp_window = Window.find(window_title)
+        if temp_window:
+            hwnd = temp_window.hwnd
+    return start_inspector(hwnd=hwnd)