Remove remaining direct references to the User model. Closes GitLab issue #225.

This commit is contained in:
Roberto Rosario
2016-01-28 05:55:15 -04:00
parent 27d8644438
commit 2ebb84f18f
15 changed files with 36 additions and 34 deletions

View File

@@ -1,6 +1,7 @@
from __future__ import unicode_literals
from django.contrib.auth.models import Group, User
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group
from rest_framework import generics
@@ -83,7 +84,7 @@ class APIUserListView(generics.ListCreateAPIView):
mayan_object_permissions = {'GET': (permission_user_view,)}
mayan_view_permissions = {'POST': (permission_user_create,)}
permission_classes = (MayanPermission,)
queryset = User.objects.all()
queryset = get_user_model().objects.all()
serializer_class = UserSerializer
def get(self, *args, **kwargs):
@@ -108,7 +109,7 @@ class APIUserView(generics.RetrieveUpdateDestroyAPIView):
'DELETE': (permission_user_delete,)
}
permission_classes = (MayanPermission,)
queryset = User.objects.all()
queryset = get_user_model().objects.all()
serializer_class = UserSerializer
def delete(self, *args, **kwargs):