Admin: calculate stats from the Admin page (mozilla#4063)

Author: functionzz

pontoon/administration/static/css/admin.css

@@ -1,13 +1,19 @@
-.add {
-  float: right;
-}
-
 menu.controls {
   display: flex;
   align-items: center;
   gap: 10px;
 }
 
+menu.controls .search-wrapper {
+  flex: 1;
+}
+
+menu.controls #add-project,
+menu.controls #toggle-projects {
+  flex-shrink: 0;
+  width: 200px;
+}
+
 .disabled-projects {
   display: none;
 }
@@ -24,3 +30,14 @@ menu.controls {
 .request-toggle.back:before {
   content: '';
 }
+
+#recalculate-stats {
+  float: right;
+  width: 200px;
+  background: var(--button-background-2);
+  color: var(--light-grey-7);
+}
+
+#recalculate-stats:hover {
+  background: var(--button-background-hover-2);
+}

pontoon/administration/static/js/admin.js

@@ -18,3 +18,33 @@ document.addEventListener('DOMContentLoaded', () => {
     toggleBtn.dataset.showDisabled = (!showingDisabled).toString();
   });
 });
+
+$(function () {
+  $('#recalculate-stats').on('click', function (e) {
+    e.preventDefault();
+
+    const button = $(this),
+      title = button.html();
+
+    if (button.is('.in-progress')) {
+      return;
+    }
+
+    button.addClass('in-progress').html('Calculating...');
+
+    $.ajax({
+      url: '/admin/calculate-stats/',
+      success() {
+        button.html('Done');
+      },
+      error() {
+        button.html('Whoops!');
+      },
+      complete() {
+        setTimeout(function () {
+          button.removeClass('in-progress').html(title);
+        }, 2000);
+      },
+    });
+  });
+});

pontoon/administration/tasks.py

@@ -0,0 +1,18 @@
+import logging
+
+from celery import shared_task
+
+from django.core.management import call_command
+
+from pontoon.base.tasks import PontoonTask
+
+
+log = logging.getLogger(__name__)
+
+
+@shared_task(base=PontoonTask, name="calculate_stats")
+def calculate_stats_task():
+    try:
+        call_command("calculate_stats")
+    except Exception as err:
+        log.error(f"Calculate Stats failed: {err}")

pontoon/administration/templates/admin.html

@@ -19,7 +19,17 @@
 {% block class %}admin{% endblock %}
 
 {% block heading %}
-  {{ Heading.heading(title='Admin', subtitle="Don't mess it up. Please.") }}
+  <section id="heading">
+    <div class="container clearfix">
+      <div class="controls">
+        <button id="recalculate-stats" class="button">
+          Re-calculate stats
+        </button>
+      </div>
+      <h1 id="title">Admin</h1>
+      <h2 id="subtitle">Don't mess it up. Please.</h2>
+    </div>
+  </section>
 {% endblock %}
 
 {% block bottom %}
@@ -38,6 +48,7 @@
         </div>
         <a
           class="add button small"
+          id="add-project"
           href="{{ url('pontoon.admin.project.new') }}"
           >Add new project</a
         >

pontoon/administration/urls.py

@@ -36,6 +36,12 @@
             ]
         ),
     ),
+    # Calculate Stats
+    path(
+        "calculate-stats/",
+        views.manually_calculate_stats,
+        name="pontoon.admin.calculate-stats",
+    ),
     # AJAX view: Get slug
     path("get-slug/", views.get_slug, name="pontoon.admin.project.slug"),
     # AJAX view: Get project locales

pontoon/administration/views.py

@@ -18,6 +18,7 @@
     RepositoryInlineFormSet,
     TagInlineFormSet,
 )
+from pontoon.administration.tasks import calculate_stats_task
 from pontoon.base import utils
 from pontoon.base.models import (
     Entity,
@@ -544,6 +545,19 @@ def manage_project_strings(request, slug=None):
     return render(request, "admin_project_strings.html", data)
 
 
+@login_required(redirect_field_name="", login_url="/403")
+@require_AJAX
+def manually_calculate_stats(request):
+    if not request.user.has_perm("base.can_manage_project"):
+        return HttpResponseForbidden(
+            "Forbidden: You don't have permission for calculating statistics"
+        )
+
+    calculate_stats_task.delay()
+
+    return HttpResponse("ok")
+
+
 @login_required(redirect_field_name="", login_url="/403")
 @require_AJAX
 def manually_sync_project(request, slug):