aiData/T3_YeLiTe.py

# coding=utf-8
import sys
import os
import asyncio
import logging

# 将项目根目录添加到 sys.path
project_root = os.path.dirname(os.path.abspath(__file__))
if project_root not in sys.path:
    sys.path.append(project_root)

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger("T3_YeLiTe")

try:
    from Apps.YeLiTe.Service import YiLaiTeService
    from Apps.YeLiTe import Opener, Crawler, Kit
except KeyboardInterrupt:
    logger.info("\n🛑 用户在初始化阶段手动停止了程序。")
    sys.exit(0)
except Exception as e:
    logger.error(f"❌ 初始化导入失败: {e}")
    sys.exit(1)

async def run_process():
    logger.info("=== 开始全流程任务 (驿来特): 打开小程序 -> 爬取数据 ===")

    # 步骤 0: 初始化基础服务
    logger.info(">>> 步骤 0: 初始化基础服务 (数据库连接)...")
    service = YiLaiTeService()
    await service.init_db()

    try:
        # 启动前清空临时目录
        Kit.clear_temp_dir()

        # 步骤 1: 启动小程序
        logger.info(">>> 步骤 1: 启动 驿来特 小程序...")
        success = await Opener.open_mini_program()
        if not success:
            logger.error("❌ 无法成功打开小程序，任务终止。")
            return

        logger.info("✅ 小程序启动成功，等待 5 秒确保界面稳定...")
        await asyncio.sleep(5)

        # 步骤 2: 执行爬取任务
        logger.info(">>> 步骤 2: 开始执行场站爬取任务...")
        # 调用 Crawler 的 main，并传入 service
        await Crawler.main(service=service)
            
        logger.info("✅ 爬取任务完成！")
    
    except Exception as e:
        logger.error(f"❌ 运行异常: {e}")
    finally:
        if service:
            await service.close_db()
        logger.info("=== 全流程任务结束 ===")

if __name__ == "__main__":
    try:
        asyncio.run(run_process())
    except KeyboardInterrupt:
        logger.info("\n🛑 用户手动停止了程序 (Ctrl+C)。")
    except Exception as e:
        logger.exception(f"主程序崩溃: {e}")