Spaces:
Paused
Paused
File size: 1,144 Bytes
390dc9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
认证模块 - 提供API访问认证功能
"""
import os
import logging
import functools
from flask import request, jsonify
logger = logging.getLogger(__name__)
# 从环境变量获取API密钥
API_KEY = os.environ.get("API_KEY", "changeme")
def authenticate(func):
"""
用于API路由的认证装饰器
Args:
func: 被装饰的视图函数
Returns:
函数: 包含认证逻辑的包装函数
"""
@functools.wraps(func)
def wrapper(*args, **kwargs):
# 获取请求头中的API Key
api_key = request.headers.get("X-API-Key")
# 验证API Key
if not api_key or api_key != API_KEY:
logger.warning(f"API认证失败：{'未提供API Key' if not api_key else 'API Key无效'}")
return jsonify({
"success": False,
"error": "未提供API Key或API Key无效"
}), 401
# 认证通过，调用原始视图函数
return func(*args, **kwargs)
return wrapper |