Flash messages signals

If we send a signal when a template is rendered why not when a message
is flashed? One real world use case is in tests, this signal should make
flash messages expectation easier to implement.
This commit is contained in:
Paulo Poiati 2013-02-27 23:46:17 -03:00
parent bfeee75696
commit b2aae044ca
4 changed files with 32 additions and 1 deletions

View file

@ -35,6 +35,7 @@ except ImportError:
from jinja2 import FileSystemLoader
from .signals import message_flashed
from .globals import session, _request_ctx_stack, _app_ctx_stack, \
current_app, request
@ -361,6 +362,8 @@ def flash(message, category='message'):
flashes = session.get('_flashes', [])
flashes.append((category, message))
session['_flashes'] = flashes
message_flashed.send(current_app._get_current_object(),
message=message, category=category)
def get_flashed_messages(with_categories=False, category_filter=[]):