Merge pull request #11 from btaquee/trac

transactions

Author: XiyuanWu

transactions/admin.py

@@ -1,3 +1,6 @@
 from django.contrib import admin
+from .models import Transaction
+
+admin.site.register(Transaction)
 
 # Register your models here.

transactions/models.py

@@ -1,3 +1,19 @@
 from django.db import models
+from django.conf import settings
+from cards.models import Card, RewardRule
 
 # Create your models here.
+
+
+class Transaction(models.Model):
+    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="transactions")
+    card = models.ForeignKey(Card, on_delete=models.CASCADE, related_name="transactions")
+    merchant = models.CharField(max_length=255)
+    amount = models.DecimalField("Amount ($)", max_digits=10, decimal_places=2)
+    category = models.CharField(max_length=255, choices=RewardRule.CATEGORY_CHOICES)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    notes = models.TextField(blank=True, null=True)
+
+    def __str__(self):
+        return f"{self.merchant} - ${self.amount} ({self.user.username})"

transactions/serializers.py

@@ -0,0 +1,8 @@
+from rest_framework import serializers
+from .models import Transaction
+
+class TransactionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Transaction
+        fields = ("id", "card", "merchant", "amount", "category", "created_at", "updated_at", "notes")
+        read_only_fields = ("id", "user", "created_at", "updated_at")

transactions/urls.py

@@ -1,6 +1,12 @@
-from django.urls import path
+from django.urls import path, include
 from . import views
+from rest_framework.routers import DefaultRouter
+from .views import TransactionViewSet
+
+router = DefaultRouter()
+router.register(r'transactions', TransactionViewSet, basename='transactions')
 
 urlpatterns = [
+    path('', include(router.urls)),
     path('health/', views.HealthCheckView.as_view(), name='health'),
 ]

transactions/views.py

@@ -2,9 +2,28 @@
 from rest_framework.views import APIView
 from rest_framework.response import Response
 from rest_framework import status
+from .models import Transaction
+from .serializers import TransactionSerializer
+from rest_framework import viewsets, permissions
 
 # Create your views here.
 # Check API health
 class HealthCheckView(APIView):
     def get(self, request):
-        return Response({"status": "ok"}, status=status.HTTP_200_OK)
+        return Response({"status": "ok"}, status=status.HTTP_200_OK)
+
+
+class TransactionViewSet(viewsets.ModelViewSet):
+    serializer_class = TransactionSerializer
+    permission_classes = [permissions.IsAuthenticated]
+    
+    # AI help me understand the concept of two functions below, what is does in detail
+    
+    # Records the API should return when a user makes a GET request
+    # No filter. Will return all transactions for that user, sorted by created_at in descending order
+    def get_queryset(self):
+        return Transaction.objects.filter(user=self.request.user).select_related('card', 'card__issuer').order_by("-created_at")
+
+    # Saves the transaction to the database
+    def perform_create(self, serializer):
+        serializer.save(user=self.request.user)