{% for category, message in get_flashed_messages(with_categories=true) %} {% set alert_class = 'alert-info' %} {% if category == 'success' %} {% set alert_class = 'alert-success' %} {% elif category == 'error' %} {% set alert_class = 'alert-danger' %} {% endif %} {% endfor %}