EasyXT/tools/create_user_block.py at main · lesslinxiu/EasyXT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
"""
通达信自选股转自定义板块工具
将通达信默认自选股转换为API可读取的自定义板块
"""

import sys
from pathlib import Path
from datetime import datetime

# 添加项目路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))
sys.path.insert(0, str(project_root / "easy_xt"))

from easy_xt.tdx_client import TdxClient


def create_custom_sector():
    """创建自定义板块"""
    print("="*70)
    print("  通达信自定义板块创建工具")
    print("="*70)
    print(f"  时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n")

    with TdxClient() as client:
        tq = client.tq

        # 步骤1: 创建新的自定义板块
        print("[步骤1] 创建自定义板块")
        print("-"*70)

        sector_name = "量化选股"  # 板块名称
        stocks = ['605168.SH', '000333.SZ', '600519.SH']  # 测试股票

        print(f"  板块名称: '{sector_name}'")
        print(f"  添加股票: {stocks}")

        try:
            result = tq.send_user_block(
                block_code=sector_name,
                stocks=stocks,
                show=True
            )

            print(f"\n  [OK] 创建成功")
            print(f"  返回结果: {result}")

        except Exception as e:
            print(f"  [ERROR] 创建失败: {e}")
            return

        # 步骤2: 验证板块是否创建成功
        print("\n[步骤2] 验证板块")
        print("-"*70)

        import time
        time.sleep(1)  # 等待一下

        # 尝试获取板块
        try:
            retrieved_stocks = tq.get_stock_list_in_sector(sector_name)

            if retrieved_stocks and len(retrieved_stocks) > 0:
                print(f"  [OK] 板块创建成功！")
                print(f"  板块名称: '{sector_name}'")
                print(f"  股票数量: {len(retrieved_stocks)}")
                print(f"  股票列表: {retrieved_stocks}")
            else:
                print(f"  [WARN] 板块创建后仍无法读取")

        except Exception as e:
            print(f"  [ERROR] 验证失败: {e}")

        # 步骤3: 检查用户板块列表
        print("\n[步骤3] 检查用户板块列表")
        print("-"*70)

        try:
            user_sectors = tq.get_user_sector()

            if user_sectors:
                print(f"  [OK] 找到 {len(user_sectors)} 个用户板块:")
                for i, sector in enumerate(user_sectors, 1):
                    print(f"    {i}. {sector}")

                if sector_name in user_sectors:
                    print(f"\n  [成功] '{sector_name}' 已在用户板块列表中")
            else:
                print(f"  [INFO] 用户板块列表仍为空")
                print(f"  [提示] 这可能需要重启通达信才能显示")

        except Exception as e:
            print(f"  [ERROR] {e}")

        print("\n" + "="*70)
        print("  使用说明")
        print("="*70)

        print(f"""
成功创建自定义板块: '{sector_name}'

在交易脚本中使用:

    from easy_xt.tdx_client import TdxClient

    with TdxClient() as client:
        stocks = client.get_sector_stocks('{sector_name}')
        print(f"获取到 {{len(stocks)}} 只股票")

添加更多股票到板块:

    # 方法1: 使用send_user_block覆盖（会替换所有股票）
    tq.send_user_block(
        block_code='{sector_name}',
        stocks=['605168.SH', '000333.SZ', '600519.SH', '000001.SZ'],
        show=True
    )

    # 方法2: 在通达信中手动添加
    # 1. 按F6打开自选股设置
    # 2. 找到'{sector_name}'板块
    # 3. 右键 -> 添加股票
        """)


if __name__ == "__main__":
    create_custom_sector()