Merge pull request #40 from Kalgoc/develop

Hotfix: Release

Author: DaHunterTime

budget/views.py

@@ -12,9 +12,12 @@ def list(self, request):
         try:
             user_id = get_user_id_from_token(request)
             budget = Budget.objects.filter(username=user_id).order_by("-created_at").first()
+
             if not budget:
-                return Response({"error": "Budget not found"}, status=status.HTTP_404_NOT_FOUND)
+                return Response({"amount": None})
+
             serializer = BudgetSerializer(budget)
+
             return Response(data={"amount": serializer.data["amount"]})
         except Exception as e:
             return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
@@ -27,9 +30,12 @@ def create(self, request):
             data["username"] = user_id
 
             serializer = BudgetSerializer(data=data)
+
             if serializer.is_valid():
                 serializer.save()
+
                 return Response(data={"amount": serializer.data["amount"]}, status=status.HTTP_201_CREATED)
+
             return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
         except Exception as e:
             return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
@@ -39,9 +45,12 @@ def destroy(self, request):
         try:
             user_id = get_user_id_from_token(request)
             budget = Budget.objects.filter(username=user_id).order_by("-created_at").first()
+
             if not budget:
                 return Response({"error": "Budget not found"}, status=status.HTTP_404_NOT_FOUND)
+
             budget.delete()
+
             return Response(status=status.HTTP_204_NO_CONTENT)
         except Exception as e:
             return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
@@ -51,12 +60,26 @@ def partial_update(self, request):
         try:
             user_id = get_user_id_from_token(request)
             budget = Budget.objects.filter(username=user_id).order_by("-created_at").first()
+
             if not budget:
+                data = request.data.copy()
+                data["username"] = user_id
+                serializer = BudgetSerializer(data=data)
+
+                if serializer.is_valid():
+                    serializer.save()
+
+                    return Response(data={"amount": serializer.data["amount"]})
+
                 return Response({"error": "Budget not found"}, status=status.HTTP_404_NOT_FOUND)
+
             serializer = BudgetSerializer(budget, data=request.data, partial=True)
+
             if serializer.is_valid():
                 serializer.save()
+
                 return Response(data={"amount": serializer.data["amount"]})
+
             return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
         except Exception as e:
             return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)