开发指南¶

from easyths.core import BaseOperation
from easyths.models.operations import PluginMetadata, OperationResult

class MyOperation(BaseOperation):
    """我的操作"""

def _get_metadata(self) -> PluginMetadata:
    return PluginMetadata(
        name="MyOperation",              # 操作类名
        version="1.0.0",                  # 版本号
        description="操作描述",            # 描述
        author="你的名字",                 # 作者
        operation_name="my_operation",    # 操作名称（API 调用使用）
        parameters={                      # 参数定义
            "param1": {
                "type": "string",
                "required": True,
                "description": "参数说明"
            }
        }
    )

def validate(self, params: Dict[str, Any]) -> bool:
    """验证操作参数"""
    # 检查必需参数
    required_params = ["param1"]
    for param in required_params:
        if param not in params:
            self.logger.error(f"缺少必需参数: {param}")
            return False

    # 验证参数格式
    if not isinstance(params["param1"], str):
        self.logger.error("参数类型错误")
        return False

    return True

def execute(self, params: Dict[str, Any]) -> OperationResult:
    """执行操作"""
    try:
        # 你的操作逻辑
        result_data = {"key": "value"}

        return OperationResult(
            success=True,
            data=result_data
        )
    except Exception as e:
        return OperationResult(
            success=False,
            error=str(e)
        )

def pre_execute(self, params: Dict[str, Any]) -> bool:
    """执行前钩子"""
    # 调用基类方法：设置焦点、关闭弹窗
    return super().pre_execute(params)

# 获取指定条件的控件
control = self.get_control_with_children(
    parent_control,
    class_name="Edit",           # 控件类名
    control_type="Edit",         # 控件类型
    title="标题",                 # 控件标题
    title_re=r"^正则.*",         # 标题正则匹配
    auto_id="1032"               # 自动化 ID（重要）
)

# 获取主窗口
main_window = self.get_main_window(wrapper_obj=True)

# 获取面板
main_panel = main_window.children(control_type="Pane")[0]

# 获取编辑框并输入内容
edit = self.get_control_with_children(main_panel, control_type="Edit", auto_id="1032")
edit.type_keys("内容")

# 获取按钮并点击
button = self.get_control_with_children(main_panel, control_type="Button", auto_id="2")
button.click()

# 切换左侧菜单
self.switch_left_menus("查询[F4]", "资金股票")

# 获取主窗口
main_window = self.get_main_window(wrapper_obj=True)

# 关闭弹窗
self.close_pop_dialog()

# 等待弹窗
self.wait_for_pop_dialog(timeout=1.0)

# 睡眠
self.sleep(0.1)

# 检查弹窗是否存在
if self.is_exist_pop_dialog():
    # 处理弹窗
    pass

from typing import Dict, Any
from easyths.core import BaseOperation
from easyths.models.operations import PluginMetadata, OperationResult

class SimpleQueryOperation(BaseOperation):
    """简单查询操作"""

    def _get_metadata(self) -> PluginMetadata:
        return PluginMetadata(
            name="SimpleQueryOperation",
            version="1.0.0",
            description="简单查询示例",
            author="your_name",
            operation_name="simple_query",
            parameters={}
        )

    def validate(self, params: Dict[str, Any]) -> bool:
        return True

    def execute(self, params: Dict[str, Any]) -> OperationResult:
        # 切换菜单
        self.switch_left_menus("查询[F4]", "资金股票")

        # 获取数据
        main_window = self.get_main_window(wrapper_obj=True)
        main_panel = main_window.children(control_type="Pane")[0]
        text_controls = main_panel.children(control_type="Text")

        result_data = {}
        for control in text_controls:
            auto_id = control.element_info.automation_id
            if auto_id == "1016":
                result_data["可用金额"] = control.window_text()

        return OperationResult(success=True, data=result_data)