Add top_users

apps/group_stat_api.py

@@ -6,6 +6,37 @@ from datetime import datetime
 
 group_stat_api = Blueprint('group_stat_api', __name__)
 
+
+@group_stat_api.route("/top_users", methods=["GET"])
+def top_users():
+    number = request.args.get("number")
+
+    if not number: number = 3
+
+    number = int(number)
+
+    members = []
+    counts = []
+
+    query = (Member
+        .select(fn.count(Message.id).alias("count"), Member)
+        .join(Message)
+        .group_by(Member)
+        .order_by(SQL("count").desc())
+    )
+
+    for group in query:
+        members.append(group.first_name)
+        counts.append(group.count)
+
+    members.insert(0, "Others")
+    counts.insert(0, sum(counts[number:]))
+
+    return jsonify({
+        "members": members[:number + 1],
+        "counts": counts[:number + 1]
+    })
+
 @group_stat_api.route("/chat_activity", methods=["GET"])
 def chat_activity():
     from_date = request.args.get("from_date")
@@ -28,11 +59,11 @@ def chat_activity():
         to_date = datetime.strptime(to_date, date_format)
         
         query = (Message
-                 .select(fn.date_trunc(group_by, Message.timestamp).alias('range'), fn.count(Message.id).alias('count'))
+            .select(fn.date_trunc(group_by, Message.timestamp).alias('range'), fn.count(Message.id).alias('count'))
-                 .where((Message.timestamp >= from_date) & (Message.timestamp <= to_date))
+            .where((Message.timestamp >= from_date) & (Message.timestamp <= to_date))
-                 .group_by(fn.date_trunc(group_by, Message.timestamp))
+            .group_by(fn.date_trunc(group_by, Message.timestamp))
-                 .order_by(SQL("range"))
+            .order_by(SQL("range"))
-                 )
+        )
 
         for group in query:
             dates.append(group.range.strftime("%Y-%m-%d"))