Add /chat_activity

modified:   apps/group_stat_api.py

apps/group_stat_api.py

@@ -1,33 +1,44 @@
 from flask import Blueprint, jsonify, request
+from peewee import SQL, fn
 from database import Message, Member
 
 from datetime import datetime
 
 group_stat_api = Blueprint('group_stat_api', __name__)
 
+@group_stat_api.route("/chat_activity", methods=["GET"])
+def chat_activity():
+    from_date = request.args.get("from_date")
+    to_date = request.args.get("to_date")
 
-@group_stat_api.route("/top_users",methods=["GET"])
-def top_users():
-    number = request.args.get("number")
+    date_format = request.args.get("date_format")
+    group_by = request.args.get("group_by")
 
-    if not number: number = 3
+    if not date_format:
+        date_format = "%Y-%m-%d"
 
-    number = int(number)
+    if not group_by:
+        group_by = "day"
 
-    members = []
+    dates = []
     counts = []
 
-    for user in Member.select():
-        members.append(user.first_name)
-        counts.append(Message.select().where(Message.user_id == user.user_id).count())
+    if from_date and to_date:
+        from_date = datetime.strptime(from_date, date_format)
+        to_date = datetime.strptime(to_date, date_format)
         
-    # Sorting
-    counts, members = (list(t) for t in zip(*sorted(zip(counts,members),reverse=True)))
+        query = (Message
+                 .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))
+                 .group_by(fn.date_trunc(group_by, Message.timestamp))
+                 .order_by(SQL("range"))
+                 )
 
-    members.insert(number, "Others")
-    counts.insert(number,  sum(counts[number:]))
+        for group in query:
+            dates.append(group.range.strftime("%Y-%m-%d"))
+            counts.append(group.count)
 
     return jsonify({
-        "members":members[:number+1],
-        "counts":counts[:number+1]
+        "date": dates,
+        "counts": counts
     })