Function lifespan

src/main.py:88–232

Application lifespan manager

(app: FastAPI)

@asynccontextmanager
async def lifespan(app: FastAPI):
    """Application lifespan manager"""
    # Startup
    print("=" * 60)
    print("Flow2API Starting...")
    print("=" * 60)

    # Get config from setting.toml
    config_dict = config.get_raw_config()

    # Check if database exists (determine if first startup)
    is_first_startup = not db.db_exists()

    # Initialize database tables structure
    await db.init_db()

    # Handle database initialization based on startup type
    if is_first_startup:
        print("First startup detected. Initializing database and configuration from setting.toml...")
        await db.init_config_from_toml(config_dict, is_first_startup=True)
        print("Database and configuration initialized successfully.")
    else:
        print("Existing database detected. Checking for missing tables and columns...")
        await db.check_and_migrate_db(config_dict)
        print("Database migration check completed.")

    # 启动时统一把数据库配置同步到内存，避免 personal/brower 相关运行时配置遗漏。
    await db.reload_config_to_memory()
    generation_handler.file_cache.set_timeout(config.cache_timeout)
    cache_cleanup_enabled = await generation_handler.file_cache.refresh_cleanup_task()
    captcha_config = await db.get_captcha_config()

    # 尽量在浏览器服务启动前就拿到 token 快照，后续并发管理和预热共用。
    tokens = await token_manager.get_all_tokens()

    # Initialize browser captcha service if needed
    browser_service = None
    if captcha_config.captcha_method == "personal":
        from .services.browser_captcha_personal import (
            BrowserCaptchaService,
            PERSONAL_POOL_MAX_TOTAL_RESIDENT_TABS,
            resolve_effective_browser_count,
            resolve_effective_personal_max_resident_tabs,
        )
        browser_service = await BrowserCaptchaService.get_instance(db)
        print("Browser captcha service initialized (nodriver mode)")

        warmup_limit = max(1, min(
            PERSONAL_POOL_MAX_TOTAL_RESIDENT_TABS,
            resolve_effective_browser_count(config.browser_count)
            * resolve_effective_personal_max_resident_tabs(config.personal_max_resident_tabs),
        ))
        warmup_project_ids = await token_manager.get_personal_warmup_project_ids(
            tokens=tokens,
            limit=warmup_limit,
        )

        warmed_slots = []