304 lines
9.2 KiB
Python
304 lines
9.2 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
任务配置示例
|
||
展示如何配置不同类型的定时策略任务
|
||
"""
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 添加src目录到路径
|
||
current_dir = Path(__file__).parent
|
||
src_dir = current_dir.parent / "src"
|
||
sys.path.insert(0, str(src_dir))
|
||
|
||
from loguru import logger
|
||
from src.data.tushare_fetcher import TushareFetcher
|
||
from src.data.stock_pool_manager import StockPoolManager
|
||
from src.strategy.kline_pattern_strategy import KLinePatternStrategy
|
||
from src.execution.strategy_executor import StrategyExecutor
|
||
from src.execution.task_scheduler import TaskScheduler
|
||
from src.utils.notification import NotificationManager
|
||
from src.utils.config_loader import config_loader
|
||
|
||
|
||
def setup_common_tasks():
|
||
"""设置常见的策略任务配置"""
|
||
|
||
print("=" * 80)
|
||
print("📋 常见策略任务配置示例")
|
||
print("=" * 80)
|
||
|
||
# 初始化组件
|
||
fetcher = TushareFetcher()
|
||
pool_manager = StockPoolManager(fetcher)
|
||
notification_manager = NotificationManager(config_loader.get('notification', {}))
|
||
executor = StrategyExecutor(pool_manager, notification_manager)
|
||
scheduler = TaskScheduler()
|
||
|
||
# 策略配置
|
||
strategy_config = {
|
||
'min_entity_ratio': 0.55,
|
||
'final_yang_min_ratio': 0.40,
|
||
'max_turnover_ratio': 40.0,
|
||
'timeframes': ['daily'],
|
||
'pullback_tolerance': 0.02,
|
||
'monitor_days': 30,
|
||
'pullback_confirmation_days': 7
|
||
}
|
||
|
||
# 注册策略
|
||
kline_strategy = KLinePatternStrategy(
|
||
data_fetcher=fetcher,
|
||
notification_manager=notification_manager,
|
||
config=strategy_config
|
||
)
|
||
executor.register_strategy("kline_pattern", kline_strategy)
|
||
|
||
print("🎯 任务配置场景:")
|
||
print()
|
||
|
||
# 场景1: 开盘前扫描热门股票
|
||
print("📊 场景1: 开盘前热门股票扫描")
|
||
print("-" * 60)
|
||
|
||
task_1 = executor.create_task_function(
|
||
strategy_id="kline_pattern",
|
||
stock_pool_rule="tushare_hot",
|
||
stock_pool_params={"limit": 50},
|
||
max_stocks=30,
|
||
send_notification=True
|
||
)
|
||
|
||
scheduler.add_task(
|
||
task_id="pre_market_hot_scan",
|
||
name="开盘前热门股票K线扫描",
|
||
func=task_1,
|
||
schedule_rule="weekdays at 09:00",
|
||
enabled=False
|
||
)
|
||
|
||
print("✅ 配置完成:")
|
||
print(" 任务: 开盘前热门股票K线扫描")
|
||
print(" 时间: 每个工作日 09:00")
|
||
print(" 股票池: 同花顺热榜前50只")
|
||
print(" 分析: 最多30只股票")
|
||
print(" 通知: 启用")
|
||
print()
|
||
|
||
# 场景2: 午间龙头股扫描
|
||
print("🐲 场景2: 午间龙头股扫描")
|
||
print("-" * 60)
|
||
|
||
task_2 = executor.create_task_function(
|
||
strategy_id="kline_pattern",
|
||
stock_pool_rule="leading_stocks",
|
||
stock_pool_params={
|
||
"top_boards": 8,
|
||
"stocks_per_board": 3,
|
||
"min_score": 70.0
|
||
},
|
||
max_stocks=20,
|
||
send_notification=True
|
||
)
|
||
|
||
scheduler.add_task(
|
||
task_id="midday_leading_scan",
|
||
name="午间龙头股K线扫描",
|
||
func=task_2,
|
||
schedule_rule="weekdays at 12:30",
|
||
enabled=False
|
||
)
|
||
|
||
print("✅ 配置完成:")
|
||
print(" 任务: 午间龙头股K线扫描")
|
||
print(" 时间: 每个工作日 12:30")
|
||
print(" 股票池: 热门板块前8个，每板块前3只，评分>70")
|
||
print(" 分析: 最多20只龙头股")
|
||
print(" 通知: 启用")
|
||
print()
|
||
|
||
# 场景3: 收盘后综合扫描
|
||
print("🌅 场景3: 收盘后综合扫描")
|
||
print("-" * 60)
|
||
|
||
task_3 = executor.create_task_function(
|
||
strategy_id="kline_pattern",
|
||
stock_pool_rule="combined_hot",
|
||
stock_pool_params={
|
||
"limit_per_source": 30,
|
||
"final_limit": 50
|
||
},
|
||
max_stocks=40,
|
||
send_notification=True
|
||
)
|
||
|
||
scheduler.add_task(
|
||
task_id="after_market_comprehensive_scan",
|
||
name="收盘后综合热门股扫描",
|
||
func=task_3,
|
||
schedule_rule="weekdays at 15:30",
|
||
enabled=False
|
||
)
|
||
|
||
print("✅ 配置完成:")
|
||
print(" 任务: 收盘后综合热门股扫描")
|
||
print(" 时间: 每个工作日 15:30")
|
||
print(" 股票池: 合并热门(同花顺+东财)，各取30只，合并后50只")
|
||
print(" 分析: 最多40只股票")
|
||
print(" 通知: 启用")
|
||
print()
|
||
|
||
# 场景4: 高频监控
|
||
print("⚡ 场景4: 高频监控扫描")
|
||
print("-" * 60)
|
||
|
||
task_4 = executor.create_task_function(
|
||
strategy_id="kline_pattern",
|
||
stock_pool_rule="tushare_hot",
|
||
stock_pool_params={"limit": 20},
|
||
max_stocks=15,
|
||
send_notification=False # 高频不通知，避免打扰
|
||
)
|
||
|
||
scheduler.add_task(
|
||
task_id="high_freq_monitor",
|
||
name="高频热门股监控",
|
||
func=task_4,
|
||
schedule_rule="every 15 minutes",
|
||
enabled=False
|
||
)
|
||
|
||
print("✅ 配置完成:")
|
||
print(" 任务: 高频热门股监控")
|
||
print(" 时间: 每15分钟执行一次")
|
||
print(" 股票池: 同花顺热榜前20只")
|
||
print(" 分析: 最多15只股票")
|
||
print(" 通知: 关闭(避免频繁打扰)")
|
||
print()
|
||
|
||
# 场景5: 自定义股票池
|
||
print("🎯 场景5: 自定义股票池扫描")
|
||
print("-" * 60)
|
||
|
||
# 创建自定义股票池
|
||
custom_stocks = [
|
||
"000001.SZ", # 平安银行
|
||
"000002.SZ", # 万科A
|
||
"600000.SH", # 浦发银行
|
||
"600036.SH", # 招商银行
|
||
"000858.SZ", # 五粮液
|
||
"600519.SH", # 贵州茅台
|
||
"000725.SZ", # 京东方A
|
||
"002415.SZ" # 海康威视
|
||
]
|
||
|
||
pool_manager.create_custom_rule("my_watchlist", custom_stocks)
|
||
|
||
task_5 = executor.create_task_function(
|
||
strategy_id="kline_pattern",
|
||
stock_pool_rule="my_watchlist",
|
||
stock_pool_params={},
|
||
max_stocks=len(custom_stocks),
|
||
send_notification=True
|
||
)
|
||
|
||
scheduler.add_task(
|
||
task_id="custom_watchlist_scan",
|
||
name="自选股K线形态扫描",
|
||
func=task_5,
|
||
schedule_rule="daily at 21:00",
|
||
enabled=False
|
||
)
|
||
|
||
print("✅ 配置完成:")
|
||
print(" 任务: 自选股K线形态扫描")
|
||
print(" 时间: 每日 21:00")
|
||
print(f" 股票池: 自定义股票池({len(custom_stocks)}只)")
|
||
print(" 分析: 全部自选股")
|
||
print(" 通知: 启用")
|
||
print()
|
||
|
||
# 显示所有任务状态
|
||
print("📋 所有配置任务总览:")
|
||
print("-" * 60)
|
||
|
||
task_status = scheduler.get_task_status()
|
||
for task_id, status in task_status.items():
|
||
print(f"🔹 {status['name']}")
|
||
print(f" ID: {task_id}")
|
||
print(f" 规则: {status['schedule_rule']}")
|
||
print(f" 状态: {status['status']}")
|
||
print(f" 启用: {'是' if status['enabled'] else '否'}")
|
||
print()
|
||
|
||
# 使用指南
|
||
print("📖 使用指南:")
|
||
print("-" * 60)
|
||
print("1. 根据实际需求选择合适的任务配置")
|
||
print("2. 调整股票池参数和分析数量")
|
||
print("3. 设置合适的执行时间")
|
||
print("4. 启用需要的任务: scheduler.enable_task('task_id')")
|
||
print("5. 启动调度器: scheduler.start()")
|
||
print("6. 立即测试: scheduler.execute_task_now('task_id')")
|
||
print()
|
||
|
||
print("⚙️ 高级配置技巧:")
|
||
print("-" * 60)
|
||
print("• 开盘前(09:00): 扫描热门股，发现隔夜机会")
|
||
print("• 午间时段(12:30): 扫描龙头股，捕捉强势股")
|
||
print("• 收盘后(15:30): 综合扫描，总结全天机会")
|
||
print("• 高频监控(15分钟): 实时跟踪，但关闭通知")
|
||
print("• 晚间复盘(21:00): 扫描自选股，制定明日策略")
|
||
print()
|
||
|
||
return scheduler, executor
|
||
|
||
|
||
def demo_task_execution(scheduler, executor):
|
||
"""演示任务执行"""
|
||
|
||
print("🎬 任务执行演示:")
|
||
print("-" * 60)
|
||
|
||
# 立即执行一个任务进行测试
|
||
print("立即执行任务: pre_market_hot_scan")
|
||
success = scheduler.execute_task_now("pre_market_hot_scan")
|
||
print(f"执行结果: {'成功' if success else '失败'}")
|
||
print()
|
||
|
||
# 显示执行统计
|
||
task_status = scheduler.get_task_status()
|
||
for task_id, status in task_status.items():
|
||
if status['total_executions'] > 0:
|
||
print(f"📊 任务: {status['name']}")
|
||
print(f" 执行次数: {status['total_executions']}")
|
||
print(f" 成功率: {status['success_rate']:.1f}%")
|
||
if status['last_execution_time']:
|
||
print(f" 最后执行: {status['last_execution_time']}")
|
||
print()
|
||
|
||
print("💡 提示: 在生产环境中启用任务并启动调度器")
|
||
print(" scheduler.enable_task('task_id')")
|
||
print(" scheduler.start()")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 设置日志
|
||
logger.remove()
|
||
logger.add(sys.stdout, level="INFO", format="{time:HH:mm:ss} | {level} | {message}")
|
||
|
||
try:
|
||
scheduler, executor = setup_common_tasks()
|
||
demo_task_execution(scheduler, executor)
|
||
|
||
print("=" * 80)
|
||
print("🎉 任务配置示例演示完成!")
|
||
print("💼 可根据实际需求调整参数和时间规则")
|
||
print("🚀 启用任务并启动调度器即可自动运行")
|
||
print("=" * 80)
|
||
|
||
except Exception as e:
|
||
logger.error(f"演示过程中发生错误: {e}")
|
||
import traceback
|
||
traceback.print_exc() |