Add ADMIN endpoints for bills and payments
This commit is contained in:
parent
f358acca05
commit
b9b605f407
3 changed files with 36 additions and 7 deletions
|
@ -33,6 +33,8 @@ router.register(r'bill', payviews.BillViewSet, basename='bill')
|
||||||
router.register(r'payment', payviews.PaymentViewSet, basename='payment')
|
router.register(r'payment', payviews.PaymentViewSet, basename='payment')
|
||||||
|
|
||||||
# admin/staff urls
|
# admin/staff urls
|
||||||
|
router.register(r'admin/bill', payviews.AdminBillViewSet, basename='admin/bill')
|
||||||
|
router.register(r'admin/payment', payviews.AdminPaymentViewSet, basename='admin/payment')
|
||||||
router.register(r'admin/vmhost', vmviews.VMHostViewSet)
|
router.register(r'admin/vmhost', vmviews.VMHostViewSet)
|
||||||
router.register(r'admin/opennebula', oneviews.VMViewSet, basename='opennebula')
|
router.register(r'admin/opennebula', oneviews.VMViewSet, basename='opennebula')
|
||||||
router.register(r'admin/opennebula_raw', oneviews.RawVMViewSet)
|
router.register(r'admin/opennebula_raw', oneviews.RawVMViewSet)
|
||||||
|
|
|
@ -4,7 +4,7 @@ from .models import Bill, Payment
|
||||||
class BillSerializer(serializers.ModelSerializer):
|
class BillSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Bill
|
model = Bill
|
||||||
fields = ['owner', 'amount']
|
fields = ['owner', 'amount', 'due_date', 'creation_date', 'starting_date', 'ending_date', 'paid']
|
||||||
|
|
||||||
class PaymentSerializer(serializers.ModelSerializer):
|
class PaymentSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
|
|
|
@ -6,8 +6,9 @@ from .models import Bill, Payment
|
||||||
from .serializers import BillSerializer, PaymentSerializer
|
from .serializers import BillSerializer, PaymentSerializer
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
|
###
|
||||||
|
# Standard user views:
|
||||||
|
|
||||||
# to be implemented
|
|
||||||
class BalanceViewSet(viewsets.ViewSet):
|
class BalanceViewSet(viewsets.ViewSet):
|
||||||
# here we return a number
|
# here we return a number
|
||||||
# number = sum(payments) - sum(bills)
|
# number = sum(payments) - sum(bills)
|
||||||
|
@ -20,10 +21,9 @@ class BalanceViewSet(viewsets.ViewSet):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class BillViewSet(viewsets.ModelViewSet):
|
class BillViewSet(viewsets.ReadOnlyModelViewSet):
|
||||||
serializer_class = BillSerializer
|
serializer_class = BillSerializer
|
||||||
permission_classes = [permissions.IsAuthenticated]
|
permission_classes = [permissions.IsAuthenticated]
|
||||||
http_method_names = ['get']
|
|
||||||
|
|
||||||
def get_queryset(self):
|
def get_queryset(self):
|
||||||
return Bill.objects.filter(owner=self.request.user)
|
return Bill.objects.filter(owner=self.request.user)
|
||||||
|
@ -31,10 +31,19 @@ class BillViewSet(viewsets.ModelViewSet):
|
||||||
def unpaid(self, request):
|
def unpaid(self, request):
|
||||||
return Bill.objects.filter(owner=self.request.user, paid=False)
|
return Bill.objects.filter(owner=self.request.user, paid=False)
|
||||||
|
|
||||||
class PaymentViewSet(viewsets.ModelViewSet):
|
class PaymentViewSet(viewsets.ReadOnlyModelViewSet):
|
||||||
|
serializer_class = PaymentSerializer
|
||||||
|
permission_classes = [permissions.IsAuthenticated]
|
||||||
|
|
||||||
|
def get_queryset(self):
|
||||||
|
return Payment.objects.filter(owner=self.request.user)
|
||||||
|
|
||||||
|
###
|
||||||
|
# Admin views.
|
||||||
|
|
||||||
|
class AdminPaymentViewSet(viewsets.ModelViewSet):
|
||||||
serializer_class = PaymentSerializer
|
serializer_class = PaymentSerializer
|
||||||
permission_classes = [permissions.IsAuthenticated]
|
permission_classes = [permissions.IsAuthenticated]
|
||||||
http_method_names = ['get', 'post']
|
|
||||||
|
|
||||||
def get_queryset(self):
|
def get_queryset(self):
|
||||||
return Payment.objects.filter(owner=self.request.user)
|
return Payment.objects.filter(owner=self.request.user)
|
||||||
|
@ -42,7 +51,25 @@ class PaymentViewSet(viewsets.ModelViewSet):
|
||||||
def create(self, request):
|
def create(self, request):
|
||||||
serializer = self.get_serializer(data=request.data)
|
serializer = self.get_serializer(data=request.data)
|
||||||
serializer.is_valid(raise_exception=True)
|
serializer.is_valid(raise_exception=True)
|
||||||
serializer.save(owner=request.user,timestamp=datetime.now())
|
serializer.save(timestamp=datetime.now())
|
||||||
|
|
||||||
|
headers = self.get_success_headers(serializer.data)
|
||||||
|
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
||||||
|
|
||||||
|
class AdminBillViewSet(viewsets.ModelViewSet):
|
||||||
|
serializer_class = BillSerializer
|
||||||
|
permission_classes = [permissions.IsAuthenticated]
|
||||||
|
|
||||||
|
def get_queryset(self):
|
||||||
|
return Bill.objects.filter(owner=self.request.user)
|
||||||
|
|
||||||
|
def unpaid(self, request):
|
||||||
|
return Bill.objects.filter(owner=self.request.user, paid=False)
|
||||||
|
|
||||||
|
def create(self, request):
|
||||||
|
serializer = self.get_serializer(data=request.data)
|
||||||
|
serializer.is_valid(raise_exception=True)
|
||||||
|
serializer.save(created_at=datetime.now())
|
||||||
|
|
||||||
headers = self.get_success_headers(serializer.data)
|
headers = self.get_success_headers(serializer.data)
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
||||||
|
|
Loading…
Reference in a new issue