trading.ai/scripts/data_collector.py
2025-09-18 22:31:32 +08:00

56 lines
1.3 KiB
Python
Executable File

#!/usr/bin/env python3
"""
数据采集服务
定时运行策略扫描
"""
import time
import schedule
import sys
from pathlib import Path
# 添加项目根目录到路径
current_dir = Path(__file__).parent
project_root = current_dir.parent
sys.path.insert(0, str(project_root))
from src.strategy.kline_pattern_strategy import KLinePatternStrategy
from src.utils.config_loader import ConfigLoader
from loguru import logger
def run_strategy():
"""运行策略扫描"""
try:
logger.info('开始运行策略扫描...')
config = ConfigLoader()
strategy = KLinePatternStrategy(config)
# 这里可以添加具体的策略运行逻辑
# strategy.scan_all_stocks()
logger.info('策略扫描完成')
except Exception as e:
logger.error(f'策略运行失败: {e}')
def main():
"""主函数"""
logger.info('数据采集服务已启动')
# 每天9点和15点运行策略
schedule.every().day.at('09:00').do(run_strategy)
schedule.every().day.at('15:00').do(run_strategy)
# 立即运行一次(用于测试)
logger.info('立即执行一次策略扫描...')
run_strategy()
# 主循环
while True:
schedule.run_pending()
time.sleep(60)
if __name__ == '__main__':
main()