annotate gpp/bio/badges.py @ 322:c3d3d7114749
Fix #148; Django now requires AJAX posts to present the CSRF token. Added code suggested by Django docs to shoutbox.js. Since shoutbox.js is on every page, it should cover all cases.
author |
Brian Neal <bgneal@gmail.com> |
date |
Sat, 12 Feb 2011 21:37:17 +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()
|