X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/1876ec0c8caf982704410cee305d0cde218421d4..30beae7541b44060bdd42b55fe525a1113452b55:/forum/actions/user.py

diff --git a/forum/actions/user.py b/forum/actions/user.py
index 74771a6..6170a17 100644
--- a/forum/actions/user.py
+++ b/forum/actions/user.py
@@ -1,7 +1,7 @@
 from django.utils.translation import ugettext as _
 from django.db.models import F
 from forum.models.action import ActionProxy
-from forum.models import Award
+from forum.models import Award, Badge
 from forum import settings
 from forum.settings import APP_SHORT_NAME
 
@@ -23,6 +23,25 @@ class EditProfileAction(ActionProxy):
             'profile_link': self.hyperlink(self.user.get_profile_url(), _('profile')),
         }
 
+class BonusRepAction(ActionProxy):
+    def process_data(self, value):
+        self._value = value
+
+    def repute_users(self):
+        self.repute(self.user, self._value)
+        self.user.message_set.create(message=_("Congratulations, you have been awarded an extra %s reputation points.") % self._value +
+                                     '<br />%s' % self.extra.get('message', _('Thank you')))
+        
+    def describe(self, viewer=None):
+        value = self.extra.get('value', _('unknown'))
+        message = self.extra.get('message', '')
+
+        return _("%(user)s %(was_were)s awarded %(value)s reputation points: %(message)s") % {
+            'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),
+            'was_were': self.viewer_or_user_verb(viewer, self.user, _('were'), _('was')),
+            'value': value, 'message': message
+        }
+
 class AwardAction(ActionProxy):
     def process_data(self, badge, trigger):
         self.__dict__['_badge'] = badge
@@ -39,8 +58,17 @@ class AwardAction(ActionProxy):
         award.save()
         award.badge.awarded_count = F('awarded_count') + 1
         award.badge.save()
-        self.user.message_set.create(message=_("""Congratulations, you have received a badge '%(badge_name)s'
-                                     Check out <a href=\"%(profile_url)s\">your profile</a>.""") %
+
+        if award.badge.type == Badge.GOLD:
+            self.user.gold += 1
+        if award.badge.type == Badge.SILVER:
+            self.user.silver += 1
+        if award.badge.type == Badge.BRONZE:
+            self.user.bronze += 1
+
+        self.user.save()
+
+        self.user.message_set.create(message=_("""Congratulations, you have received a badge '%(badge_name)s'. Check out <a href=\"%(profile_url)s\">your profile</a>.""") %
                                      dict(badge_name=award.badge.name, profile_url=self.user.get_profile_url()))
 
     def cancel_action(self):