annotate bio/badges.py @ 917:0365fdbb4d78

Fix app conflict with messages. Django's messages app label conflicts with our messages app. We can't easily rename our label as that will make us rename database tables. Since our app came first we'll just customize Django messages label. For Django 1.7.7 upgrade.
author Brian Neal <bgneal@gmail.com>
date Mon, 06 Apr 2015 20:02:25 -0500
parents 5892c05886a9
children
rev   line source
bgneal@204 1 """This module contains user profile badge-related functionality."""
bgneal@316 2 import logging
bgneal@204 3
bgneal@204 4 from bio.models import Badge
bgneal@204 5 from bio.models import BadgeOwnership
bgneal@204 6
bgneal@204 7
bgneal@204 8 # Numeric ID's for badges that are awarded for user actions:
bgneal@204 9 (CONTRIBUTOR_PIN, CALENDAR_PIN, NEWS_PIN, LINK_PIN, DOWNLOAD_PIN,
bgneal@823 10 SECURITY_PIN, POTD_PIN, MAP_PIN) = range(8)
bgneal@204 11
bgneal@204 12
bgneal@204 13 def award_badge(badge_id, user):
bgneal@204 14 """This function awards the badge specified by badge_id
bgneal@204 15 to the given user. If the user already has the badge,
bgneal@204 16 the badge count is incremented by one.
bgneal@204 17 """
bgneal@204 18 try:
bgneal@204 19 badge = Badge.objects.get(numeric_id=badge_id)
bgneal@204 20 except Badge.DoesNotExist:
bgneal@316 21 logging.error("Can't award badge with numeric_id = %d", badge_id)
bgneal@204 22 return
bgneal@204 23
bgneal@789 24 profile = user.profile
bgneal@204 25
bgneal@204 26 # Does the user already have badges of this type?
bgneal@204 27 try:
bgneal@204 28 bo = BadgeOwnership.objects.get(profile=profile, badge=badge)
bgneal@204 29 except BadgeOwnership.DoesNotExist:
bgneal@204 30 # No badge of this type, yet
bgneal@204 31 bo = BadgeOwnership(profile=profile, badge=badge, count=1)
bgneal@204 32 else:
bgneal@204 33 # Already have this badge
bgneal@204 34 bo.count += 1
bgneal@204 35 bo.save()
bgneal@400 36
bgneal@400 37 logging.info('Awarded %s with the badge: %s', user.username, badge.name)