maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

Allow category filtering in get_flashed_messages to allow rending categories in separate html blocks

Browse source
This commit is contained in:
Steven Osborn 2011-10-16 19:44:37 -07:00 • committed by Ron DuPlain
parent 676b3a4c13
commit fa069f94de
2 changed files with 30 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
flask/helpers.py
View file

@ -264,7 +264,7 @@ def flash(message, category='message'):
session.setdefault('_flashes', []).append((category, message))
def get_flashed_messages(with_categories=False):
def get_flashed_messages(with_categories=False, category_filter=[]):
"""Pulls all flashed messages from the session and returns them.
Further calls in the same request to the function will return
the same messages. By default just the messages are returned,
@ -282,12 +282,18 @@ def get_flashed_messages(with_categories=False):
.. versionchanged:: 0.3
`with_categories` parameter added.
.. versionchanged: 0.9
`category_filter` parameter added.
:param with_categories: set to `True` to also receive categories.
:param category_filter: whitelist of categories to limit return values
"""
flashes = _request_ctx_stack.top.flashes
if flashes is None:
_request_ctx_stack.top.flashes = flashes = session.pop('_flashes') \
if '_flashes' in session else []
if category_filter:
flashes = filter(lambda f: f[0] in category_filter, flashes)
if not with_categories:
return [x[1] for x in flashes]
return flashes