forked from uncloud/uncloud
Move user view to uncloud_pay
This commit is contained in:
parent
225f20c91b
commit
f5eadd6ddb
4 changed files with 29 additions and 2 deletions
|
|
@ -1,9 +1,11 @@
|
|||
from django.shortcuts import render
|
||||
from django.contrib.auth import get_user_model
|
||||
from rest_framework import viewsets, permissions, status
|
||||
from rest_framework.response import Response
|
||||
from rest_framework.decorators import action
|
||||
|
||||
from .models import Bill, Payment
|
||||
from .serializers import BillSerializer, PaymentSerializer
|
||||
from .serializers import BillSerializer, PaymentSerializer, UserSerializer
|
||||
from datetime import datetime
|
||||
|
||||
###
|
||||
|
|
@ -38,6 +40,17 @@ class PaymentViewSet(viewsets.ReadOnlyModelViewSet):
|
|||
def get_queryset(self):
|
||||
return Payment.objects.filter(owner=self.request.user)
|
||||
|
||||
class UserViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
serializer_class = UserSerializer
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
|
||||
def get_queryset(self):
|
||||
return get_user_model().objects.all()
|
||||
|
||||
@action(detail=True)
|
||||
def balance(self, request):
|
||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
||||
|
||||
###
|
||||
# Admin views.
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue