我正在使用Flask-Restless构建一个需要API密钥的API,它将位于Authorization
HTTP标头中。
在预处理器的Flask-Restless示例here中:
def check_auth(instance_id=None, **kw):
# Here, get the current user from the session.
current_user = ...
# Next, check if the user is authorized to modify the specified
# instance of the model.
if not is_authorized_to_modify(current_user, instance_id):
raise ProcessingException(message='Not Authorized',
status_code=401)
manager.create_api(Person, preprocessors=dict(GET_SINGLE=[check_auth]))
如何检索Authorization
函数中的check_auth
标题?
我尝试访问Flask response
对象,但在此函数范围内它是None
。 kw
参数也是一个空字典。
答案 0 :(得分:7)
在正常的Flask请求 - 响应周期中,当运行Flask-Restful预处理器和后处理器时,request
context处于活动状态。
因此,使用:
from flask import request, abort
def check_auth(instance_id=None, **kw):
current_user = None
auth = request.headers.get('Authorization', '').lower()
try:
type_, apikey = auth.split(None, 1)
if type_ != 'your_api_scheme':
# invalid Authorization scheme
ProcessingException(message='Not Authorized',
status_code=401)
current_user = user_for_apikey[apikey]
except (ValueError, KeyError):
# split failures or API key not valid
ProcessingException(message='Not Authorized',
status_code=401)
应该工作。