annotate gpp/bio/badges.py @ 393:9af6bd45c1f8
Another try at #191 after getting some good advice in a django-users thread.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Thu, 24 Mar 2011 00:15:34 +0000 |
| parents |
767cedc7d12a |
| children |
47f4259ce511 |
| 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@204
|
10 SECURITY_PIN) = range(6)
|
|
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@204
|
24 profile = user.get_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()
|
