mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-18 22:44:24 +01:00
31 lines
1019 B
Python
31 lines
1019 B
Python
from flask_restx import Resource
|
|
from sqlalchemy import func
|
|
|
|
from CTFd.api.v1.statistics import statistics_namespace
|
|
from CTFd.models import Users
|
|
from CTFd.utils.decorators import admins_only
|
|
|
|
|
|
@statistics_namespace.route("/users")
|
|
class UserStatistics(Resource):
|
|
@admins_only
|
|
def get(self):
|
|
registered = Users.query.count()
|
|
confirmed = Users.query.filter_by(verified=True).count()
|
|
data = {"registered": registered, "confirmed": confirmed}
|
|
return {"success": True, "data": data}
|
|
|
|
|
|
@statistics_namespace.route("/users/<column>")
|
|
class UserPropertyCounts(Resource):
|
|
@admins_only
|
|
def get(self, column):
|
|
if column in Users.__table__.columns.keys():
|
|
prop = getattr(Users, column)
|
|
data = (
|
|
Users.query.with_entities(prop, func.count(prop)).group_by(prop).all()
|
|
)
|
|
return {"success": True, "data": dict(data)}
|
|
else:
|
|
return {"success": False, "message": "That could not be found"}, 404
|