fix: budget model, id type

Author: mjsifri

Diff for: budget/migrations/0001_initial.py

@@ -1,24 +1,24 @@
-# Generated by Django 5.0.6 on 2024-06-10 15:06
+# Generated by Django 5.0.6 on 2024-06-10 16:41
 
-import uuid
 from django.db import migrations, models
 
 
 class Migration(migrations.Migration):
 
     initial = True
 
-    dependencies = []
+    dependencies = [
+    ]
 
     operations = [
         migrations.CreateModel(
-            name="Budget",
+            name='Budget',
             fields=[
-                ("id", models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
-                ("username", models.UUIDField()),
-                ("amount", models.IntegerField()),
-                ("created_at", models.DateTimeField(auto_now_add=True)),
-                ("updated_at", models.DateTimeField(auto_now=True)),
+                ('id', models.AutoField(primary_key=True, serialize=False)),
+                ('username', models.UUIDField()),
+                ('amount', models.IntegerField()),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
             ],
         ),
     ]

Diff for: budget/models.py

@@ -5,7 +5,7 @@
 
 
 class Budget(models.Model):
-    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    id = models.AutoField(primary_key=True)
     username = models.UUIDField()
     amount = models.IntegerField()
     created_at = models.DateTimeField(auto_now_add=True)

Diff for: budget/serializers.py

@@ -8,3 +8,8 @@ class BudgetSerializer(serializers.Serializer):
 
     def create(self, validated_data):
         return Budget.objects.create(**validated_data)
+
+    def update(self, instance, validated_data):
+        instance.amount = validated_data.get("amount", instance.amount)
+        instance.save()
+        return instance

Diff for: budget/views.py

@@ -30,7 +30,7 @@ def get_user_id_from_token(self, request):
     def list(self, request):
         try:
             username = self.get_user_id_from_token(request)
-            budget = Budget.objects.get(username=username)
+            budget = Budget.objects.filter(username=username).order_by('-created_at').first()
             serializer = BudgetSerializer(budget)
             return Response(serializer.data)
         except Budget.DoesNotExist:
@@ -58,7 +58,7 @@ def create(self, request):
     def destroy(self, request):
         try:
             username = self.get_user_id_from_token(request)
-            budget = Budget.objects.get(username=username)
+            budget = Budget.objects.filter(username=username).order_by('-created_at').first()
             budget.delete()
             return Response(status=status.HTTP_204_NO_CONTENT)
         except Budget.DoesNotExist:
@@ -70,7 +70,7 @@ def destroy(self, request):
     def partial_update(self, request):
         try:
             username = self.get_user_id_from_token(request)
-            budget = Budget.objects.get(username=username)
+            budget = Budget.objects.filter(username=username).order_by('-created_at').first()
             serializer = BudgetSerializer(budget, data=request.data, partial=True)
             if serializer.is_valid():
                 serializer.save()