Flask: 'Response'对象不可迭代，生产响应时出错

Question

我在使用Flask 0.10.1的时候，似乎无法从异常中生成响应了（在0.9版本时也出现过同样的问题）。这段代码：

from flask import Flask, jsonify
from werkzeug.exceptions import HTTPException
import flask, werkzeug

print 'Flask version: %s' % flask.__version__
print 'Werkzeug version: %s' % werkzeug.__version__

app = Flask(__name__)
app.config['PROPAGATE_EXCEPTIONS'] = True

class JSONException(HTTPException):
    response = None

    def get_body(self, environ):
        return jsonify(a=1)

    def get_headers(self, environ):
        return [('Content-Type', 'application/json')]

@app.route('/x')
def x():
    return jsonify(a=1)

@app.route('/y')
def y():
    raise JSONException()

c = app.test_client()
r = c.get('x')
print r.data
r = c.get('y')
print r.data

输出的是

Flask version: 0.10.1
Werkzeug version: 0.9.4
{
  "a": 1
}
Traceback (most recent call last):
  File "flask_error.py", line 33, in <module>
    print r.data
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 881, in get_data
    self._ensure_sequence()
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 938, in _ensure_sequence
    self.make_sequence()
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 953, in make_sequence
    self.response = list(self.iter_encoded())
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 81, in _iter_encoded
    for item in iterable:
  File "/home/path/lib/python2.7/site-packages/werkzeug/wsgi.py", line 682, in __next__
    return self._next()
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 81, in _iter_encoded
    for item in iterable:
  File "/home/path/lib/python2.7/site-packages/werkzeug/wsgi.py", line 682, in __next__
    return self._next()
  File "/home/path/lib/python2.7/site-packages/werkzeug/wrappers.py", line 81, in _iter_encoded
    for item in iterable:
TypeError: 'Response' object is not iterable

这个错误追踪信息让我感到意外。

Answer 1

jsonify() 生成的是一个 完整的响应对象，而不是单纯的响应内容，所以应该使用 HTTPException.get_response()，而不是 .get_body()：

class JSONException(HTTPException):    
    def get_response(self, environ):
        return jsonify(a=1)

另外一种方法是直接使用 json.dumps() 来生成响应内容：

class JSONException(HTTPException):
    def get_body(self, environ):
        return json.dumps({a: 1})

    def get_headers(self, environ):
        return [('Content-Type', 'application/json')]