views.py 文件源码

def post(self, request, *args, **kwargs):
        """
        Returns a token and user_id, for credentials provided.
        ---
        response_serializer: employees.serializers.EmployeeAuthenticationResponse
        responseMessages:
        - code: 404
          message: Not found
        - code: 500
          message: Unable to log in with provided credentials.
        parameters:
        - name: username
          required: true
          paramType: string
        - name: password
          required: true
          :paramType: string
        """
        try:
            response = super(CustomObtainAuthToken, self).post(request, *args, **kwargs)
            token = Token.objects.get(key=response.data['token'])
            employee = get_object_or_404(Employee, pk=token.user_id)
            return Response({'token': token.key,
                             'user_id': token.user_id,
                             'reset_password_code': employee.reset_password_code,
                             'is_base_profile_complete': employee.is_base_profile_complete,
                             'is_password_reset_required': employee.is_password_reset_required,
                             'is_staff': employee.is_staff})
        except Exception as e:
            print(e)
            raise NotAcceptable(config.USER_UNABLE_TO_LOG)