@@ -32,7 +32,7 @@ def list(self, request):
32
32
username = self .get_user_id_from_token (request )
33
33
budget = Budget .objects .filter (username = username ).order_by ("-created_at" ).first ()
34
34
serializer = BudgetSerializer (budget )
35
- return Response (serializer .data )
35
+ return Response (data = { "amount" : serializer .data [ "amount" ]} )
36
36
except Budget .DoesNotExist :
37
37
return Response ({"error" : "Budget not found" }, status = status .HTTP_404_NOT_FOUND )
38
38
except Exception as e :
@@ -49,7 +49,7 @@ def create(self, request):
49
49
print (serializer )
50
50
if serializer .is_valid ():
51
51
serializer .save ()
52
- return Response (serializer .data , status = status .HTTP_201_CREATED )
52
+ return Response (data = { "amount" : serializer .data [ "amount" ]} , status = status .HTTP_201_CREATED )
53
53
return Response (serializer .errors , status = status .HTTP_400_BAD_REQUEST )
54
54
except Exception as e :
55
55
return Response ({"error" : str (e )}, status = status .HTTP_500_INTERNAL_SERVER_ERROR )
@@ -74,7 +74,7 @@ def partial_update(self, request):
74
74
serializer = BudgetSerializer (budget , data = request .data , partial = True )
75
75
if serializer .is_valid ():
76
76
serializer .save ()
77
- return Response (serializer .data )
77
+ return Response (data = { "amount" : serializer .data [ "amount" ]} )
78
78
return Response (serializer .errors , status = status .HTTP_400_BAD_REQUEST )
79
79
except Budget .DoesNotExist :
80
80
return Response ({"error" : "Budget not found" }, status = status .HTTP_404_NOT_FOUND )
0 commit comments