frytrader 0.1.0__py3-none-any.whl

easytrader/__init__.py

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+import urllib3
+
+from easytrader import exceptions
+from easytrader.api import use, follower
+from easytrader.log import logger
+
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+__version__ = "0.1.0"
+__author__ = "shidenggui, Fryt1"

easytrader/api.py

@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+import logging
+import sys
+
+import six
+
+from easytrader.joinquant_follower import JoinQuantFollower
+from easytrader.log import logger
+from easytrader.ricequant_follower import RiceQuantFollower
+from easytrader.xq_follower import XueQiuFollower
+from easytrader.xqtrader import XueQiuTrader
+
+if sys.version_info <= (3, 5):
+    raise TypeError("不支持 Python3.5 及以下版本，请升级")
+
+
+def use(broker, debug=False, **kwargs):
+    """用于生成特定的券商对象
+    :param broker: 券商名支持 
+        例如 ['miniqmt', 'xq', '雪球', 'gj_client', '国金客户端', "universal_client", "通用同花顺客户端", "ths", "同花顺客户端"] 等
+    :param debug: 控制 debug 日志的显示, 默认为 False
+    :param initial_assets: [雪球参数] 控制雪球初始资金，默认为一百万
+    :return the class of trader
+
+    Usage::
+
+        >>> import easytrader
+        >>> user = easytrader.use('xq')
+        >>> user.prepare('xq.json')
+    """
+    if "background" in kwargs:
+        if not isinstance(kwargs["background"], bool):
+            raise TypeError("background 必须是布尔值")
+        if kwargs["background"] and broker.lower() not in (
+            "ths", "同花顺客户端"
+        ):
+            raise NotImplementedError("后台只读模式仅支持 ths 远航版")
+
+    if debug:
+        logger.setLevel(logging.DEBUG)
+
+    if broker.lower() in ["xq", "雪球"]:
+        return XueQiuTrader(**kwargs)
+
+    if broker.lower() in ["yh_client", "银河客户端"]:
+        from .yh_clienttrader import YHClientTrader
+
+        return YHClientTrader()
+
+    if broker.lower() in ["ht_client", "华泰客户端"]:
+        from .ht_clienttrader import HTClientTrader
+
+        return HTClientTrader()
+
+    if broker.lower() in ["wk_client", "五矿客户端"]:
+        from easytrader.wk_clienttrader import WKClientTrader
+
+        return WKClientTrader()
+
+    if broker.lower() in ["htzq_client", "海通证券客户端"]:
+        from easytrader.htzq_clienttrader import HTZQClientTrader
+
+        return HTZQClientTrader()
+
+    if broker.lower() in ["gj_client", "国金客户端"]:
+        from .gj_clienttrader import GJClientTrader
+
+        return GJClientTrader()
+
+    if broker.lower() in ["gf_client", "广发客户端"]:
+        from .gf_clienttrader import GFClientTrader
+
+        return GFClientTrader()
+
+    if broker.lower() in ["universal_client", "通用同花顺客户端"]:
+        from easytrader.universal_clienttrader import UniversalClientTrader
+
+        return UniversalClientTrader()
+
+    if broker.lower() in ["ths", "同花顺客户端"]:
+        background = kwargs.pop("background", False)
+        if not isinstance(background, bool):
+            raise TypeError("background 必须是布尔值")
+        if background:
+            from .background_trader import BackgroundThsTrader
+
+            return BackgroundThsTrader(**kwargs)
+        from .clienttrader import ClientTrader
+
+        return ClientTrader()
+    
+    if broker.lower() in ["miniqmt"]:
+        try:
+            import xtquant
+        except:
+            logger.error("miniqmt 相关组件 xtqimt 未安装, 请执行 pip install easytrader[xtquant]安装")
+        from easytrader.miniqmt.miniqmt_trader import MiniqmtTrader
+
+        return MiniqmtTrader()
+
+    raise NotImplementedError
+
+
+def follower(platform, **kwargs):
+    """用于生成特定的券商对象
+    :param platform:平台支持 ['jq', 'joinquant', '聚宽’]
+    :param initial_assets: [雪球参数] 控制雪球初始资金，默认为一万,
+        总资金由 initial_assets * 组合当前净值 得出
+    :param total_assets: [雪球参数] 控制雪球总资金，无默认值,
+        若设置则覆盖 initial_assets
+    :return the class of follower
+
+    Usage::
+
+        >>> import easytrader
+        >>> user = easytrader.use('xq')
+        >>> user.prepare('xq.json')
+        >>> jq = easytrader.follower('jq')
+        >>> jq.login(user='username', password='password')
+        >>> jq.follow(users=user, strategies=['strategies_link'])
+    """
+    if platform.lower() in ["rq", "ricequant", "米筐"]:
+        return RiceQuantFollower()
+    if platform.lower() in ["jq", "joinquant", "聚宽"]:
+        return JoinQuantFollower()
+    if platform.lower() in ["xq", "xueqiu", "雪球"]:
+        return XueQiuFollower(**kwargs)
+    raise NotImplementedError

