update calculate metrics task return value

membermatters · May 31, 2024 · defd85f · defd85f
1 parent da579e2
commit defd85f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/memberportal/api_metrics/tasks.py b/memberportal/api_metrics/tasks.py
@@ -25,10 +25,6 @@ def setup_periodic_tasks(sender, **kwargs):
 @app.task
 def calculate_metrics():
     logger.info("Calculating metrics!")
-    metric_results = {
-        "member_count": [],
-        "subscription_count": [],
-    }
 
     # get the count of all the different member profile states
     logger.debug("Calculating member count total")
@@ -68,4 +64,9 @@ def calculate_metrics():
     except Exception as e:
         logger.error(f"Failed to update Prometheus metrics: {e}")
 
-    return metric_results
+    return {
+        "member_count": Profile.objects.count(),
+        "subscription_count": Profile.objects.filter(
+            subscription_status__in=["active", "cancelling"]
+        ).count(),
+    }