from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView


def with_user_jwt_factory(user_serializer_class):
    class WithUserTokenObtainPairSerializer(TokenObtainPairSerializer):
        @classmethod
        def get_token(cls, user):
            token = super().get_token(user)

            token['userdata'] = user_serializer_class(user).data

            return token

    class WithUserTokenObtainPairView(TokenObtainPairView):
        swagger_schema = None
        serializer_class = WithUserTokenObtainPairSerializer

    return WithUserTokenObtainPairView


class NoSwaggerTokenRefreshView(TokenRefreshView):
    swagger_schema = None