from rest_framework import viewsets, status from rest_framework.decorators import action from rest_framework.response import Response from userausfall.models import User, MissingUserAttribute, PasswordMismatch from userausfall.rest_api.serializers import UserSerializer, UserActivationSerializer class UserViewSet(viewsets.ModelViewSet): queryset = User.objects.all() serializer_class = UserSerializer @action(detail=True, methods=["post"]) def activate(self, request, pk=None): """Create the corresponding LDAP account.""" user: User = self.get_object() serializer = UserActivationSerializer(data=request.data) if serializer.is_valid(): try: user.create_ldap_account(serializer.validated_data["password"]) except (MissingUserAttribute, PasswordMismatch) as e: return Response({"message": str(e)}, status=status.HTTP_400_BAD_REQUEST) return Response(status=status.HTTP_204_NO_CONTENT) else: return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)