|
|
|
@ -1,8 +1,8 @@ |
|
|
|
|
from django.contrib.auth.models import User, Group |
|
|
|
|
from rest_framework import viewsets |
|
|
|
|
from rest_framework import permissions |
|
|
|
|
from rest_framework import viewsets, permissions |
|
|
|
|
from server.api import serializers, models |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserViewSet(viewsets.ModelViewSet): |
|
|
|
|
queryset = User.objects.all() |
|
|
|
|
serializer_class = serializers.UserSerializer |
|
|
|
@ -11,6 +11,7 @@ class UserViewSet(viewsets.ModelViewSet): |
|
|
|
|
def get_queryset(self): |
|
|
|
|
return [self.request.user] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AccountViewSet(viewsets.ModelViewSet): |
|
|
|
|
queryset = models.Account.objects.all() |
|
|
|
|
serializer_class = serializers.AccountSerializer |
|
|
|
@ -19,6 +20,7 @@ class AccountViewSet(viewsets.ModelViewSet): |
|
|
|
|
def get_queryset(self): |
|
|
|
|
return self.queryset.filter(users=self.request.user) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class StackViewSet(viewsets.ModelViewSet): |
|
|
|
|
queryset = models.Stack.objects.all() |
|
|
|
|
serializer_class = serializers.StackSerializer |
|
|
|
@ -27,6 +29,7 @@ class StackViewSet(viewsets.ModelViewSet): |
|
|
|
|
def get_queryset(self): |
|
|
|
|
return self.queryset.filter(account__users=self.request.user) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TransactionViewSet(viewsets.ModelViewSet): |
|
|
|
|
queryset = models.Transaction.objects.all() |
|
|
|
|
serializer_class = serializers.TransactionSerializer |
|
|
|
|