28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
from rest_framework import generics, status
|
|
from rest_framework.response import Response
|
|
|
|
from rest_framework_simplejwt.tokens import RefreshToken
|
|
|
|
from core.apps.accounts.models.user import User
|
|
from core.apps.accounts.serializers.login import LoginSerializer
|
|
from core.apps.accounts.permissions.permissions import HasRolePermission
|
|
|
|
|
|
class LoginApiView(generics.GenericAPIView):
|
|
serializer_class = LoginSerializer
|
|
queryset = User.objects.all()
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.data)
|
|
if serializer.is_valid(raise_exception=True):
|
|
user = serializer.validated_data.get('user')
|
|
token = RefreshToken.for_user(user)
|
|
user_data = {
|
|
'role': user.role.name,
|
|
'permissions': user.role.permissions.values_list('code', flat=True),
|
|
}
|
|
return Response(
|
|
{"access": str(token.access_token), "refresh": str(token), 'user_data': user_data},
|
|
status=status.HTTP_200_OK
|
|
)
|