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()