解决通用同花顺交易端输入验证码

[ouchengxian]

env

OS: win7/ win10 / mac / linux PYTHON_VERSION: 3.x EASYTRADER_VERSION: 0.xx.xx BROKER_TYPE: gj / ht / xq / xxx

problem

how to repeat

1、通过修改 grid_strategies.py文件 2、安装Tesseract-OCR

class Xls(BaseStrategy):
‘’‘通过将 Grid 另存为 xls #文件再读取的方式获取 grid 内容’‘’ def init(self, tmp_folder: Optional[str] = None):
‘’‘param tmp_folder: 用于保持临时文件的文件夹’‘’ super().init() self.tmp_folder = tmp_folder def get(self, control_id: int) -> List[Dict]: logger.info("保存 grid 内容为 xls 文件模式") grid = self._get_grid(control_id)
# ctrl+s 保存 grid 内容为 xls 文件 self._set_foreground(grid) # setFocus buggy, instead of SetForegroundWindow grid.type_keys("^s", set_foreground=False) ################################ #检查是否需要输入验证码 if (self._trader.app.top_window().window( class_name="Static", title_re="验证码" ).exists(timeout=1)): #检查是否有验证码输入Window file_path = "tmp.png" count = 5 found = False while count > 0: self._trader.app.top_window().window( control_id=0x965, class_name="Static" ).capture_as_image().save( file_path ) # 保存验证码图片 captcha_num = captcha_recognize(file_path).strip() # 识别验证码 captcha_num = "".join(captcha_num.split()) logger.info("captcha result-->" + captcha_num) if len(captcha_num) == 4: ###### 模拟输入验证码##### editor = self._trader.app.top_window().window( control_id=0x964, class_name="Edit" ) editor.select() editor.type_keys(captcha_num) ######################### self._trader.app.top_window().set_focus() pywinauto.keyboard.SendKeys("{ENTER}") # 模拟发送enter，点击确定 if self._trader.app.window(title='另存为').exists(timeout=1): found = True break logger.info(f"captcha result:{captcha_num} error") count -= 1 self._trader.wait(0.1) self._trader.app.top_window().window( control_id=0x965, class_name="Static" ).click() if not found: self._trader.app.top_window().Button2.click() # 点击取消 #上述是验证码处理 ##################################
temp_path = tempfile.mktemp(suffix=".xls", dir=self.tmp_folder)
self._set_foreground(self._trader.app.top_window()) # alt+s保存，alt+y替换已存在的文件 self._trader.app.top_window().Edit1.set_edit_text(temp_path) self._trader.wait(0.1) self._trader.app.top_window().type_keys("%{s}%{y}", set_foreground=False) # Wait until file save complete otherwise pandas can not find file self._trader.wait(0.2) if self._trader.is_exist_pop_dialog(): self._trader.app.top_window().Button2.click() self._trader.wait(0.2)
return self._format_grid_data(temp_path)

[huaixu1358]

方便给一份完整代码我吗

[Ziven-L]

您好，可以给我一份完整代码吗？

[skyformat99]

推送pr啊

[yangzhipeng1108]

class Xls(BaseStrategy): """ 通过将 Grid 另存为 xls 文件再读取的方式获取 grid 内容 """

def __init__(self, tmp_folder: Optional[str] = None):
    """
    :param tmp_folder: 用于保持临时文件的文件夹
    """
    super().__init__()
    self.tmp_folder = tmp_folder

def get(self, control_id: int) -> List[Dict]:
    grid = self._get_grid(control_id)

    # ctrl+s 保存 grid 内容为 xls 文件
    self._set_foreground(grid)  # setFocus buggy, instead of SetForegroundWindow
    grid.type_keys("^s", set_foreground=False)
    # count = 10
    # while count > 0:
    #     if self._trader.is_exist_pop_dialog():
    #         break
    #     self._trader.wait(0.2)
    #     count -= 1

    if  (self._trader.app.top_window().window(class_name="Static", title_re="验证码").exists(timeout=1)): #检查是否有验证码输入Window
        file_path = "tmp.png"
        count = 5
        found = False
        while count > 0:
            self._trader.app.top_window().window(
                control_id=0x965, class_name="Static"
            ).capture_as_image().save(
                file_path
            )  # 保存验证码

            captcha_num = captcha_recognize(file_path).strip()  # 识别验证码
            captcha_num = "".join(captcha_num.split())
            logger.info("captcha result-->" + captcha_num)
            if len(captcha_num) == 4:

                editor = self._trader.app.top_window().window(
                    control_id=0x964, class_name="Edit")
                editor.select()
                editor.type_keys(captcha_num)

                self._trader.app.top_window().set_focus()
                pywinauto.keyboard.SendKeys("{ENTER}")  # 模拟发送enter，点击确定
                if self._trader.app.window(title='另存为').exists(timeout=1):
                    found = True
                    break
                logger.info(f"captcha result:{captcha_num} error")
            count -= 1
            self._trader.wait(0.1)
            self._trader.app.top_window().window(
                control_id=0x965, class_name="Static"
            ).click()
        if not found:
            self._trader.app.top_window().Button2.click()  # 点击取消

    temp_path = tempfile.mktemp(suffix=".xls", dir=self.tmp_folder)
    self._set_foreground(self._trader.app.top_window())

    # alt+s保存，alt+y替换已存在的文件
    self._trader.app.top_window().Edit1.set_edit_text(temp_path)
    self._trader.wait(0.1)
    self._trader.app.top_window().type_keys("%{s}%{y}", set_foreground=False)
    # Wait until file save complete otherwise pandas can not find file
    self._trader.wait(0.2)
    if self._trader.is_exist_pop_dialog():
        self._trader.app.top_window().Button2.click()
        self._trader.wait(0.2)

    return self._format_grid_data(temp_path)

def _format_grid_data(self, data: str) -> List[Dict]:
    with open(data, encoding="gbk", errors="replace") as f:
        content = f.read()

    df = pd.read_csv(
        StringIO(content),
        delimiter="\t",
        dtype=self._trader.config.GRID_DTYPE,
        na_filter=False,
    )
    return df.to_dict("records")

[yangzhipeng1108]

Tesseract 使用参考这两文章https://blog.csdn.net/u010454030/article/details/80515501 和 https://blog.csdn.net/wang_hugh/article/details/80760940