easytrader/background_focus.py

@@ -0,0 +1,147 @@
+"""Bounded thread-specific activation veto for the x86 Voyager UI thread.
+
+The helper and hook are built from the included native sources. They never
+patch executable files or send input; Windows removes the hook if the host
+exits. Each query owns its helper and explicitly stops it in finally.
+"""
+
+import hashlib
+import os
+import struct
+import subprocess
+import tempfile
+import time
+from pathlib import Path
+
+from easytrader.background_runtime import BackgroundError
+
+_HASHES = {
+    "focus_guard32.dll": (
+        "6f9143a13d75c0e2b874a6854bb1bede94f0408bae7e4dff086941e88481236b"
+    ),
+    "focus_host32.exe": (
+        "5753bae4146a96b394f4981858e2d425330595c23b236b42bc0c90e19eb49f24"
+    ),
+}
+
+
+def pe_machine(path):
+    """Reject a bitness mismatch before a window is moved or hooked."""
+    with Path(path).open("rb") as stream:
+        if stream.read(2) != b"MZ":
+            raise BackgroundError("原生保护文件不是 PE 可执行文件")
+        stream.seek(0x3C)
+        offset_bytes = stream.read(4)
+        if len(offset_bytes) != 4:
+            raise BackgroundError("PE 文件头不完整")
+        offset = struct.unpack("<I", offset_bytes)[0]
+        if not 64 <= offset <= 16 * 1024 * 1024:
+            raise BackgroundError("PE 文件头位置无效")
+        stream.seek(offset)
+        if stream.read(4) != b"PE\0\0":
+            raise BackgroundError("PE 文件签名无效")
+        machine = stream.read(2)
+        if len(machine) != 2:
+            raise BackgroundError("PE 架构信息缺失")
+        return struct.unpack("<H", machine)[0]
+
+
+class ThreadActivationGuard:
+    def __init__(self, directory=None):
+        self.directory = (
+            Path(directory)
+            if directory
+            else Path(__file__).resolve().parent / "native" / "bin"
+        )
+        for name, expected in _HASHES.items():
+            path = self.directory / name
+            if not path.is_file():
+                raise BackgroundError("缺少原生防激活保护，请重新安装完整包")
+            if hashlib.sha256(path.read_bytes()).hexdigest() != expected:
+                raise BackgroundError("原生防激活保护校验失败，请重新安装")
+            if pe_machine(path) != 0x14C:
+                raise BackgroundError("当前防激活保护仅支持 x86 下单窗口")
+        self.host = None
+        self.temp = None
+        self.log = None
+        self.folder = None
+
+    def start(self, backend, guard):
+        if self.host is not None:
+            raise BackgroundError("原生防激活保护已在运行")
+        guard.check()
+        hwnd, pid, thread = backend.hook_target()
+        self.temp = tempfile.TemporaryDirectory(prefix="frytrader-focus-")
+        self.folder = Path(self.temp.name)
+        self.log = (self.folder / "host.log").open("wb")
+        (self.folder / "target.txt").write_text(
+            "%d,0,%d,%d,%d" % (hwnd, pid, thread, os.getpid()),
+            encoding="ascii",
+        )
+        environment = os.environ.copy()
+        environment["FRYTRADER_FOCUS_DIRECTORY"] = str(self.folder)
+        environment["FRYTRADER_FOCUS_DLL"] = str(
+            self.directory / "focus_guard32.dll"
+        )
+        try:
+            self.host = subprocess.Popen(
+                [str(self.directory / "focus_host32.exe")],
+                env=environment,
+                stdin=subprocess.DEVNULL,
+                stdout=self.log,
+                stderr=subprocess.STDOUT,
+                shell=False,
+            )
+            deadline = min(time.monotonic() + 2, guard.deadline)
+            ready = self.folder / "ready.txt"
+            while not ready.exists():
+                guard.check()
+                if self.host.poll() is not None:
+                    raise BackgroundError(
+                        "防激活宿主未就绪（exit=%s）" % self.host.returncode
+                    )
+                if time.monotonic() >= deadline:
+                    raise BackgroundError("防激活宿主启动超时")
+                time.sleep(0.01)
+            if ready.read_text(encoding="ascii") != "ok":
+                raise BackgroundError("防激活宿主就绪信号无效")
+            self.check()
+            guard.check()
+        except BaseException:
+            self.close()
+            raise
+
+    def check(self):
+        if self.host is None or self.host.poll() is not None:
+            raise BackgroundError("防激活保护已退出，已停止后台操作")
+
+    def close(self):
+        failed = None
+        try:
+            if self.host is not None:
+                if self.host.poll() is None:
+                    (self.folder / "stop.txt").write_text(
+                        "stop", encoding="ascii"
+                    )
+                    try:
+                        self.host.wait(timeout=2)
+                    except subprocess.TimeoutExpired:
+                        # Only our own helper is terminated.
+                        self.host.terminate()
+                        self.host.wait(timeout=2)
+                        failed = "防激活宿主未能正常卸载，已终止自己的宿主"
+                if self.host.returncode != 0 and failed is None:
+                    failed = (
+                        "防激活宿主退出异常（exit=%s）" % self.host.returncode
+                    )
+        finally:
+            self.host = None
+            if self.log is not None:
+                self.log.close()
+                self.log = None
+            if self.temp is not None:
+                self.temp.cleanup()
+                self.temp = None
+                self.folder = None
+        if failed:
+            raise BackgroundError(failed)

