Flask应用配置注入后，如何在@authorize装饰器中注入认证URL？

嘿，我刚接触Flask Injector的时候也被这个问题卡过——它确实对类的注入支持得很好，但装饰器这种函数式的东西怎么搞依赖注入呢？其实咱们可以把验证逻辑封装成服务类，再让装饰器从注入器里拿这个服务实例，完美解决硬编码的问题！

步骤1：把验证逻辑封装成可注入的服务类 #

首先，咱们把get_token_info的逻辑移到一个类里，这样就能利用Flask Injector给它注入配置的auth URL了。比如在core目录下创建auth_service.py：

from flask import request, abort
import requests

class AuthService:
    def __init__(self, auth_url):
        # 构造函数注入auth_url，完全不用硬编码
        self.auth_url = auth_url

    def validate_token(self):
        session = requests.Session()
        # 先处理Authorization头缺失的情况，更健壮
        try:
            auth_header = request.headers['Authorization']
        except KeyError:
            abort(401, description="Authorization header is required")
        
        res = session.get(self.auth_url, headers={'Authorization': auth_header})
        if res.status_code != requests.codes.ok:
            abort(401, description="Invalid or expired authentication token")

步骤2：修改装饰器，从注入器获取服务实例 #

接下来，修改api/auth.py里的@authorize装饰器，让它从Flask应用的Injector实例中拿到AuthService：

from flask import current_app
from core.auth_service import AuthService

def authorize(func):
    def decorator(*args, **kwargs):
        # 从当前应用的Injector中获取AuthService实例
        injector = current_app.injector
        auth_service = injector.get(AuthService)
        # 调用验证方法
        auth_service.validate_token()
        return func(*args, **kwargs)
    return decorator

这里current_app.injector是Flask Injector初始化后绑定到应用上的注入器实例，只有在请求上下文里才可用——而装饰器正好是在处理请求时执行的，所以完全没问题。

步骤3：配置Flask Injector，完成依赖绑定 #

现在在__init__.py里，咱们要把AuthService和配置里的auth URL绑定起来，让注入器知道怎么创建AuthService实例：

from flask import Flask
from flask_injector import FlaskInjector
import json
from core.auth_service import AuthService

def create_app():
    app = Flask(__name__)
    
    # 加载config.json里的配置
    with open('api/config/config.json', 'r') as f:
        config = json.load(f)
    app.config.update(config)

    # 配置依赖注入规则
    def configure(binder):
        # 把auth_url从配置里取出来，绑定给AuthService的构造函数
        binder.bind(AuthService, to=AuthService(app.config['AUTH_URL']))
        # 如果需要更规范的构造注入（比如多个字符串依赖），可以用限定符：
        # from injector import qualifier
        # @qualifier
        # class AuthUrl: pass
        # binder.bind(str, to=app.config['AUTH_URL'], qualifier=AuthUrl)
        # binder.bind(AuthService, to=lambda i: AuthService(i.get(str, qualifier=AuthUrl)))

    # 初始化Flask Injector
    FlaskInjector(app=app, modules=[configure])

    # 注册路由蓝图
    from api.namespace.routes import bp
    app.register_blueprint(bp)

    return app

别忘了在api/config/config.json里加上你的auth URL配置：

{
  "AUTH_URL": "your_actual_auth_url_here"
}

为什么这么做？ #

• 把验证逻辑封装成服务类，符合依赖注入的思想，后续修改验证逻辑或者替换Auth服务时，只需要改动这个类就行，不用碰装饰器和路由。
• 完全避免了硬编码URL，配置集中管理，不同环境（开发/测试/生产）可以用不同的config.json。
• 方便测试：写单元测试时，可以给注入器绑定一个模拟的AuthService，不用真的调用外部Auth接口。

这样改造后，你在路由上用@authorize装饰器时，就自动使用注入的Auth URL做验证啦！

内容的提问来源于stack exchange，提问作者Tyrannas