28 lines
603 B
Python
28 lines
603 B
Python
from django import template
|
|
from django.contrib.messages import constants as message_constants
|
|
|
|
|
|
MESSAGE_LEVEL_TYPES = {
|
|
message_constants.DEBUG: "info",
|
|
message_constants.INFO: "info",
|
|
message_constants.SUCCESS: "success",
|
|
message_constants.WARNING: "notice",
|
|
message_constants.ERROR: "error",
|
|
}
|
|
|
|
|
|
register = template.Library()
|
|
|
|
|
|
@register.filter
|
|
def pnotify_type(message):
|
|
try:
|
|
level = message.level
|
|
except AttributeError:
|
|
return 'info'
|
|
else:
|
|
try:
|
|
return MESSAGE_LEVEL_TYPES[level]
|
|
except KeyError:
|
|
return "info"
|