orbit-oss/flask
4
5
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Set the content length automatically before calling wrap_file

Browse source
This commit is contained in:
Armin Ronacher 2013-05-30 14:24:29 +01:00
parent ffd9296507
commit 47572c5b40
2 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
flask/helpers.py
View file

@ -516,6 +516,7 @@ def send_file(filename_or_fp, mimetype=None, as_attachment=False,
if file is None: if file is None:
file = open(filename, 'rb') file = open(filename, 'rb')
mtime = os.path.getmtime(filename) mtime = os.path.getmtime(filename)
headers['Content-Length'] = os.path.getsize(filename)
data = wrap_file(request.environ, file) data = wrap_file(request.environ, file)
rv = current_app.response_class(data, mimetype=mimetype, headers=headers, rv = current_app.response_class(data, mimetype=mimetype, headers=headers,

1
flask/testsuite/basic.py
View file

@ -786,6 +786,7 @@ class BasicFunctionalityTestCase(FlaskTestCase):
def test_static_files(self): def test_static_files(self):
app = flask.Flask(__name__) app = flask.Flask(__name__)
app.testing = True
rv = app.test_client().get('/static/index.html') rv = app.test_client().get('/static/index.html')
self.assert_equal(rv.status_code, 200) self.assert_equal(rv.status_code, 200)
self.assert_equal(rv.data.strip(), b'<h1>Hello World!</h1>') self.assert_equal(rv.data.strip(), b'<h1>Hello World!</h1>')