自制小工具---QMT持仓导入通达信自定义版块

admin

本文将详细解析一个实用的量化交易工具——将QMT持仓数据导入通达信自定义板块的Python程序。该工具能够自动获取QMT交易系统中的持仓股票，并将其转换为通达信软件可识别的板块文件格式。

程序概述

这个Python脚本主要实现以下功能：

• 1. 通过图形界面获取用户配置信息
• 2. 连接QMT交易系统获取持仓数据
• 3. 将持仓股票转换为通达信格式
• 4. 生成通达信自定义板块文件
  

代码结构解析

• 1. 日志系统配置
  

def setup_logger():
    # 创建logs目录
    if not os.path.exists('logs'):
        os.makedirs('logs')
        
    # 配置日志文件名（使用日期）
    log_file = os.path.join('logs', f'qmt_import_{datetime.now().strftime("%Y%m%d")}.log')
   
    # 创建日志记录器
    logger = logging.getLogger('QMTImport')
    logger.setLevel(logging.INFO)
   
    # 创建文件处理器（限制单个文件大小为5MB，最多保留5个文件）
    file_handler = RotatingFileHandler(log_file, maxBytes=5*1024*1024, backupCount=5, encoding='utf-8')
    file_handler.setLevel(logging.INFO)

这段代码配置了一个完善的日志系统：
• 自动创建logs目录存储日志文件

• 按日期生成日志文件名

• 使用RotatingFileHandler实现日志轮转，防止单个日志文件过大

• 同时输出到文件和终端控制台

• 2. 图形用户界面
  

class ConfigGUI:
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("仓鼠智能交易系统--QMT持仓导入通达信")
        self.root.geometry("800x600")
        
        # 默认配置
        self.qmt_path = tk.StringVar(value=r"D:\通达信\国金QMT交易端模拟\userdata_mini")
        self.tdx_path = tk.StringVar(value=r"D:\通达信\自用版")

GUI类使用tkinter库创建，主要特点包括：
• 设置默认路径，方便用户直接使用

• 提供浏览按钮选择目录

• 包含日志显示区域，实时反馈操作结果

• 输入验证功能，确保必要参数已填写

• 3. QMT连接模块
  

def connect_qmt():
    try:
        # 生成会话ID
        session_id = int(time.time()) % 1000000
        
        # 创建交易对象
        xt_trader = XtQuantTrader(QMT_PATH, session_id)
        xt_trader.start()
        
        # 连接交易服务器
        if xt_trader.connect() != 0:
            log_message("错误: 连接QMT交易服务器失败")
            return None

连接QMT的核心步骤：

• 1. 生成唯一的会话ID
• 2. 创建XtQuantTrader交易对象
• 3. 启动交易连接
• 4. 订阅指定账户
• 5. 持仓数据获取
  

def get_positions(xt_trader):
    account = StockAccount(QMT_ACCOUNT)
    positions = xt_trader.query_stock_positions(account)
   
    # 获取股票名称和最新价格
    stock_info = get_stock_info(stock_codes)
   
    # 计算盈亏百分比
    profit_percent = (price / cost - 1) * 100

获取持仓数据后，程序还会：
• 查询股票的实时行情数据

• 计算每只股票的盈亏比例

• 格式化输出持仓明细

• 5. 通达信板块导出
  

def export_to_tdx(positions):
    # 转换代码格式为通达信格式
    if code.endswith('.SH'):
        tdx_code = '1' + code.split('.')[0]
    elif code.endswith('.SZ'):
        tdx_code = '0' + code.split('.')[0]
        
    # 写入板块文件
    with open(block_file, 'w', encoding='gbk') as f:
        f.write('\n'.join(stock_codes))

代码转换规则：
• 上海市场股票：前缀1+股票代码

• 深圳市场股票：前缀0+股票代码

• 使用GBK编码写入文件，确保通达信能正确识别

技术要点总结

• 1. 多线程安全：通过生成唯一会话ID确保多实例运行安全
• 2. 错误处理：完善的异常捕获和日志记录机制
• 3. 性能优化：批量获取股票行情数据，减少API调用次数
• 4. 兼容性：处理不同市场股票代码的转换规则
• 5. 用户体验：图形界面与日志反馈相结合
  

