qka 0.1.0__tar.gz

qka-0.1.0/.github/workflows/release.yml

@@ -0,0 +1,70 @@
+name: Python package
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  Test:
+    runs-on: windows-latest
+    permissions: write-all
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0  # Ensure full history for semantic-release
+
+      - name: Set up Python 3.10
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'  # You can specify your required Python version here
+
+      - name: Install Flit
+        run: |
+          pip install flit
+        shell: bash
+
+      - name: Check release status
+        id: release-status
+        run: |
+          pip install python-semantic-release
+          if semantic-release --noop --strict version; then
+            echo "::set-output name=released::true"
+          else
+            echo "::set-output name=released::false"
+          fi
+        shell: bash
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Semantic Release Version
+        if: steps.release-status.outputs.released == 'true'
+        run: |
+          semantic-release version
+        shell: bash
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }}
+
+      - name: Release to PyPI
+        if: steps.release-status.outputs.released == 'true'
+        run: |
+          flit build
+          flit publish --repository pypi
+        shell: bash
+        env:
+          FLIT_USERNAME: __token__
+          FLIT_PASSWORD: ${{ secrets.PYPI_TOKEN }}
+
+      - name: Release to GitHub
+        if: steps.release-status.outputs.released == 'true'
+        run: |
+          git fetch --tags
+          for file in ./dist/*; do
+            gh release upload "${{ steps.release-status.outputs.tag }}" $file
+          done
+        shell: bash
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}

qka-0.1.0/.gitignore

@@ -0,0 +1,164 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+*.ipynb

qka-0.1.0/.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "editor.tabSize": 4
+}

qka-0.1.0/CHANGELOG.md

@@ -0,0 +1,23 @@
+# CHANGELOG
+
+
+## v0.1.0 (2025-01-19)
+
+### Bug Fixes
+
+- 优化  ([`c77eae5`](https://github.com/zsrl/qka/commit/c77eae539e5bd64674a6e280af1eb19c8f9c75e5))
+
+### Features
+
+- Qmt服务端与客户端
+  ([`23baabf`](https://github.com/zsrl/qka/commit/23baabfc001ed1ac3f24c17cef55f190f86fc23a))
+
+- 初始化项目  ([`c6fbea5`](https://github.com/zsrl/qka/commit/c6fbea5eea52311856b67bba48b11c5e298d0cf3))
+
+### Performance Improvements
+
+- 初始化包  ([`9ec3368`](https://github.com/zsrl/qka/commit/9ec3368bdf6d866ff5334f379c0e7a2a2c883663))
+
+- 初始化框架  ([`e549e45`](https://github.com/zsrl/qka/commit/e549e45c6f4c16ab04076c7854e4cef25013b263))
+
+- 增加发布流程  ([`e08523a`](https://github.com/zsrl/qka/commit/e08523a02b08f36bcca339dacac0059b0821eb8e))

qka-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 量化投资技术
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

qka-0.1.0/PKG-INFO

@@ -0,0 +1,77 @@
+Metadata-Version: 2.3
+Name: qka
+Version: 0.1.0
+Summary: 
+Author-email: myc <mayuanchi1029@gmail.com>
+Description-Content-Type: text/markdown
+Classifier: License :: OSI Approved :: MIT License
+Requires-Dist: xtquant
+Requires-Dist: fastapi
+Requires-Dist: uvicorn
+Project-URL: Home, https://github.com/zsrl/qka
+
+# qka (快量化)
+
+快捷量化助手（Quick Quantitative Assistant）是一个简洁易用，可实操A股的量化交易框架。
+
+## 安装
+
+```bash
+pip install qka
+```
+
+## 使用方法
+
+### QMTServer
+
+```python
+from qka.server import QMTServer
+
+server = QMTServer("YOUR_ACCOUNT_ID", "YOUR_QMT_PATH")
+# 服务器启动时会打印生成的token
+server.start()
+```
+
+### QMTClient
+
+#### 查询
+
+```python
+from qka.client import QMTClient
+
+client = QMTClient(token="服务器打印的token")
+# 调用接口
+result = client.api("query_stock_asset")
+```
+
+#### 下单
+
+```python
+from qka.client import QMTClient
+from xtquant import xtconstant
+
+client = QMTClient(token="服务器打印的token")
+# 调用接口
+result = client.api("order_stock", stock_code='600000.SH', order_type=xtconstant.STOCK_BUY, order_volume =1000, price_type=xtconstant.FIX_PRICE, price=10.5)
+```
+<!-- ```python
+datas = qka.data(
+  stock_list=[], 
+  period='tick', 
+  indicators=[
+    'MA',
+    'BOLL'
+  ]
+)
+
+
+def strategy(bar, bars, borker):
+  pass
+
+res = qka.backtest(datas, start_time='', end_time='', strategy=strategy)
+
+borker = qka.broker(type='qmt', config={})
+
+qka.trade(datas, start_time='', strategy=strategy, borker=borker)
+
+``` -->

qka-0.1.0/README.md

@@ -0,0 +1,65 @@
+# qka (快量化)
+
+快捷量化助手（Quick Quantitative Assistant）是一个简洁易用，可实操A股的量化交易框架。
+
+## 安装
+
+```bash
+pip install qka
+```
+
+## 使用方法
+
+### QMTServer
+
+```python
+from qka.server import QMTServer
+
+server = QMTServer("YOUR_ACCOUNT_ID", "YOUR_QMT_PATH")
+# 服务器启动时会打印生成的token
+server.start()
+```
+
+### QMTClient
+
+#### 查询
+
+```python
+from qka.client import QMTClient
+
+client = QMTClient(token="服务器打印的token")
+# 调用接口
+result = client.api("query_stock_asset")
+```
+
+#### 下单
+
+```python
+from qka.client import QMTClient
+from xtquant import xtconstant
+
+client = QMTClient(token="服务器打印的token")
+# 调用接口
+result = client.api("order_stock", stock_code='600000.SH', order_type=xtconstant.STOCK_BUY, order_volume =1000, price_type=xtconstant.FIX_PRICE, price=10.5)
+```
+<!-- ```python
+datas = qka.data(
+  stock_list=[], 
+  period='tick', 
+  indicators=[
+    'MA',
+    'BOLL'
+  ]
+)
+
+
+def strategy(bar, bars, borker):
+  pass
+
+res = qka.backtest(datas, start_time='', end_time='', strategy=strategy)
+
+borker = qka.broker(type='qmt', config={})
+
+qka.trade(datas, start_time='', strategy=strategy, borker=borker)
+
+``` -->

qka-0.1.0/pyproject.toml

@@ -0,0 +1,46 @@
+[build-system]
+requires = ["flit_core >=3.2,<4"]
+build-backend = "flit_core.buildapi"
+
+[project]
+name = "qka"
+version = "0.1.0"
+description = ""
+authors = [{name = "myc", email = "mayuanchi1029@gmail.com"}]
+readme = "README.md"
+license = {file = "LICENSE"}
+classifiers = ["License :: OSI Approved :: MIT License"]
+dependencies = [
+    "xtquant",
+    "fastapi",
+    "uvicorn"
+]
+
+[project.urls]
+Home = "https://github.com/zsrl/qka"
+
+[tool.semantic_release]
+version_toml = [
+    "pyproject.toml:project.version"
+]
+
+branch = "main"
+upload_to_PyPI = true
+upload_to_release = true
+commit_author = "github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>"
+
+[tool.semantic_release.commit_parser_options]
+allowed_tags = [
+    "build",
+    "chore",
+    "ci",
+    "docs",
+    "feat",
+    "fix",
+    "perf",
+    "style",
+    "refactor",
+    "test"
+]
+minor_tags = ["feat"]
+patch_tags = ["fix", "perf"]

qka-0.1.0/qka/__init__.py

@@ -0,0 +1,5 @@
+from qka.data import data
+from qka.backtest import backtest
+from qka.trade import create_trader
+
+__all__ = ['data', 'backtest', 'create_trader']

qka-0.1.0/qka/anis.py

@@ -0,0 +1,6 @@
+# ANSI 转义序列
+RED = "\033[91m"
+GREEN = "\033[92m"
+YELLOW = "\033[93m"
+BLUE = "\033[94m"
+RESET = "\033[0m"

qka-0.1.0/qka/backtest.py

@@ -0,0 +1,3 @@
+
+def backtest():
+  pass

qka-0.1.0/qka/client.py

@@ -0,0 +1,43 @@
+import requests
+from typing import Any, Dict
+from qka.logger import logger
+
+class QMTClient:
+    def __init__(self, base_url: str = "http://localhost:8000", token: str = None):
+        """初始化交易客户端
+        Args:
+            base_url: API服务器地址，默认为本地8000端口
+            token: 访问令牌，必须与服务器的token一致
+        """
+        self.base_url = base_url.rstrip('/')
+        self.session = requests.Session()
+        if not token:
+            raise ValueError("必须提供访问令牌(token)")
+        self.token = token
+        self.headers = {"X-Token": self.token}
+
+    def api(self, method_name: str, **params) -> Any:
+        """通用调用接口方法
+        Args:
+            method_name: 要调用的接口名称
+            **params: 接口参数，作为关键字参数传入
+        Returns:
+            接口返回的数据
+        """
+        try:
+            response = self.session.post(
+                f"{self.base_url}/api/{method_name}",
+                json=params or {},
+                headers=self.headers
+            )
+            response.raise_for_status()
+            result = response.json()
+            
+            if not result.get('success'):
+                raise Exception(f"API调用失败: {result.get('detail')}")
+            
+            return result.get('data')
+        except Exception as e:
+            logger.error(f"调用 {method_name} 失败: {str(e)}")
+            raise
+

qka-0.1.0/qka/data.py

@@ -0,0 +1,54 @@
+import time
+from datetime import datetime
+from xtquant import xtdata
+import numpy as np
+from qka.logger import logger
+
+class QMTData:
+    def __init__(self, stocks=None, sector=None, indicators=None):
+        self.stocks = stocks
+        if sector is not None:
+            self.stocks = xtdata.get_stock_list_in_sector(sector)
+        
+        self.indicators = indicators
+    
+    def get(self, period, start_time='', end_time=''):
+
+        for stock in self.stocks:
+            xtdata.download_history_data(stock_code=stock, period=period, start_time=start_time, end_time=end_time, incrementally=True)
+        
+        res = xtdata.get_local_data(stock_list=self.stocks, period=period, start_time=start_time, end_time=end_time)
+
+        return res
+    
+    def subscribe(self, callback):
+
+        delays = []
+
+        def task(res):
+            for code, item in res.items():
+                rate = (item['lastPrice'] - item['lastClose']) / item['lastClose']
+                timetag = datetime.fromtimestamp(item['time'] / 1000)
+                current_time = datetime.fromtimestamp(time.time())
+                delay_seconds = (current_time - timetag).total_seconds()
+                if delay_seconds < 1000:
+
+                    # delays.append(delay_seconds)
+
+                    # if len(delays) >= 1_000_000:
+                    #     arr = np.array(delays)
+                    #     mean = arr.mean()
+                    #     delays.clear()
+                    #     logger.info(f'平均延迟 {mean}')
+
+                    if delay_seconds > 3:
+                        logger.warning(f'{code} 延迟 {delay_seconds}')
+
+                callback(code, item)
+
+        xtdata.subscribe_whole_quote(code_list=self.stocks, callback=task)
+
+        xtdata.run()
+
+def data(stocks=None, sector=None, indicators=None):
+    return QMTData(stocks, sector, indicators)

qka-0.1.0/qka/logger.py

@@ -0,0 +1,69 @@
+import logging
+import requests
+import os
+import re
+from datetime import date
+
+formatter = logging.Formatter('[%(levelname)s][%(asctime)s]%(message)s', datefmt='%Y-%m-%d %H:%M:%S')
+
+class RemoveAnsiEscapeCodes(logging.Filter):
+    def filter(self, record):
+        record.msg = re.sub(r'\033\[[0-9;]*m', '', str(record.msg))
+        return True
+
+def create_logger():
+
+    logger = logging.getLogger('log')
+    logger.setLevel(logging.DEBUG)
+
+    stream_handler = logging.StreamHandler()
+    stream_handler.setFormatter(formatter)
+    logger.addHandler(stream_handler)
+
+    # 创建日志文件夹（如果不存在）
+    log_dir = 'logs'
+    if not os.path.exists(log_dir):
+        os.makedirs(log_dir)
+
+    # 创建文件处理器，并将日志写入文件
+    file_handler = logging.FileHandler(f"{log_dir}/{date.today().strftime('%Y-%m-%d')}.log", encoding='utf-8')
+    file_handler.setLevel(logging.DEBUG)
+    file_handler.setFormatter(formatter)
+    file_handler.addFilter(RemoveAnsiEscapeCodes())
+    logger.addHandler(file_handler)
+
+    return logger
+
+
+class WeChatHandler(logging.Handler):
+    def __init__(self, webhook_url):
+        super().__init__()
+        self.webhook_url = webhook_url
+
+    def emit(self, record):
+        log_entry = self.format(record)
+        payload = {
+            "msgtype": "text",
+            "text": {
+                "content": log_entry
+            }
+        }
+        try:
+            response = requests.post(self.webhook_url, json=payload)
+            response.raise_for_status()
+        except requests.exceptions.RequestException as e:
+            print(f"Failed to send log to WeChat: {e}")
+
+def add_wechat_handler(logger, level, wechat_webhook_url):
+
+    if wechat_webhook_url is not None:
+
+        # 创建企业微信处理器，并将日志发送到企业微信
+        wechat_handler = WeChatHandler(wechat_webhook_url)
+        wechat_handler.setLevel(level or logging.INFO)
+        wechat_handler.setFormatter(formatter)
+        wechat_handler.addFilter(RemoveAnsiEscapeCodes())
+        logger.addHandler(wechat_handler)
+
+
+logger = create_logger()

qka-0.1.0/qka/server.py

@@ -0,0 +1,126 @@
+from fastapi import FastAPI, HTTPException, Header, Depends
+from pydantic import BaseModel
+import inspect
+from typing import Any, Dict
+from qka.trade import create_trader
+import uvicorn
+import uuid
+import hashlib
+
+class QMTServer:
+    def __init__(self, account_id: str, mini_qmt_path: str, host: str = "0.0.0.0", port: int = 8000, token: str = None):
+        """初始化交易服务器
+        Args:
+            account_id: 账户ID
+            mini_qmt_path: miniQMT路径
+            host: 服务器地址，默认0.0.0.0
+            port: 服务器端口，默认8000
+            token: 可选的自定义token
+        """
+        self.account_id = account_id
+        self.mini_qmt_path = mini_qmt_path
+        self.host = host
+        self.port = port
+        self.app = FastAPI()
+        self.trader = None
+        self.account = None
+        self.token = token if token else self.generate_token()  # 使用自定义token或生成固定token
+        print(f"\n授权Token: {self.token}\n")  # 打印token供客户端使用
+
+    def generate_token(self) -> str:
+        """生成基于机器码的固定token"""
+        # 获取机器码（例如MAC地址）
+        mac = uuid.getnode()
+        # 将机器码转换为字符串
+        mac_str = str(mac)
+        # 使用SHA256哈希生成固定长度的token
+        token = hashlib.sha256(mac_str.encode()).hexdigest()
+        return token
+
+    async def verify_token(self, x_token: str = Header(...)):
+        """验证token的依赖函数"""
+        if x_token != self.token:
+            raise HTTPException(status_code=401, detail="无效的Token")
+        return x_token
+
+    def init_trader(self):
+        """初始化交易对象"""
+        self.trader, self.account = create_trader(self.account_id, self.mini_qmt_path)
+
+    def convert_to_dict(self, obj):
+        """将结果转换为可序列化的字典"""
+        # 如果是基本类型，直接返回
+        if isinstance(obj, (int, float, str, bool)):
+            return obj
+        # 如果已经是字典类型，直接返回
+        elif isinstance(obj, dict):
+            return obj
+        # 如果是列表或元组，递归转换每个元素
+        elif isinstance(obj, (list, tuple)):
+            return [self.convert_to_dict(item) for item in obj]
+        # 如果是自定义对象，获取所有公开属性
+        elif hasattr(obj, '__dir__'):
+            attrs = obj.__dir__()
+            # 过滤掉内部属性和方法
+            public_attrs = {attr: getattr(obj, attr) 
+                           for attr in attrs 
+                           if not attr.startswith('_') and not callable(getattr(obj, attr))}
+            return public_attrs
+        # 其他类型直接返回
+        return str(obj)  # 将无法处理的类型转换为字符串
+
+    def convert_method_to_endpoint(self, method_name: str, method):
+        """将 XtQuantTrader 方法转换为 FastAPI 端点"""
+        sig = inspect.signature(method)
+        param_names = list(sig.parameters.keys())
+        
+        # 创建动态的请求模型
+        class_fields = {
+            '__annotations__': {}  # 添加类型注解字典
+        }
+        
+        for param in param_names:
+            if param in ['self', 'account']:
+                continue
+            class_fields['__annotations__'][param] = Any
+            class_fields[param] = None
+
+        RequestModel = type(f'{method_name}Request', (BaseModel,), class_fields)
+
+        async def endpoint(request: RequestModel, token: str = Depends(self.verify_token)):
+            try:
+                params = request.dict(exclude_unset=True)
+                if 'account' in param_names:
+                    params['account'] = self.account
+                result = getattr(self.trader, method_name)(**params)
+                converted_result = self.convert_to_dict(result)
+                return {'success': True, 'data': converted_result}
+            except Exception as e:
+                raise HTTPException(status_code=500, detail=str(e))
+
+        self.app.post(f'/api/{method_name}')(endpoint)
+
+    def setup_routes(self):
+        """设置所有路由"""
+        trader_methods = inspect.getmembers(
+            self.trader.__class__, 
+            predicate=lambda x: inspect.isfunction(x) or inspect.ismethod(x)
+        )
+        
+        excluded_methods = {'__init__', '__del__', 'register_callback', 'start', 'stop',
+                          'connect', 'sleep', 'run_forever', 'set_relaxed_response_order_enabled'}
+        
+        for method_name, method in trader_methods:
+            if not method_name.startswith('_') and method_name not in excluded_methods:
+                self.convert_method_to_endpoint(method_name, method)
+
+    def start(self):
+        """启动服务器"""
+        self.init_trader()
+        self.setup_routes()
+        uvicorn.run(self.app, host=self.host, port=self.port)
+
+def qmt_server(account_id: str, mini_qmt_path: str, host: str = "0.0.0.0", port: int = 8000, token: str = None):
+    """快速创建并启动服务器的便捷函数"""
+    server = QMTServer(account_id, mini_qmt_path, host, port, token)
+    server.start()

qka-0.1.0/qka/trade.py

@@ -0,0 +1,71 @@
+from xtquant.xttrader import XtQuantTrader, XtQuantTraderCallback
+from xtquant.xttype import StockAccount
+import random
+from qka.utils import timestamp_to_datetime_string, parse_order_type, convert_to_current_date
+from qka.anis import RED, GREEN, YELLOW, BLUE, RESET
+from qka.logger import logger
+
+error_orders = []
+
+class MyXtQuantTraderCallback(XtQuantTraderCallback):
+    def on_disconnected(self):
+        """
+        连接状态回调
+        :return:
+        """
+        print("connection lost")
+    def on_stock_order(self, order):
+        """
+        委托信息推送
+        :param order: XtOrder对象
+        :return:
+        """
+        # 委托
+        if order.order_status == 50:
+            logger.info(f"{BLUE}【已委托】{RESET} {parse_order_type(order.order_type)} 代码:{order.stock_code} 名称:{order.order_remark} 委托价格:{order.price:.2f} 委托数量:{order.order_volume} 订单编号:{order.order_id} 委托时间:{timestamp_to_datetime_string(convert_to_current_date(order.order_time))}")
+        elif order.order_status == 53 or order.order_status == 54:
+            logger.warning(f"{YELLOW}【已撤单】{RESET} {parse_order_type(order.order_type)} 代码:{order.stock_code} 名称:{order.order_remark} 委托价格:{order.price:.2f} 委托数量:{order.order_volume} 订单编号:{order.order_id} 委托时间:{timestamp_to_datetime_string(convert_to_current_date(order.order_time))}")
+
+    def on_stock_trade(self, trade):
+        """
+        成交信息推送
+        :param trade: XtTrade对象
+        :return:
+        """
+        logger.info(f"{GREEN}【已成交】{RESET} {parse_order_type(trade.order_type)} 代码:{trade.stock_code} 名称:{trade.order_remark} 成交价格:{trade.traded_price:.2f} 成交数量:{trade.traded_volume} 成交编号:{trade.order_id} 成交时间:{timestamp_to_datetime_string(convert_to_current_date(trade.traded_time))}")
+
+    def on_order_error(self, data):
+        if data.order_id in error_orders:
+            return
+        error_orders.append(data.order_id)
+        logger.error(f"{RED}【委托失败】{RESET}错误信息:{data.error_msg.strip()}")
+
+    def on_cancel_error(self, data):
+        if data.order_id in error_orders:
+            return
+        error_orders.append(data.order_id)
+        logger.error(f"{RED}【撤单失败】{RESET}错误信息:{data.error_msg.strip()}")
+
+
+def create_trader(account_id, mini_qmt_path):
+    # 创建session_id
+    session_id = int(random.randint(100000, 999999))
+    # 创建交易对象
+    xt_trader = XtQuantTrader(mini_qmt_path, session_id)
+    # 启动交易对象
+    xt_trader.start()
+    # 连接客户端
+    connect_result = xt_trader.connect()
+
+    if connect_result == 0:
+        logger.debug(f"{GREEN}【miniQMT连接成功】{RESET} 路径:{mini_qmt_path}")
+
+    # 创建账号对象
+    account = StockAccount(account_id)
+    # 订阅账号
+    xt_trader.subscribe(account)
+    logger.debug(f"{GREEN}【账号订阅成功】{RESET} 账号ID:{account_id}")
+    # 注册回调类
+    xt_trader.register_callback(MyXtQuantTraderCallback())
+
+    return xt_trader, account

qka-0.1.0/qka/utils.py

@@ -0,0 +1,50 @@
+from xtquant import xtconstant
+from datetime import datetime
+from qka.anis import RED, GREEN, YELLOW, BLUE, RESET
+
+def add_stock_suffix(stock_code):
+    """
+    为给定的股票代码添加相应的后缀。
+    """
+    # 检查股票代码是否为6位数字
+    if len(stock_code) != 6 or not stock_code.isdigit():
+        raise ValueError("股票代码必须是6位数字")
+
+    # 根据股票代码的前缀添加相应的后缀
+    if stock_code.startswith("00") or stock_code.startswith("30") or stock_code.startswith("15") or stock_code.startswith("16") or stock_code.startswith("18") or stock_code.startswith("12"):
+        return f"{stock_code}.SZ"  # 深圳证券交易所
+    elif stock_code.startswith("60") or stock_code.startswith("68") or stock_code.startswith("11"):
+        return f"{stock_code}.SH"  # 上海证券交易所
+    elif stock_code.startswith("83") or stock_code.startswith("43"):
+        return f"{stock_code}.BJ"  # 北京证券交易所
+    
+    return f"{stock_code}.SH"
+
+def timestamp_to_datetime_string(timestamp):
+    """
+    将时间戳转换为时间字符串。
+
+    :param timestamp: 时间戳（秒级）
+    :return: 格式化的时间字符串 'YYYY-MM-DD HH:MM:SS'
+    """
+    dt_object = datetime.fromtimestamp(timestamp)
+    time_string = dt_object.strftime('%Y-%m-%d %H:%M:%S')
+    return time_string
+
+def parse_order_type(order_type):
+    if order_type == xtconstant.STOCK_BUY:
+        return f"{RED}买入{RESET}"
+    elif order_type == xtconstant.STOCK_SELL:
+        return f"{GREEN}卖出{RESET}"
+
+def convert_to_current_date(timestamp):
+    # 将时间戳转换为 datetime 对象
+    dt = datetime.fromtimestamp(timestamp)
+    
+    # 获取当前日期
+    current_date = datetime.now().date()
+    
+    # 创建一个新的 datetime 对象，使用当前日期和原始时间戳的时间部分
+    new_dt = datetime.combine(current_date, dt.time())
+
+    return new_dt.timestamp()

qka-0.1.0/test.py

@@ -0,0 +1,4 @@
+from qka.server import QMTServer
+
+server = QMTServer(account_id='55003152', mini_qmt_path=r'D:\国金QMT交易端模拟\userdata_mini')
+server.start()