jumpserver/apps/authentication/serializers/password_mfa.py

# -*- coding: utf-8 -*-
#
from django.utils.translation import ugettext_lazy as _
from rest_framework import serializers

from common.drf.fields import EncryptedField

__all__ = [
    'MFAChallengeSerializer', 'MFASelectTypeSerializer',
    'PasswordVerifySerializer', 'ResetPasswordCodeSerializer',
]


class ResetPasswordCodeSerializer(serializers.Serializer):
    form_type = serializers.ChoiceField(
        choices=[('sms', _('SMS')), ('email', _('Email'))], default='email'
    )
    email = serializers.CharField(allow_blank=True)
    sms = serializers.CharField(allow_blank=True)

    def create(self, attrs):
        error = []
        validate_backends = {
            'email': _('Email'), 'sms': _('SMS')
        }
        form_type = attrs.get('form_type', 'email')
        validate_backend_input = attrs.get(form_type)
        if not validate_backend_input:
            error.append(_('The {} cannot be empty').format(
                validate_backends.get(validate_backend_input))
            )
        if error:
            raise serializers.ValidationError(error)


class PasswordVerifySerializer(serializers.Serializer):
    password = EncryptedField()


class MFASelectTypeSerializer(serializers.Serializer):
    type = serializers.CharField()
    username = serializers.CharField(required=False, allow_blank=True, allow_null=True)


class MFAChallengeSerializer(serializers.Serializer):
    type = serializers.CharField(write_only=True, required=False, allow_blank=True)
    code = serializers.CharField(write_only=True)

    def create(self, validated_data):
        pass

    def update(self, instance, validated_data):
        pass