Add notifications when a badge is awarded

eslint.config.mjs

@@ -59,6 +59,7 @@ export default [
         jQuery: 'readonly',
         Clipboard: 'readonly',
         Chart: 'readonly',
+        confetti: 'readonly',
         NProgress: 'readonly',
         diff_match_patch: 'readonly',
         Highcharts: 'readonly',

pontoon/base/forms.py

@@ -7,18 +7,17 @@
 from django import forms
 from django.conf import settings
 from django.core.exceptions import ValidationError
-from django.utils import timezone
 
 from pontoon.base import utils
 from pontoon.base.models import (
     Locale,
-    PermissionChangelog,
     ProjectLocale,
     User,
     UserProfile,
 )
 from pontoon.sync.formats import are_compatible_files
 from pontoon.teams.utils import log_group_members
+from pontoon.translations.views import _send_badge_notification
 
 
 class HtmlField(forms.CharField):
@@ -91,6 +90,8 @@ class UserPermissionLogFormMixin:
     def __init__(self, *args, **kwargs):
         self.user = kwargs.pop("user")
         super().__init__(*args, **kwargs)
+        # Track if user reached new level for Community Builder Badge
+        self.community_builder_level_reached = 0
 
     def assign_users_to_groups(self, group_name, users):
         """
@@ -102,42 +103,11 @@ def assign_users_to_groups(self, group_name, users):
 
         group.user_set.clear()
 
-        before_count = self.user.badges_promotion_count
-        now = timezone.now()
-
         if users:
             group.user_set.add(*users)
 
         log_group_members(self.user, group, (add_users, remove_users))
 
-        after_count = self.user.badges_promotion_count
-
-        # TODO:
-        # This code is the only consumer of the PermissionChangelog
-        # model, so we should refactor in the future to simplify
-        # how promotions are retrieved. (see #2195)
-
-        # Check if user was demoted from Manager to Translator
-        # In this case, it doesn't count as a promotion
-        if group_name == "managers":
-            removal = PermissionChangelog.objects.filter(
-                performed_by=self.user,
-                action_type=PermissionChangelog.ActionType.REMOVED,
-                created_at__gte=now,
-            )
-            if removal:
-                for item in removal:
-                    if "managers" in item.group.name:
-                        after_count -= 1
-
-        # Award Community Builder badge
-        if (
-            after_count > before_count
-            and after_count in settings.BADGES_PROMOTION_THRESHOLDS
-        ):
-            # TODO: Send a notification to the user
-            pass
-
 
 class LocalePermsForm(UserPermissionLogFormMixin, forms.ModelForm):
     translators = forms.ModelMultipleChoiceField(
@@ -158,9 +128,29 @@ def save(self, *args, **kwargs):
         translators = self.cleaned_data.get("translators", User.objects.none())
         managers = self.cleaned_data.get("managers", User.objects.none())
 
+        before_count = self.user.badges_promotion_count
+
         self.assign_users_to_groups("translators", translators)
         self.assign_users_to_groups("managers", managers)
 
+        after_count = self.user.badges_promotion_count
+
+        # Award Community Builder badge
+        if (
+            after_count > before_count
+            and after_count in settings.BADGES_PROMOTION_THRESHOLDS
+        ):
+            self.community_builder_level_reached = (
+                settings.BADGES_PROMOTION_THRESHOLDS.index(after_count) + 1
+            )
+            _send_badge_notification(
+                self.user,
+                "Community Builder Badge",
+                self.community_builder_level_reached,
+            )
+
+        return self.community_builder_level_reached
+
 
 class ProjectLocalePermsForm(UserPermissionLogFormMixin, forms.ModelForm):
     translators = forms.ModelMultipleChoiceField(

pontoon/base/models/user.py

@@ -1,3 +1,4 @@
+from datetime import timedelta
 from hashlib import md5
 from urllib.parse import quote, urlencode
 
@@ -7,7 +8,7 @@
 from django.conf import settings
 from django.contrib.auth.models import User
 from django.contrib.contenttypes.models import ContentType
-from django.db.models import Count, Q
+from django.db.models import Count, Exists, OuterRef, Q
 from django.urls import reverse
 from django.utils import timezone
 
@@ -243,9 +244,27 @@ def badges_review_count(self):
 @property
 def badges_promotion_count(self):
     """Role promotions performed by user that count towards their badges"""
-    return self.changed_permissions_log.filter(
+    added_entries = self.changed_permissions_log.filter(
         action_type="added",
         created_at__gte=settings.BADGES_START_DATE,
+    )
+
+    # Check if user was demoted from Manager to Translator.
+    # In this case, it doesn't count as a promotion.
+    #
+    # TODO:
+    # This code is the only consumer of the PermissionChangelog model, so we should
+    # refactor to simplify how promotions are retrieved. (see #2195)
+    return added_entries.exclude(
+        Exists(
+            self.changed_permissions_log.filter(
+                performed_by=OuterRef("performed_by"),
+                performed_on=OuterRef("performed_on"),
+                action_type="removed",
+                created_at__gt=OuterRef("created_at"),
+                created_at__lte=OuterRef("created_at") + timedelta(milliseconds=10),
+            )
+        )
     ).count()