实际应用价值

该工具解决了量化交易中的一个常见需求：将QMT系统中的持仓股票快速导入到通达信软件中进行分析。主要优势包括：

• 1. 自动化：无需手动记录和输入股票代码
• 2. 实时性：获取最新持仓和行情数据
• 3. 可视化：直观显示盈亏情况
• 4. 灵活性：可自定义板块名称和存储路径
  

对于同时使用QMT和通达信的投资者，这个工具能显著提高工作效率，实现两个平台之间的数据无缝衔接。

1. #!/usr/bin/env python
   
2. # -*- coding: utf-8 -*-
   
3. 
   
4. from __future__ import print_function
   
5. import os
   
6. import sys
   
7. import json
   
8. import uuid
   
9. import shutil
   
10. import platform
    
11. import re
    
12. from datetime import datetime
    
13. import errno
    
14. 
    
15. def get_storage_path():
    
16.     """获取配置文件路径"""
    
17.     system = platform.system().lower()
    
18.     home = os.path.expanduser('~')
    
19.    
    
20.     if system == 'windows':
    
21.         return os.path.join(os.getenv('APPDATA'), 'Cursor', 'User', 'globalStorage', 'storage.json')
    
22.     elif system == 'darwin':  # macOS
    
23.         return os.path.join(home, 'Library', 'Application Support', 'Cursor', 'User', 'globalStorage', 'storage.json')
    
24.     else:  # Linux
    
25.         return os.path.join(home, '.config', 'Cursor', 'User', 'globalStorage', 'storage.json')
    
26. 
    
27. def get_main_js_path():
    
28.     """获取main.js文件路径"""
    
29.     system = platform.system().lower()
    
30.    
    
31.     if system == 'darwin':  # macOS
    
32.         return '/Applications/Cursor.app/Contents/Resources/app/out/main.js'
    
33.     elif system == 'windows':  # Windows
    
34.         user_profile = os.getenv('LOCALAPPDATA')  # 使用LOCALAPPDATA而不是USERPROFILE
    
35.         if not user_profile:
    
36.             return None
    
37.         return os.path.join(user_profile, 'Programs', 'cursor', 'resources', 'app', 'out', 'main.js')
    
38.     return None
    
39. 
    
40. def generate_random_id():
    
41.     """生成随机ID (64位十六进制)"""
    
42.     return uuid.uuid4().hex + uuid.uuid4().hex
    
43. 
    
44. def generate_uuid():
    
45.     """生成UUID"""
    
46.     return str(uuid.uuid4())
    
47. 
    
48. def backup_file(file_path):
    
49.     """创建文件备份"""
    
50.     if os.path.exists(file_path):
    
51.         backup_path = '{}.backup_{}'.format(
    
52.             file_path,
    
53.             datetime.now().strftime('%Y%m%d_%H%M%S')
    
54.         )
    
55.         shutil.copy2(file_path, backup_path)
    
56.         print('已创建备份文件:', backup_path)
    
57. 
    
58. def ensure_dir_exists(path):
    
59.     """确保目录存在（兼容 Python 2/3）"""
    
60.     if not os.path.exists(path):
    
61.         try:
    
62.             os.makedirs(path)
    
63.         except OSError as e:
    
64.             if e.errno != errno.EEXIST:
    
65.                 raise
    
66. 
    
67. def update_main_js(file_path):
    
68.     """更新main.js文件中的UUID生成方式"""
    
69.     if not os.path.exists(file_path):
    
70.         print('警告: main.js 文件不存在:', file_path)
    
71.         return False
    
72. 
    
73.     # 创建备份
    
74.     backup_file(file_path)
    
75. 
    
76.     try:
    
77.         # 读取文件内容
    
78.         with open(file_path, 'r') as f:
    
79.             content = f.read()
    
80. 
    
81.         system = platform.system().lower()
    
82.         if system == 'darwin':
    
83.             # macOS: 替换 ioreg 命令
    
84.             new_content = re.sub(
    
85.                 r'ioreg -rd1 -c IOPlatformExpertDevice',
    
86.                 'UUID=$(uuidgen | tr \'[:upper:]\' \'[:lower:]\');echo \"IOPlatformUUID = \"$UUID\";',
    
87.                 content
    
88.             )
    
