X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/19a817e69c4518215a96e0456bbea06aa4e03e63..5806688fd0986ae297d9c500382585cecbaa72d3:/forum/actions/user.py?ds=sidebyside diff --git a/forum/actions/user.py b/forum/actions/user.py index 7018a5d..0e894ba 100644 --- a/forum/actions/user.py +++ b/forum/actions/user.py @@ -1,56 +1,56 @@ -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 import settings - -class UserJoinsAction(ActionProxy): - def repute_users(self): - self.repute(self.user, int(settings.INITIAL_REP)) - -class EditProfileAction(ActionProxy): - pass - -class AwardAction(ActionProxy): - def process_data(self, badge, trigger): - self.__dict__['_badge'] = badge - self.__dict__['_trigger'] = trigger - - def process_action(self): - badge = self.__dict__['_badge'] - trigger = self.__dict__['_trigger'] - - award = Award(user=self.user, badge=badge, trigger=trigger, action=self) - if self.node: - award.node = self.node - - 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 your profile.""") % - dict(badge_name=award.badge.name, profile_url=self.user.get_profile_url())) - - def cancel_action(self): - award = self.award - badge = award.badge - badge.awarded_count = F('awarded_count') - 1 - badge.save() - award.delete() - - @classmethod - def get_for(cls, user, badge, node=False): - try: - if node is False: - return Award.objects.get(user=user, badge=badge).action - else: - return Award.objects.get(user=user, node=node, badge=badge).action - except: - return None - - def describe(self, viewer=None): - return _("%(user)s %(were_was)s awarded the %(badge_name)s badge") % { - 'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)), - 'were_was': self.viewer_or_user_verb(viewer, self.user, _('were'), _('was')), - 'badge_name': self.award.all()[0].badge.name, +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 import settings + +class UserJoinsAction(ActionProxy): + def repute_users(self): + self.repute(self.user, int(settings.INITIAL_REP)) + +class EditProfileAction(ActionProxy): + pass + +class AwardAction(ActionProxy): + def process_data(self, badge, trigger): + self.__dict__['_badge'] = badge + self.__dict__['_trigger'] = trigger + + def process_action(self): + badge = self.__dict__['_badge'] + trigger = self.__dict__['_trigger'] + + award = Award(user=self.user, badge=badge, trigger=trigger, action=self) + if self.node: + award.node = self.node + + 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 your profile.""") % + dict(badge_name=award.badge.name, profile_url=self.user.get_profile_url())) + + def cancel_action(self): + award = self.award + badge = award.badge + badge.awarded_count = F('awarded_count') - 1 + badge.save() + award.delete() + + @classmethod + def get_for(cls, user, badge, node=False): + try: + if node is False: + return Award.objects.get(user=user, badge=badge).action + else: + return Award.objects.get(user=user, node=node, badge=badge).action + except: + return None + + def describe(self, viewer=None): + return _("%(user)s %(were_was)s awarded the %(badge_name)s badge") % { + 'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)), + 'were_was': self.viewer_or_user_verb(viewer, self.user, _('were'), _('was')), + 'badge_name': self.award.badge.name, } \ No newline at end of file