Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-02-16 18:00:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

make IS_TEST not required, enhancement in Login view

Browse Source
This commit is contained in:
Mateusz Sikora 2014-05-30 11:17:25 +02:00
parent 7825890462
commit 44e0f33b25
2 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rest_auth/urls.py
View File

@ -27,13 +27,6 @@ urlpatterns = patterns('rest_auth.views',
name='rest_password_change'), name='rest_password_change'),
) )
if settings.DEBUG and not settings.IS_TEST: if getattr(settings, 'IS_TEST', False):
urlpatterns += patterns('',
# Swagger Docs
url(r'^docs/',
include('rest_framework_swagger.urls')),
)
if settings.IS_TEST:
from django.contrib.auth.tests import urls from django.contrib.auth.tests import urls
urlpatterns += patterns('', url(r'^test-admin/', include(urls))) urlpatterns += patterns('', url(r'^test-admin/', include(urls)))

6
rest_auth/views.py
View File

@ -68,6 +68,8 @@ class Login(LoggedOutRESTAPIView, GenericAPIView):
""" """
serializer_class = LoginSerializer serializer_class = LoginSerializer
token_model = Token
token_serializer = TokenSerializer
def post(self, request): def post(self, request):
# Create a serializer with request.DATA # Create a serializer with request.DATA
@ -86,8 +88,8 @@ class Login(LoggedOutRESTAPIView, GenericAPIView):
login(request, user) login(request, user)
# Return REST Token object with OK HTTP status # Return REST Token object with OK HTTP status
token, created = Token.objects.get_or_create(user=user) token, created = self.token_model.objects.get_or_create(user=user)
return Response(TokenSerializer(token).data, return Response(self.token_serializer(token).data,
status=status.HTTP_200_OK) status=status.HTTP_200_OK)
else: else:
return Response({'error': 'This account is disabled.'}, return Response({'error': 'This account is disabled.'},