Mercurial > public > sg101
comparison accounts/stats.py @ 581:ee87ea74d46b
For Django 1.4, rearranged project structure for new manage.py.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Sat, 05 May 2012 17:10:48 -0500 |
| parents | gpp/accounts/stats.py@e94570675664 |
| children |
comparison
equal
deleted
inserted
replaced
| 580:c525f3e0b5d0 | 581:ee87ea74d46b |
|---|---|
| 1 """ | |
| 2 This module performs user account related statistics. | |
| 3 | |
| 4 """ | |
| 5 import logging | |
| 6 | |
| 7 from django.db.models.signals import post_save | |
| 8 from django.contrib.auth.models import User | |
| 9 | |
| 10 from core.services import get_redis_connection | |
| 11 | |
| 12 | |
| 13 # Redis key names | |
| 14 USER_COUNT_KEY = "accounts:user_count" | |
| 15 NEW_USERS_KEY = "accounts:new_users" | |
| 16 | |
| 17 | |
| 18 logger = logging.getLogger(__name__) | |
| 19 | |
| 20 | |
| 21 def on_user_save(sender, **kwargs): | |
| 22 """ | |
| 23 This function is our signal handler for when a User object is saved. | |
| 24 | |
| 25 """ | |
| 26 from accounts.tasks import user_stats_task | |
| 27 | |
| 28 if kwargs['created']: | |
| 29 user = kwargs['instance'] | |
| 30 | |
| 31 # kick off a task to update user stats | |
| 32 | |
| 33 user_stats_task.delay(user.id) | |
| 34 | |
| 35 | |
| 36 def update_user_stats(user_id): | |
| 37 """ | |
| 38 This function is given a new user id and is responsible for updating various | |
| 39 user account statistics. | |
| 40 | |
| 41 """ | |
| 42 try: | |
| 43 user = User.objects.get(pk=user_id) | |
| 44 except User.DoesNotExist: | |
| 45 logger.warning("update_user_stats: user id %d doesn't exist", user_id) | |
| 46 return | |
| 47 | |
| 48 redis = get_redis_connection() | |
| 49 | |
| 50 # update the count of registered users | |
| 51 | |
| 52 count = redis.incr(USER_COUNT_KEY) | |
| 53 if count == 1: | |
| 54 # it is likely redis got wiped out; update it now | |
| 55 count = User.objects.all().count() | |
| 56 redis.set(USER_COUNT_KEY, count) | |
| 57 | |
| 58 # update the list of new users | |
| 59 | |
| 60 pipeline = redis.pipeline() | |
| 61 pipeline.lpush(NEW_USERS_KEY, user.username) | |
| 62 pipeline.ltrim(NEW_USERS_KEY, 0, 9) | |
| 63 pipeline.execute() | |
| 64 | |
| 65 | |
| 66 def get_user_count(redis=None): | |
| 67 """ | |
| 68 This function returns the current count of users. | |
| 69 | |
| 70 """ | |
| 71 if redis is None: | |
| 72 redis = get_redis_connection() | |
| 73 return redis.get(USER_COUNT_KEY) | |
| 74 | |
| 75 | |
| 76 def get_new_users(redis=None): | |
| 77 """ | |
| 78 This function returns a list of new usernames. | |
| 79 | |
| 80 """ | |
| 81 if redis is None: | |
| 82 redis = get_redis_connection() | |
| 83 return redis.lrange(NEW_USERS_KEY, 0, -1) | |
| 84 | |
| 85 | |
| 86 def get_user_stats(redis=None): | |
| 87 """ | |
| 88 This function returns a tuple of the user stats. Element 0 is the user count | |
| 89 and element 1 is the list of new users. | |
| 90 | |
| 91 """ | |
| 92 if redis is None: | |
| 93 redis = get_redis_connection() | |
| 94 return get_user_count(redis), get_new_users(redis) | |
| 95 | |
| 96 | |
| 97 post_save.connect(on_user_save, sender=User, dispatch_uid='accounts.stats') |