From: Bryan <btbonval@gmail.com>
Date: Fri, 9 Jan 2015 22:33:10 +0000 (-0500)
Subject: regularly schedule file count updates for #377
X-Git-Tag: release-20150131~4
X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=321c9fd8be8eb3776ee56f628d2918d689ac4e2c;p=oweals%2Fkarmaworld.git

regularly schedule file count updates for #377
---

diff --git a/karmaworld/apps/courses/management/commands/fix_note_counts.py b/karmaworld/apps/courses/management/commands/fix_note_counts.py
index 1b896ab..3197fd2 100644
--- a/karmaworld/apps/courses/management/commands/fix_note_counts.py
+++ b/karmaworld/apps/courses/management/commands/fix_note_counts.py
@@ -1,18 +1,9 @@
 from django.core.management.base import BaseCommand
-from karmaworld.apps.notes.models import *
-from karmaworld.apps.courses.models import *
+from karmaworld.apps.courses.tasks import fix_note_counts
 
 class Command(BaseCommand):
     help = """Set the field file_count on every Course and School
             to the correct value."""
 
     def handle(self, *args, **kwargs):
-        for c in Course.objects.all():
-            c.update_note_count()
-            print "Updated course {c}".format(c=c)
-
-        for s in School.objects.all():
-            s.update_note_count()
-            print "Updated school {s}".format(s=s)
-
-
+        fix_note_counts()
diff --git a/karmaworld/apps/courses/tasks.py b/karmaworld/apps/courses/tasks.py
new file mode 100644
index 0000000..1674bff
--- /dev/null
+++ b/karmaworld/apps/courses/tasks.py
@@ -0,0 +1,21 @@
+from karmaworld.apps.courses.models import Course
+from karmaworld.apps.courses.models import School
+
+from celery import task
+from celery.utils.log import get_task_logger
+
+logger = get_task_logger(__name__)
+
+@task(name="fix_note_counts")
+def fix_note_counts():
+    """
+    Set the field file_count on every Course and School to the correct value.
+    """
+
+    for c in Course.objects.all():
+        c.update_note_count()
+        print "Updated course {c}".format(c=c)
+
+    for s in School.objects.all():
+        s.update_note_count()
+        print "Updated school {s}".format(s=s)
diff --git a/karmaworld/settings/prod.py b/karmaworld/settings/prod.py
index b83cb9b..4329cdc 100644
--- a/karmaworld/settings/prod.py
+++ b/karmaworld/settings/prod.py
@@ -93,6 +93,10 @@ CELERYBEAT_SCHEDULE = {
         'task': 'get_extract_keywords_results',
         'schedule': timedelta(minutes=20),
     },
+    'update-scoreboard': {
+        'task': 'fix_note_counts',
+        'schedule': timedelta(days=1),
+    },
 }
 
 CELERY_TIMEZONE = 'UTC'