【发布时间】:2015-06-30 15:23:37
【问题描述】:
我正在使用 Pyramid 和 Cornice 编写一些 RESTful Python 应用程序,并制作了一个简单的 Cornice resource:
@resource(collection_path='/users/', path='/users/{id}')
class UsersResource(object):
def __init__(self, request):
self.request = request
@view(renderer='json', content_type=content_type)
@my_wrapper
def get(self):
return {'user_id': self.request.matchdict['id']}
您可能已经注意到,除了 Cornice 的 view 装饰器,我还在这里添加了一个额外的装饰器 (my_decorator),我打算将其用作包装器,以便在响应中添加一些额外的信息:
def my_wrapper(method):
def wrapper(*args, **kw):
time_start = time()
profiler = sqltap.start()
fn_result = method(*args, **kw)
stats = profiler.collect()
time_end = time()
result = {
'info': {
'api_version': args[0].request.registry.settings.api_version,
'request_path': args[0].request.path_info,
'request_method': args[0].request.method,
'current_time': datetime.datetime.now(pytz.utc).strftime('%Y-%m-%dT%H:%M:%SZ'),
'execution_time': time_end - time_start,
'total_queries': len(stats),
'query_time': stats[0].duration if len(stats) > 0 else 0,
},
}
result.update(fn_result)
return result
return wrapper
除非我在我的 view 装饰器中定义 Cornice validators,否则这很好用:
from validators import validate_int
@resource(collection_path='/users/', path='/users/{id}')
class UsersResource(object):
def __init__(self, request):
self.request = request
@view(renderer='json', content_type=content_type, validators=validate_int) # added validators
@my_wrapper
def get(self):
return {'user_id': self.request.matchdict['id']}
验证器.py
import responses
def validate_int(request):
should_be_int = request.matchdict['id']
try:
int(should_be_int)
except:
raise responses._400('This doesn\'t look like a valid ID.')
responses.py
class _400(exc.HTTPError):
def __init__(self, desc):
body = {'status': 400, 'message': 'Bad Request', 'description': desc}
Response.__init__(self, json.dumps(body))
使用这样的代码,my_wrapper 仅在验证通过时才包装响应(这是完全可以理解的),但我想知道当默认的 HTTPException 引发时,我如何仍然用一些额外的信息包装响应(因为在那种情况下,代码根本不会到达my_wrapper)?
【问题讨论】:
标签: python rest pyramid httpexception cornice