Day 17 | 依赖注入与中间件

#!/usr/bin/env python3
"""
Day 17 - 依赖注入示例
演示如何创建可复用的依赖函数
"""

from fastapi import Header, HTTPException
from typing import Optional

# ========== 依赖函数：验证 Token ==========
# 这是一个普通的异步函数
# 但它的参数 x_token 声明了来源于 Header，这很关键！
# Header(...) 表示这个参数必须从 HTTP Header 中获取
async def verify_token(x_token: str = Header(...)):
    """
    验证请求 Token
    
    参数:
        x_token: 从 Header 中获取的 Token（必须提供）
    
    返回:
        str: 验证通过的 Token
    
    抛出:
        HTTPException: Token 无效时抛出 400 错误
    """
    # 简单的 Token 验证逻辑（实际应该查询数据库或验证 JWT）
    if x_token != "fake-secret":
        # 如果 Token 对不上，直接抛出 400 错误
        # 这个异常会被 FastAPI 捕获，直接中断请求
        # 路由函数不会被执行
        raise HTTPException(
            status_code=400, 
            detail="Invalid X-Token header"
        )
    
    # 如果验证通过，返回 Token（虽然下面的路由可能不想用它）
    # 主要是为了通过验证，让路由函数能够执行
    return x_token

# ========== 可选依赖：获取当前用户 ==========
# 这个依赖依赖于 verify_token，形成依赖链
async def get_current_user(x_token: str = Header(...)):
    """
    获取当前用户信息（依赖于 Token 验证）
    
    参数:
        x_token: 从 Header 中获取的 Token
    
    返回:
        dict: 用户信息
    """
    # 先验证 Token
    if x_token != "fake-secret":
        raise HTTPException(status_code=400, detail="Invalid token")
    
    # 模拟根据 Token 查询用户信息
    # 实际应该从数据库或缓存中查询
    return {
        "id": 1,
        "username": "admin",
        "email": "admin@example.com"
    }

#!/usr/bin/env python3
"""
Day 17 - 依赖注入与中间件实战
演示如何使用 Depends 和 Middleware
"""

import time
from fastapi import FastAPI, Depends, Request
from fastapi.middleware.cors import CORSMiddleware
from 17_deps import verify_token, get_current_user

# 初始化 FastAPI 应用
app = FastAPI(
    title="Dependency Injection & Middleware Demo",
    description="演示依赖注入和中间件的使用",
    version="1.0.0"
)

# ========== 1. 全局中间件：配置 CORS (跨域) ==========
# 这是 Web 开发必配的，否则前端 Vue/React 请求会被浏览器拦截
# CORS (Cross-Origin Resource Sharing) 允许不同域名的前端访问 API
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 允许任何来源（生产环境应该指定具体域名！）
    allow_credentials=True,  # 允许携带 Cookie
    allow_methods=["*"],  # 允许所有 HTTP 方法
    allow_headers=["*"],  # 允许所有 Header
)

# ========== 2. 自定义中间件：统计耗时 ==========
# @app.middleware("http") 表示这是一个 HTTP 中间件
# 中间件会在每个请求前后执行
@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
    """
    统计请求处理时间并添加到 Response Header
    
    参数:
        request: 进来的请求对象
        call_next: 把请求交给下一个环节处理的函数
    
    返回:
        Response: 添加了耗时信息的响应
    """
    # 1. 记下开始时间
    start_time = time.time()
    
    # 2. 等待业务逻辑处理完（这里可能要很久）
    # call_next(request) 会调用路由函数，并返回响应
    response = await call_next(request)
    
    # 3. 算出耗时（秒）
    process_time = time.time() - start_time
    
    # 4. 把耗时悄悄塞到 Response Header 里返回给前端
    # 前端可以通过 Response Headers 查看处理时间
    response.headers["X-Process-Time"] = str(process_time)
    
    return response

# ========== 3. 路由实战 ==========

# 场景 A: 公开接口（不需要 Token）
@app.get("/")
async def public_api():
    """
    公开接口：任何人都可以访问
    """
    return {"msg": "Everyone can see me"}

# 场景 B: 保护接口（需要 Token）
# dependencies=[Depends(verify_token)]: 
# 在执行 read_items 前，FastAPI 会先去跑 verify_token 函数
# 如果 verify_token 报错，read_items 根本不会执行
@app.get("/items/", dependencies=[Depends(verify_token)])
async def read_items():
    """
    保护接口：需要 Token 才能访问
    
    依赖:
        verify_token: 验证请求 Header 中的 x-token
    """
    return [
        {"item": "Secret Data A"},
        {"item": "Secret Data B"}
    ]

# 场景 C: 使用依赖的返回值
# 如果依赖函数返回了值，可以通过参数接收
@app.get("/me")
async def get_current_user_info(current_user: dict = Depends(get_current_user)):
    """
    获取当前用户信息
    
    依赖:
        get_current_user: 验证 Token 并返回用户信息
    
    参数:
        current_user: 从依赖函数获取的用户信息
    """
    return {
        "message": "Your profile",
        "user": current_user
    }

# 场景 D: 多个依赖
# 可以同时使用多个依赖
@app.get("/admin/", dependencies=[Depends(verify_token), Depends(get_current_user)])
async def admin_only():
    """
    管理员接口：需要 Token 和用户信息
    
    依赖:
        verify_token: 验证 Token
        get_current_user: 获取用户信息
    """
    return {"message": "Admin area", "status": "restricted"}