import schedule
import time
import subprocess
import sys
import os
from datetime import datetime
import calendar
from config import CONFIG
from logger_config import setup_logger

# 设置日志记录器
logger = setup_logger('scheduler')

def run_sendmsg():
    """执行sendmsg.py脚本，周末不执行"""
    # 获取当前日期是星期几（0是星期一，6是星期日）
    current_weekday = datetime.now().weekday()
    
    # 如果是周末（周六或周日），不执行任务
    if current_weekday >= 5:  # 5是周六，6是周日
        print(f"[{datetime.now()}] 今天是周末，不执行任务")
        return
    
    script_path = os.path.join(os.path.dirname(__file__), 'sendmsg.py')
    logger.info("=" * 50)
    logger.info("开始执行每日消息发送任务...")
    logger.info(f"脚本路径: {script_path}")
    logger.info(f"当前工作目录: {os.getcwd()}")
    
    try:
        logger.info("正在启动sendmsg.py子进程...")
        result = subprocess.run([sys.executable, script_path], 
                              capture_output=True, text=True, cwd=os.path.dirname(__file__))
        
        logger.info(f"子进程执行完成，返回码: {result.returncode}")
        
        if result.returncode == 0:
            logger.info("任务执行成功")
            if result.stdout:
                logger.info(f"标准输出: {result.stdout}")
        else:
            logger.error("任务执行失败")
            if result.stderr:
                logger.error(f"错误输出: {result.stderr}")
            if result.stdout:
                logger.info(f"标准输出: {result.stdout}")
                
    except Exception as e:
        logger.error(f"执行出错: {str(e)}")
        logger.error(f"错误类型: {type(e).__name__}")
        import traceback
        logger.error(f"异常堆栈: {traceback.format_exc()}")
    
    logger.info("=" * 50)

def main():
    """主函数 - 设置定时任务"""
    logger.info("=" * 50)
    logger.info("定时任务调度器启动")
    logger.info("=" * 50)
    
    # 显示配置信息
    logger.info(f"检查时间: {CONFIG['checking_time']}")
    logger.info(f"发送时间: {CONFIG['sending_time']}")
    logger.info(f"文件路径: {CONFIG['file_path']}")
    logger.info(f"消息接收者: {CONFIG['messages_reciever']}")
    logger.info(f"飞书接收者数量: {len(CONFIG['open_id'])}")
    
    # 每天在配置的时间执行任务
    schedule.every().day.at(CONFIG['checking_time']).do(run_sendmsg)
    
    logger.info(f"定时任务已设置，将在每天 {CONFIG['checking_time']} 执行sendmsg.py")
    logger.info("按Ctrl+C停止程序")
    
    # 显示下次执行时间
    next_run = schedule.next_run()
    if next_run:
        logger.info(f"下次执行时间: {next_run.strftime('%Y-%m-%d %H:%M:%S')}")
    
    try:
        while True:
            schedule.run_pending()
            time.sleep(60)  # 每分钟检查一次
            
            # 每分钟显示一次状态信息
            current_time = datetime.now()
            logger.debug(f"当前时间: {current_time.strftime('%Y-%m-%d %H:%M:%S')}, 等待下次执行...")
            
    except KeyboardInterrupt:
        logger.info(f"\n收到中断信号，正在停止定时任务...")
        logger.info(f"定时任务已停止")
    except Exception as e:
        logger.error(f"定时任务运行过程中出现异常: {str(e)}")
        logger.error(f"异常类型: {type(e).__name__}")
        import traceback
        logger.error(f"异常堆栈: {traceback.format_exc()}")
        raise

if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        logger.error(f"程序启动失败: {str(e)}")
        logger.error(f"异常类型: {type(e).__name__}")
        import traceback
        logger.error(f"异常堆栈: {traceback.format_exc()}")
        sys.exit(1)