76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
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("/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")
|
|
to_date = request.args.get("to_date")
|
|
|
|
date_format = request.args.get("date_format")
|
|
group_by = request.args.get("group_by")
|
|
|
|
if not date_format:
|
|
date_format = "%Y-%m-%d"
|
|
|
|
if not group_by:
|
|
group_by = "day"
|
|
|
|
dates = []
|
|
counts = []
|
|
|
|
if from_date and to_date:
|
|
from_date = datetime.strptime(from_date, date_format)
|
|
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'))
|
|
.where((Message.timestamp >= from_date) & (Message.timestamp <= to_date))
|
|
.group_by(fn.date_trunc(group_by, Message.timestamp))
|
|
.order_by(SQL("range"))
|
|
)
|
|
|
|
for group in query:
|
|
dates.append(group.range.strftime("%Y-%m-%d"))
|
|
counts.append(group.count)
|
|
|
|
return jsonify({
|
|
"date": dates,
|
|
"counts": counts
|
|
})
|