forked from uncloud/uncloud
Fix payment creation
This commit is contained in:
parent
06ab21c577
commit
f358acca05
2 changed files with 7 additions and 5 deletions
|
@ -4,9 +4,9 @@ from .models import Bill, Payment
|
||||||
class BillSerializer(serializers.ModelSerializer):
|
class BillSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Bill
|
model = Bill
|
||||||
fields = ['user', 'amount']
|
fields = ['owner', 'amount']
|
||||||
|
|
||||||
class PaymentSerializer(serializers.ModelSerializer):
|
class PaymentSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Payment
|
model = Payment
|
||||||
fields = ['user', 'amount', 'source', 'timestamp']
|
fields = ['owner', 'amount', 'source', 'timestamp']
|
||||||
|
|
|
@ -1,8 +1,10 @@
|
||||||
from django.shortcuts import render
|
from django.shortcuts import render
|
||||||
from rest_framework import viewsets, permissions
|
from rest_framework import viewsets, permissions, status
|
||||||
|
from rest_framework.response import Response
|
||||||
|
|
||||||
from .models import Bill, Payment
|
from .models import Bill, Payment
|
||||||
from .serializers import BillSerializer, PaymentSerializer
|
from .serializers import BillSerializer, PaymentSerializer
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
|
||||||
# to be implemented
|
# to be implemented
|
||||||
|
@ -35,12 +37,12 @@ class PaymentViewSet(viewsets.ModelViewSet):
|
||||||
http_method_names = ['get', 'post']
|
http_method_names = ['get', 'post']
|
||||||
|
|
||||||
def get_queryset(self):
|
def get_queryset(self):
|
||||||
return Payment.objects.filter(user=self.request.user)
|
return Payment.objects.filter(owner=self.request.user)
|
||||||
|
|
||||||
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(user=request.user,timestamp=datetime.now())
|
serializer.save(owner=request.user,timestamp=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…
Add table
Reference in a new issue