Logged Exception in update_badges script

Lucas Kanashiro
1 parent 2ce3f49a
Showing 1 changed file with 11 additions and 5 deletions Show diff stats
src/badger/management/commands/update_badges.py
@@ -6,17 +6,26 @@ from haystack.query import SearchQuerySet
 from accounts.models import User
 from badger.models import Badge
+import logging
 class Command(BaseCommand):
     help = "Update the user's badges"
+    def search(self, object_type, **opts={}):
+        try:
+            sqs = SearchQuerySet().filter(type=object_type, **opts)
+        except Exception, e:
+            logging.except(e)
+            raise
+        return sqs
+
     def handle(self, *args, **kwargs):
         for badge in Badge.objects.filter(type='auto'):
             if not badge.comparison:
                 continue
             elif badge.comparison == 'biggest':
                 order = u'-{}'.format(Badge.USER_ATTR_OPTS[badge.user_attr])
-                sqs = SearchQuerySet().filter(type='user')
+                sqs = self.search('user')
                 user = sqs.order_by(order)[0]
                 badge.awardees.add(User.objects.get(pk=user.pk))
                 continue
@@ -30,10 +39,7 @@ class Command(BaseCommand):
             )
             opts = {key: badge.value}
-            sqs = SearchQuerySet().filter(
-                type='user',
-                **opts
-            )
+            sqs = self.search('user', **opts)
             for user in sqs:
                 badge.awardees.add(User.objects.get(pk=user.pk))