Misago/misago/users/serializers/ban.py

from django.utils.translation import ugettext as _

from rest_framework import serializers

from misago.core.utils import format_plaintext_for_html
from misago.users.models import BAN_IP, Ban


__all__ = [
    'BanMessageSerializer',
    'BanDetailsSerializer',
]


def serialize_message(message):
    if message:
        return {
            'plain': message,
            'html': format_plaintext_for_html(message)
        }
    else:
        return None


class BanMessageSerializer(serializers.ModelSerializer):
    message = serializers.SerializerMethodField()

    class Meta:
        model = Ban
        fields = ('message', 'expires_on')

    def get_message(self, obj):
        if obj.user_message:
            message = obj.user_message
        elif obj.check_type == BAN_IP:
            message = _("Your IP address is banned.")
        else:
            message = _("You are banned.")

        return serialize_message(message)


class BanDetailsSerializer(serializers.ModelSerializer):
    user_message = serializers.SerializerMethodField()
    staff_message = serializers.SerializerMethodField()

    class Meta:
        model = Ban
        fields = ('user_message', 'staff_message', 'expires_on')

    def get_user_message(self, obj):
        return serialize_message(obj.user_message)

    def get_staff_message(self, obj):
        return serialize_message(obj.staff_message)