This commit is contained in:
Elijah Lucian
2021-07-15 15:13:43 -06:00
parent f0f4494553
commit 83fdd5b79a
5 changed files with 13 additions and 11 deletions

View File

@@ -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

View File

@@ -88,7 +88,7 @@ TEMPLATES = [
]
WSGI_APPLICATION = 'server.wsgi.application'
APPEND_SLASH = False
# Database
# https://docs.djangoproject.com/en/3.1/ref/settings/#databases