class UserRegister(generics.CreateAPIView):
    model = User
    serializer_class = UserSerializer
    permission_classes = [
        permissions.AllowAny  # Or anon users can't register
    ]

class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ('id', 'username', 'first_name', 'last_name')

    def create(self, validated_data):
        # user = User(email=validated_data['email'], username=validated_data['username'])
        # user.set_password(validated_data['password'])
        # user.save()
        # return user
        return User.objects.create_user(username=validated_data['username'],
                                        email=validated_data['email'],
                                        password=validated_data['password'])