annotate gpp/bio/badges.py @ 318:c550933ff5b6
Fix a bug where you'd get an error when trying to delete a forum thread (topic does not exist). Apparently when you call topic.delete() the posts would get deleted, but the signal handler for each one would run, and it would try to update the topic's post count or something, but the topic was gone? Reworked the code a bit and explicitly delete the posts first. I also added a sync() call on the parent forum since post counts were not getting adjusted.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Sat, 05 Feb 2011 21:46:52 +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()
|