89.         elif system == 'windows':
    
90.             # Windows: 替换 REG.exe 命令
    
91.             # 注意：这里使用三重引号来处理复杂的转义
    
92.             old_cmd = r'${v5[s$()]}\\REG.exe QUERY HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Cryptography /v MachineGuid'
    
93.             new_cmd = r'powershell -Command "[guid]::NewGuid().ToString().ToLower()"'
    
94.             new_content = content.replace(old_cmd, new_cmd)
    
95.         else:
    
96.             print('警告: 不支持的操作系统')
    
97.             return False
    
98. 
    
99.         # 写入修改后的内容
    
100.         with open(file_path, 'w') as f:
     
101.             f.write(new_content)
     
102. 
     
103.         # 验证修改
     
104.         success_marker = 'UUID=$(uuidgen | tr \'[:upper:]\' \'[:lower:]\');echo \"IOPlatformUUID = \"$UUID\";' if system == 'darwin' else 'powershell -Command "[guid]::NewGuid().ToString().ToLower()"'
     
105.         if success_marker in new_content:
     
106.             print('main.js 文件修改成功')
     
107.             return True
     
108.         else:
     
109.             print('警告: main.js 文件可能未被正确修改，请检查文件内容')
     
110.             print('你可以从备份文件恢复:', file_path + '.backup_*')
     
111.             return False
     
112. 
     
113.     except Exception as e:
     
114.         print('修改 main.js 时出错:', str(e))
     
115.         return False
     
116. 
     
117. def update_storage_file(file_path):
     
118.     """更新存储文件中的ID"""
     
119.     # 生成新的ID
     
120.     new_machine_id = generate_random_id()
     
121.     new_mac_machine_id = generate_random_id()
     
122.     new_dev_device_id = generate_uuid()
     
123.    
     
124.     # 确保目录存在
     
125.     ensure_dir_exists(os.path.dirname(file_path))
     
126.    
     
127.     # 读取或创建配置文件
     
128.     if os.path.exists(file_path):
     
129.         try:
     
130.             with open(file_path, 'r') as f:
     
131.                 data = json.load(f)
     
132.         except ValueError:
     
133.             data = {}
     
134.     else:
     
135.         data = {}
     
136.    
     
137.     # 更新ID
     
138.     data['telemetry.machineId'] = new_machine_id
     
139.     data['telemetry.macMachineId'] = new_mac_machine_id
     
140.     data['telemetry.devDeviceId'] = new_dev_device_id
     
141.     data['telemetry.sqmId'] = '{' + str(uuid.uuid4()).upper() + '}'
     
142.    
     
143.     # 写入文件
     
144.     with open(file_path, 'w') as f:
     
145.         json.dump(data, f, indent=4)
     
146.    
     
147.     return new_machine_id, new_mac_machine_id, new_dev_device_id
     
148. 
     
149. def main():
     
150.     """主函数"""
     
151.     try:
     
152.         # 获取配置文件路径
     
153.         storage_path = get_storage_path()
     
154.         print('配置文件路径:', storage_path)
     
155.         
     
156.         # 备份原文件
     
157.         backup_file(storage_path)
     
158.         
     
159.         # 更新ID
     
160.         machine_id, mac_machine_id, dev_device_id = update_storage_file(storage_path)
     
161.         
     
162.         # 输出结果
     
163.         print('\n已成功修改 ID:')
     
164.         print('machineId:', machine_id)
     
165.         print('macMachineId:', mac_machine_id)
     
166.         print('devDeviceId:', dev_device_id)
     
167. 
     
168.         # 处理 main.js
     
169.         system = platform.system().lower()
     
170.         if system in ['darwin', 'windows']:
     
171.             main_js_path = get_main_js_path()
     
172.             if main_js_path:
     
173.                 update_main_js(main_js_path)
     
174.         
     
175.     except Exception as e:
     
176.         print('错误:', str(e), file=sys.stderr)
     
177.         sys.exit(1)
     
178. 
     
179. if __name__ == '__main__':
     
180.     main()
     

复制代码

1

试读已结束，请付费阅读全文。

￥ 1.20付费阅读

登录后可特价购买

  本文只能试读95%,付费后可阅读全文。