easytrader/background_runtime.py

@@ -0,0 +1,133 @@
+"""Nonactivating AutoHotkey bridge, restricted to captcha controls."""
+
+import os
+import shutil
+import subprocess
+import tempfile
+import time
+from pathlib import Path
+
+
+class BackgroundError(RuntimeError):
+    """A background read could not be confirmed safely."""
+
+
+class BackgroundInterference(BackgroundError):
+    """The user desktop or foreground was unexpectedly taken by Voyager."""
+
+
+def resolve_ahk_path(configured=None):
+    """Use a configured/runtime executable; never download or install here."""
+    candidates = []
+    if configured:
+        candidates.append(Path(configured).expanduser())
+    elif os.environ.get("FRYTRADER_AHK"):
+        candidates.append(Path(os.environ["FRYTRADER_AHK"]).expanduser())
+    else:
+        local = os.environ.get("LOCALAPPDATA")
+        if local:
+            candidates.append(
+                Path(local) / "Frytrader" / "runtime" / "AutoHotkey64.exe"
+            )
+        discovered = shutil.which("AutoHotkey64.exe")
+        if discovered:
+            candidates.append(Path(discovered))
+        program_files = os.environ.get("ProgramFiles")
+        if program_files:
+            candidates.append(
+                Path(program_files) / "AutoHotkey" / "v2" / "AutoHotkey64.exe"
+            )
+    for candidate in candidates:
+        if candidate.is_file() and candidate.suffix.lower() == ".exe":
+            return candidate.resolve()
+    raise BackgroundError(
+        "未找到 AutoHotkey v2。请配置 ahk_path 或 FRYTRADER_AHK，"
+        "后台模式不会自动安装软件或退回真实键鼠输入。"
+    )
+
+
+class AutoHotkeyControls:
+    """Only ControlSendText and ControlClick NA; no foreground fallback."""
+
+    def __init__(self, ahk_path=None):
+        self.executable = resolve_ahk_path(ahk_path)
+        self.script = (
+            Path(__file__).resolve().parent
+            / "config"
+            / "background_controls.ahk"
+        )
+        if not self.script.is_file():
+            raise BackgroundError("后台控件脚本未随安装包提供")
+
+    @staticmethod
+    def validate_code(code):
+        if (
+            not isinstance(code, str)
+            or len(code) != 4
+            or not code.isascii()
+            or not code.isalnum()
+        ):
+            raise BackgroundError("验证码必须是四位 ASCII 字母或数字")
+
+    def _run(self, action, controls, control, pid, guard, code=None):
+        guard.check()
+        with tempfile.TemporaryDirectory(prefix="frytrader-na-") as folder:
+            directory = Path(folder)
+            output = directory / "result.txt"
+            command = [
+                str(self.executable),
+                "/ErrorStdOut",
+                str(self.script),
+                action,
+                str(controls.dialog),
+                str(control),
+                str(pid),
+                str(output),
+            ]
+            if code is not None:
+                command.append(code)
+            # Regular files/DEVNULL, not named-pipe stdio. No shell quoting.
+            with (directory / "helper.log").open("wb") as log:
+                process = subprocess.Popen(
+                    command,
+                    stdin=subprocess.DEVNULL,
+                    stdout=log,
+                    stderr=subprocess.STDOUT,
+                    shell=False,
+                )
+                try:
+                    deadline = min(time.monotonic() + 5, guard.deadline)
+                    while process.poll() is None:
+                        guard.check()
+                        if time.monotonic() >= deadline:
+                            raise BackgroundError("NA 控件操作超时")
+                        time.sleep(0.01)
+                    guard.check()
+                    result = (
+                        output.read_text(encoding="utf-8-sig")
+                        if output.exists()
+                        else "missing result"
+                    )
+                    if process.returncode or result != "ok":
+                        raise BackgroundError(
+                            "NA 控件操作未确认（%s，exit=%s）"
+                            % (action, process.returncode)
+                        )
+                finally:
+                    # This is our short-lived bridge, never the trader.
+                    if process.poll() is None:
+                        process.terminate()
+                        try:
+                            process.wait(timeout=1)
+                        except subprocess.TimeoutExpired:
+                            process.kill()
+                            process.wait(timeout=1)
+
+    def fill(self, controls, code, pid, guard):
+        self.validate_code(code)
+        self._run("fill", controls, controls.editor, pid, guard, code)
+
+    def click(self, controls, role, pid, guard):
+        if role not in ("ok", "cancel", "image"):
+            raise BackgroundError("后台桥接只允许验证码对话框操作")
+        self._run("click", controls, getattr(controls, role), pid, guard)