mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-18 05:51:26 +02:00
83 lines
3.4 KiB
Python
83 lines
3.4 KiB
Python
# coding: utf-8
|
|
from django.utils.translation import ugettext as _, ugettext_noop
|
|
from modoboa.lib.webutils import _render, _render_to_string, ajax_simple_response
|
|
from modoboa.admin.models import Domain, Mailbox
|
|
from django.contrib.auth.decorators \
|
|
import login_required, user_passes_test, permission_required
|
|
from modoboa.extensions.stats.grapher import *
|
|
from modoboa.lib.exceptions import *
|
|
|
|
graph_types = ['AVERAGE', 'MAX']
|
|
graph_list = [{"name" : "traffic", "label" : ugettext_noop("Average normal traffic")},
|
|
{"name" : "badtraffic", "label" : ugettext_noop("Average bad traffic")},
|
|
{"name" : "size", "label" : ugettext_noop("Average normal traffic size")}]
|
|
periods = [{"name" : "day", "label" : ugettext_noop("Day")},
|
|
{"name" : "week", "label" : ugettext_noop("Week")},
|
|
{"name" : "month", "label" : ugettext_noop("Month")},
|
|
{"name" : "year", "label" : ugettext_noop("Year")},
|
|
{"name" : "custom", "label" : ugettext_noop("Custom")}]
|
|
|
|
@login_required
|
|
@permission_required("admin.view_mailboxes")
|
|
def index(request):
|
|
deflocation = ""
|
|
if request.user.is_superuser:
|
|
deflocation = "graphs/?view=global"
|
|
else:
|
|
if len(request.user.get_domains()):
|
|
deflocation = "graphs/?view=%s" % request.user.get_domains()[0].name
|
|
else:
|
|
raise ModoboaException(_("No statistics available"))
|
|
|
|
period = request.GET.get("period", "day")
|
|
return _render(request, 'stats/index.html', {
|
|
"graphs" : graph_list,
|
|
"periods" : periods,
|
|
"period" : period,
|
|
"selection" : "stats",
|
|
"deflocation" : deflocation
|
|
})
|
|
|
|
@login_required
|
|
@user_passes_test(lambda u: u.group != "SimpleUsers")
|
|
def graphs(request):
|
|
view = request.GET.get("view", None)
|
|
if not view:
|
|
raise ModoboaException(_("Invalid request"))
|
|
period = request.GET.get("period", "day")
|
|
tplvars = dict(graphs=graph_list, period=period,
|
|
graphs_loc=parameters.get_admin("GRAPHS_LOCATION"))
|
|
if view == "global":
|
|
if not request.user.is_superuser:
|
|
raise PermDeniedException(_("you're not allowed to see those statistics"))
|
|
tplvars.update(domain=view)
|
|
else:
|
|
try:
|
|
domain = Domain.objects.get(name=view)
|
|
except Domain.DoesNotExist:
|
|
raise ModoboaException(_("Domain not found. Please enter a full name"))
|
|
if not request.user.can_access(domain):
|
|
raise PermDeniedException(_("You don't have access to this domain"))
|
|
tplvars.update(domain=domain.name)
|
|
|
|
if period == "custom":
|
|
if not request.GET.has_key("start") or not request.GET.has_key("end"):
|
|
raise ModoboaException(_("Bad custom period"))
|
|
start = request.GET["start"]
|
|
end = request.GET["end"]
|
|
G = Grapher()
|
|
period_name = "%s_%s" % (start.replace('-',''), end.replace('-',''))
|
|
for tpl_name in graph_list:
|
|
G.process(tplvars["domain"],
|
|
period_name,
|
|
str2Time(*start.split('-')),
|
|
str2Time(*end.split('-')),
|
|
tpl[tpl_name['name']])
|
|
tplvars["period_name"] = period_name
|
|
tplvars["start"] = start
|
|
tplvars["end"] = end
|
|
|
|
return ajax_simple_response(dict(
|
|
status="ok", content=_render_to_string(request, "stats/graphs.html", tplvars)
|
|
))
|