annotate bio/badges.py @ 1096:d9cd3180c12c

More GCalendar V3 conversion in progress. Built a brand new post editor. It is hardcoded into GCalendar right now. We will make it more general in the future.
author Brian Neal <bgneal@gmail.com>
date Tue, 14 Jun 2016 21:16:09 -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)