Update endpoints to handle Enterprise plans

kobo/apps/stripe/views.py

@@ -324,14 +324,14 @@ def generate_portal_link(user, organization_id, price):
             if not len(all_configs):
                 return Response({'error': "Missing Stripe billing configuration."}, status=status.HTTP_502_BAD_GATEWAY)
 
-            is_price_for_addon = price.product.metadata.get('product_type', '') == 'addon'
-
-            if is_price_for_addon:
-                """
-                Recurring add-ons aren't included in the default billing configuration.
-                This lets us hide them as an 'upgrade' option for paid plan users.
-                Here, we try getting the portal configuration that lets us switch to the provided price.
-                """
+            """
+            Recurring add-ons and the Enterprise plan aren't included in the default billing configuration.
+            This lets us hide them as an 'upgrade' option for paid plan users.
+            """
+            needs_custom_config = price.product.metadata.get('product_type', '') == 'addon'
+            needs_custom_config = needs_custom_config or price.product.metadata.get('plan_type', '') == 'enterprise'
+            if needs_custom_config:
+                # Try getting the portal configuration that lets us switch to the provided price
                 current_config = next(
                     (config for config in all_configs if (
                             config['active'] and
@@ -350,7 +350,7 @@ def generate_portal_link(user, organization_id, price):
                     )), None
                 )
 
-                if is_price_for_addon:
+                if needs_custom_config:
                     """
                     we couldn't find a custom configuration, let's try making a new one
                     add the price we're switching into to the list of prices that allow subscription updates

kpi/serializers/v2/service_usage.py

@@ -1,10 +1,13 @@
+from django.contrib.auth.models import User
+from django.conf import settings
 from django.db.models import Sum, Q
 from django.db.models.functions import Coalesce
 from django.utils import timezone
 from rest_framework import serializers
 from rest_framework.fields import empty
 
 from kobo.apps.organizations.models import Organization
+from kobo.apps.stripe.constants import ACTIVE_STRIPE_STATUSES
 from kobo.apps.trackers.models import NLPUsageCounter
 from kpi.deployment_backends.kc_access.shadow_models import (
     KobocatXForm,
@@ -219,16 +222,16 @@ def _get_organization_details(self):
             # Couldn't find organization, proceed as normal
             return
 
-        """
-        Commented out until the Enterprise plan is implemented
-
-        # If the user is in an organization, get all org users so we can query their total org usage
-        self._user_ids = list(
-            User.objects.values_list('pk', flat=True).filter(
-                organizations_organization__id=organization_id
+        if settings.STRIPE_ENABLED:
+            # if the user is in an organization and has an enterprise plan, get all org users and their total usage
+            enterprise_users = User.objects.values_list('pk', flat=True).filter(
+                organizations_organization__id=organization_id,
+                organizations_organization__djstripe_customers__subscriptions__status__in=ACTIVE_STRIPE_STATUSES,
+                organizations_organization__djstripe_customers__subscriptions__items__price__product__metadata__has_key='plan_type',
+                organizations_organization__djstripe_customers__subscriptions__items__price__product__metadata__plan_type='enterprise',
             )
-        )
-        """
+            if enterprise_users.exists():
+                self._user_ids = list(enterprise_users)
 
         # If they have a subscription, use its start date to calculate beginning of current month/year's usage
         billing_details = organization.active_subscription_billing_details