Prevent pop if flashes not in session to avoid modification to session fixes #227

This commit is contained in:
Sharoon Thomas 2011-05-17 11:41:12 -04:00
parent 8b974eb355
commit 57920a5808

View file

@ -249,7 +249,8 @@ def get_flashed_messages(with_categories=False):
"""
flashes = _request_ctx_stack.top.flashes
if flashes is None:
_request_ctx_stack.top.flashes = flashes = session.pop('_flashes', [])
_request_ctx_stack.top.flashes = flashes = session.pop('_flashes') \
if '_flashes' in session else []
if not with_categories:
return [x[1] for x in flashes]
return flashes