import schedule
import time
from update_lottery import LotteryUpdater
import logging

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s',
    filename='lottery_update.log'
)
logger = logging.getLogger(__name__)


def update_job():
    """定时更新任务"""
    try:
        logger.info("开始执行定时更新任务...")
        updater = LotteryUpdater()
        updater.update_all_lottery_data()
        logger.info("定时更新任务完成")
    except Exception as e:
        logger.error(f"定时更新任务失败: {str(e)}")


def main():
    # 设置定时任务
    # 每天凌晨2点执行更新
    schedule.every().day.at("02:00").do(update_job)

    logger.info("定时更新服务已启动")

    while True:
        schedule.run_pending()
        time.sleep(60)


if __name__ == "__main__":
    main()