1 files changed, 58 insertions, 0 deletions
diff --git a/forum/management/commands/clean_award_badges.py b/forum/management/commands/clean_award_badges.py
new file mode 100644
index 00000000..df3d2917
--- /dev/null
+++ b/forum/management/commands/clean_award_badges.py
@@ -0,0 +1,58 @@
+#-------------------------------------------------------------------------------
+# Name:        Award badges command
+# Purpose:     This is a command file croning in background process regularly to
+#              query database and award badges for user's special acitivities.
+#
+# Author:      Mike
+#
+# Created:     18/01/2009
+# Copyright:   (c) Mike 2009
+# Licence:     GPL V2
+#-------------------------------------------------------------------------------
+#!/usr/bin/env python
+#encoding:utf-8
+from django.core.management.base import NoArgsCommand
+from django.db import connection
+from django.shortcuts import get_object_or_404
+from django.contrib.contenttypes.models import ContentType
+
+from forum.models import *
+
+class Command(NoArgsCommand):
+    def handle_noargs(self, **options):
+        try:
+            self.clean_awards()
+        except Exception, e:
+            print e
+        finally:
+            connection.close()
+
+    def clean_awards(self):
+        Award.objects.all().delete()
+
+        award_type =ContentType.objects.get_for_model(Award)
+        Activity.objects.filter(content_type=award_type).delete()
+
+        for user in User.objects.all():
+            user.gold = 0
+            user.silver = 0
+            user.bronze = 0
+            user.save()
+
+        for badge in Badge.objects.all():
+            badge.awarded_count = 0
+            badge.save()
+            
+        query = "UPDATE activity SET is_auditted = 0"
+        cursor = connection.cursor()
+        try:
+            cursor.execute(query)
+        finally:
+            cursor.close()
+            connection.close()
+
+def main():
+    pass
+
+if __name__ == '__main__':
+    main()
+\ No newline at end of file