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
		Add a link
		
	
		Reference in a new issue