Add clearance viewset for student

equipment_tracker/accounts/serializers.py

@@ -52,3 +52,7 @@ class UserRegistrationSerializer(serializers.ModelSerializer):
         user.save()
 
         return user
+
+
+class ClearanceSerializer(serializers.Serializer):
+    cleared = serializers.CharField()

equipment_tracker/accounts/urls.py

@@ -1,7 +1,8 @@
 from django.contrib import admin
 from django.urls import path, include
-
+from accounts.views import ClearanceViewSet
 urlpatterns = [
     path('', include('djoser.urls')),
     path('', include('djoser.urls.jwt')),
+    path('clearance/', ClearanceViewSet.as_view()),
 ]

equipment_tracker/accounts/views.py

@@ -1,3 +1,26 @@
-from django.shortcuts import render
+from transactions.models import Transaction
+from accounts.permissions import IsStudent
+from rest_framework.permissions import IsAuthenticated
+from rest_framework import generics
+from accounts.serializers import ClearanceSerializer
+from rest_framework.response import Response
 
-# Create your views here.
+
+class ClearanceViewSet(generics.GenericAPIView):
+    # Check all transactions associated with the student. If any are uncleared or not finalized, return uncleared
+    http_method_names = ['get']
+    permission_classes = [IsAuthenticated, IsStudent]
+    serializer_class = ClearanceSerializer
+
+    def get_queryset(self):
+        user = self.request.user
+        return Transaction.objects.filter(borrower=user)
+
+    def get(self, request, *args, **kwargs):
+        queryset = self.get_queryset()
+        status_list = ["Approved", "Borrowed", "Returned: Pending Checking",
+                       "With Breakages: Pending Resolution"]
+        for transaction in queryset:
+            if transaction.transaction_status in status_list:
+                return Response({"cleared": "Uncleared"})
+        return Response({"cleared": "Cleared"})