24 lines
721 B
Python
24 lines
721 B
Python
|
from django.contrib.auth import get_user_model
|
||
|
|
from rest_framework import viewsets
|
||
|
|
from rest_framework import status
|
||
|
|
from rest_framework.decorators import action
|
||
|
|
from rest_framework.response import Response
|
||
|
|
|
||
|
|
from orchestra.api import router, SetPasswordApiMixin
|
||
|
|
from orchestra.apps.accounts.api import AccountApiMixin
|
||
|
|
|
||
|
|
from .serializers import UserSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class UserViewSet(AccountApiMixin, SetPasswordApiMixin, viewsets.ModelViewSet):
|
||
|
|
model = get_user_model()
|
||
|
|
serializer_class = UserSerializer
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
""" select related roles """
|
||
|
|
qs = super(UserViewSet, self).get_queryset()
|
||
|
|
return qs.select_related(*self.inserted)
|
||
|
|
|
||
|
|
|
||
|
|
router.register(r'users', UserViewSet)
|