Function lifespan

api_utils/app.py:266–355

FastAPI application lifecycle management

(app: FastAPI)

@asynccontextmanager
async def lifespan(app: FastAPI):
    """FastAPI application lifecycle management"""
    from .queue_worker import queue_worker

    original_streams = sys.stdout, sys.stderr
    initial_stdout, initial_stderr = _setup_logging()
    logger = state.logger

    _initialize_globals()
    _initialize_proxy_settings()
    load_excluded_models(EXCLUDED_MODELS_FILENAME)

    state.is_initializing = True
    startup_start_time = time.time()
    logger.info("Starting AI Studio Proxy Server...")

    try:
        await _start_stream_proxy()
        await _initialize_browser_and_page()

        launch_mode = get_environment_variable("LAUNCH_MODE", "unknown")
        if state.is_page_ready or launch_mode == "direct_debug_no_browser":
            state.worker_task = asyncio.create_task(queue_worker())
            logger.info("Request processing worker started.")
        else:
            raise RuntimeError("Failed to initialize browser/page, worker not started.")

        logger.info("[WATCHDOG] Starting Quota Watchdog Task...")
        watchdog_func = state.quota_watchdog
        if watchdog_func:
            app.state.watchdog_task = asyncio.create_task(watchdog_func())
        else:
            logger.warning(
                "[WATCHDOG] Quota Watchdog function not found, task not started."
            )

        # Start periodic cookie refresh task
        try:
            from browser_utils.cookie_refresh import start_periodic_refresh

            cookie_refresh_task = start_periodic_refresh()
            if cookie_refresh_task:
                app.state.cookie_refresh_task = cookie_refresh_task
        except Exception as e:
            logger.warning(f"[COOKIE-REFRESH] Failed to start periodic refresh: {e}")

        startup_duration = time.time() - startup_start_time
        logger.info(f"Server startup complete. (Took: {startup_duration:.2f}s)")
        state.is_initializing = False
        yield
    except Exception as e:
        logger.critical(f"Application startup failed: {e}", exc_info=True)
        await _shutdown_resources()
        raise RuntimeError(f"Application startup failed: {e}") from e
    finally:
        logger.info("Shutting down server...")

        # Stop periodic cookie refresh and save cookies before shutdown