From c44f42c065a7887bebfaa5750717da6054367303 Mon Sep 17 00:00:00 2001
From: Peter Odeny <odenypeter@gmail.com>
Date: Sun, 30 Jun 2024 16:26:20 +0200
Subject: [PATCH] implement subscription cancellation

---
 core/views/subscription.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/core/views/subscription.py b/core/views/subscription.py
index 9cfa6707..9f857447 100644
--- a/core/views/subscription.py
+++ b/core/views/subscription.py
@@ -27,10 +27,24 @@ class SubscriptionViewSet(viewsets.ModelViewSet):
      All the subscriptions related actions
 
      """
-    queryset = Subscription.objects.filter(cancelled=False).order_by('-create_date')
+    queryset = Subscription.objects.all()
     serializer_class = SubscriptionSerializer
     permission_classes = (AllowAny, IsAuthenticated)
 
+    def get_queryset(self):
+        queryset = (
+            super(SubscriptionViewSet, self)
+            .get_queryset()
+            .filter(organization=self.request.user.organization)
+        )
+
+        if int(self.request.query_params.get('cancelled', '0')):
+            queryset = queryset.filter(cancelled=True)
+        else:
+            queryset = queryset.filter(cancelled=False)
+
+        return queryset.order_by('-create_date')
+
     def create(self, request, *args, **kwargs):
         if settings.STRIPE_SECRET:
             stripe.api_key = settings.STRIPE_SECRET