"""任务三定时调度器"""
import sys
import signal
import time
from pathlib import Path
from datetime import datetime

# 添加项目根目录到Python路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

import schedule
from src3.main import run_once


class Task3Scheduler:
    """任务三定时调度器"""

    def __init__(self):
        self.running = True
        self.total_runs = 0
        self.success_count = 0
        self.fail_count = 0
        self.start_time = datetime.now()

        # 注册信号处理
        signal.signal(signal.SIGINT, self._signal_handler)
        signal.signal(signal.SIGTERM, self._signal_handler)

    def _signal_handler(self, signum, frame):
        """处理退出信号"""
        print("\n收到退出信号，等待当前任务完成...")
        self.running = False

    def _is_workday(self):
        """判断是否为工作日（周一到周五）"""
        return datetime.now().weekday() < 5

    def _job(self):
        """定时任务"""
        if not self._is_workday():
            print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] 今天是周末，跳过推送")
            return

        print(f"\n{'='*50}")
        print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] 开始执行过期单推送")
        print(f"{'='*50}")

        self.total_runs += 1

        try:
            run_once()
            self.success_count += 1
            print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] ✓ 执行成功")
        except Exception as e:
            self.fail_count += 1
            print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] ✗ 执行失败: {e}")
            import traceback
            traceback.print_exc()

    def start(self):
        """启动调度器"""
        from src3.config import Task3ConfigManager

        try:
            config = Task3ConfigManager()
            push_time = config.get_push_time()
        except Exception as e:
            print(f"✗ 配置加载失败: {e}")
            return

        print("="*50)
        print("任务三：TAPD过期单推送调度器")
        print("="*50)
        print(f"推送时间: 每个工作日 {push_time}")
        print(f"启动时间: {self.start_time.strftime('%Y-%m-%d %H:%M:%S')}")
        print("="*50)

        # 设置定时任务
        schedule.every().day.at(push_time).do(self._job)

        # 循环执行
        while self.running:
            schedule.run_pending()
            time.sleep(1)

        # 退出统计
        self._print_stats()

    def _print_stats(self):
        """打印运行统计"""
        print("\n" + "="*50)
        print("运行统计")
        print("="*50)
        print(f"启动时间: {self.start_time.strftime('%Y-%m-%d %H:%M:%S')}")
        print(f"结束时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
        print(f"总执行次数: {self.total_runs}")
        print(f"成功次数: {self.success_count}")
        print(f"失败次数: {self.fail_count}")
        print("="*50)


if __name__ == "__main__":
    scheduler = Task3Scheduler()
    scheduler.start()