Custom Django filter to convert message levels to valid blueprint css

blueprintcss.py

# -*- coding: utf-8 -*-

from django import template
from django.contrib import messages

register = template.Library()

LEVEL_TO_CSS = {
    messages.ERROR: "error",
    messages.WARNING: "notice",
    messages.SUCCESS: "success",
    messages.INFO: "info",
}


@register.filter(name='level2css')
def level2css(message):
    return LEVEL_TO_CSS.get(message.level, "info")