include subscription info in stats

Author: jonasjucker

bot.py

@@ -178,13 +178,28 @@ async def _stats(self, update: Update, context: CallbackContext):
             return
 
         activity_summary_text = []
-        activity_summary_text.append('*Activity summary*')
+        
+        # Query activity summary for each interval
+        activity_summary_text.append('*Activity*')
         for interval in VALID_SUMMARY_INTERVALS:
             activity_summary = self._db.get_activity_summary(interval)
             activity_summary_text.append(f"_{interval.lower()}_")
             for activity in activity_summary:
                 activity_summary_text.append(f"- {activity}")
             activity_summary_text.append('')
+
+        # Query subscription summary
+        activity_summary_text.append('*Subscriptions*')
+        subscription_summary = self._db.get_subscription_summary()
+        for station in subscription_summary:
+            activity_summary_text.append(f"- {station}")
+        
+        # Query unique subscribers
+        unique_subscribers = self._db.count_unique_subscribers()
+        activity_summary_text.append('')
+        activity_summary_text.append(f"_Unique subscribers: {unique_subscribers}_")
+        activity_summary_text.append('')
+
         activity_summary_text = "\n".join(activity_summary_text)
         await update.message.reply_markdown(activity_summary_text)
 

db.py

@@ -128,6 +128,21 @@ def get_subscriptions_by_station(self, station) -> list[int]:
                 [subscription['user_id'] for subscription in subscriptions])
         else:
             return []
+    
+    def count_unique_subscribers(self) -> list[int]:
+        sql = f"""
+            SELECT DISTINCT user_id
+            FROM subscriptions_{self._table_suffix}
+        """
+        subscribers = self._select(sql)
+        return len([subscriber['user_id'] for subscriber in subscribers])
+
+    def get_subscription_summary(self) -> list[str]:
+        stations = self.stations_with_subscribers()
+        summary = []
+        for station in stations:
+            summary.append(f"{station}: {len(self.get_subscriptions_by_station(station))}")
+        return summary
 
     def _select_with_values(self, sql, values):
         connection = self._get_db_connection()

test/test_db.py

@@ -107,3 +107,23 @@ def test_get_subscriptions_by_station(db_instance):
     # Test for a station with no subscriptions
     users = db_instance.get_subscriptions_by_station("station3")
     assert users == []
+
+def test_subscription_summary(db_instance):
+    # Add test data
+    db_instance.add_subscription("station1", 12345)
+    db_instance.add_subscription("station2", 67890)
+    db_instance.add_subscription("station1", 54321)
+
+    summary = db_instance.get_subscription_summary()
+
+    assert summary == ["station1: 2", "station2: 1"]
+
+def test_get_unique_subscribers(db_instance):
+    # Add test data
+    db_instance.add_subscription("station1", 12345)
+    db_instance.add_subscription("station2", 67890)
+    db_instance.add_subscription("station1", 54321)
+
+    unique_subscribers = db_instance.count_unique_subscribers()
+
+    assert unique_subscribers == 3