"""
量化交易系统 - 主程序入口
"""
from loguru import logger
import os
import sys


def initialize_system():
    """初始化系统"""
    logger.info("=" * 50)
    logger.info("  量化交易系统启动中...")
    logger.info("=" * 50)
    
    # 确保必要的目录存在
    os.makedirs("logs", exist_ok=True)
    os.makedirs("data", exist_ok=True)
    os.makedirs("strategies", exist_ok=True)
    os.makedirs("backup", exist_ok=True)
    
    logger.info("✅ 系统目录检查完成")
    
    # 测试数据库连接
    try:
        from test_database import test_database_connection
        if test_database_connection():
            logger.info("✅ 数据库连接正常")
        else:
            logger.warning("⚠️  数据库连接异常")
    except Exception as e:
        logger.error(f"❌ 数据库测试失败: {e}")
    
    logger.info("=" * 50)
    logger.info("  量化交易系统初始化完成！")
    logger.info("=" * 50)


def main():
    """主函数"""
    # 配置日志
    logger.add("logs/system_{time}.log", rotation="1 day", level="INFO")
    
    try:
        initialize_system()
        logger.info("🚀 系统准备就绪，等待交易指令...")
        
    except KeyboardInterrupt:
        logger.info("👋 收到中断信号，系统正在关闭...")
    except Exception as e:
        logger.error(f"❌ 系统错误: {e}")
        import traceback
        logger.error(traceback.format_exc())
        sys.exit(1)


if __name__ == "__main__":
    main()
