web/lib/django_extensions/management/commands/runjobs.py
changeset 3 526ebd3988b0
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/lib/django_extensions/management/commands/runjobs.py	Wed Jan 20 12:37:40 2010 +0100
@@ -0,0 +1,90 @@
+from django.core.management.base import LabelCommand
+from optparse import make_option
+from django_extensions.management.jobs import get_jobs, print_jobs
+
+class Command(LabelCommand):
+    option_list = LabelCommand.option_list + (
+        make_option('--list', '-l', action="store_true", dest="list_jobs",
+            help="List all jobs with their description"),
+    )
+    help = "Runs scheduled maintenance jobs."
+    args = "[hourly daily weekly monthly]"
+    label = ""
+
+    requires_model_validation = True
+
+    def usage_msg(self):
+        print "Run scheduled jobs. Please specify 'hourly', 'daily', 'weekly' or 'monthly'"
+
+    def runjobs(self, when, options):
+        verbosity = int(options.get('verbosity', 1))
+        jobs = get_jobs(when, only_scheduled=True)
+        list = jobs.keys()
+        list.sort()
+        for app_name, job_name in list:
+            job = jobs[(app_name, job_name)]
+            if verbosity>1:
+                print "Executing %s job: %s (app: %s)" % (when, job_name, app_name)
+            try:
+                job().execute()
+            except Exception, e:
+                import traceback
+                print "ERROR OCCURED IN %s JOB: %s (APP: %s)" % (when.upper(), job_name, app_name)
+                print "START TRACEBACK:"
+                traceback.print_exc()
+                print "END TRACEBACK\n"
+
+    def runjobs_by_signals(self, when, options):
+        """ Run jobs from the signals """
+        # Thanks for Ian Holsman for the idea and code
+        from django_extensions.management import signals
+        from django.db import models
+        from django.conf import settings
+
+        verbosity = int(options.get('verbosity', 1))
+        for app_name in settings.INSTALLED_APPS:
+            try:
+                __import__(app_name + '.management', '', '', [''])
+            except ImportError:
+                pass
+
+        for app in models.get_apps():
+            if verbosity>1:
+                app_name = '.'.join(app.__name__.rsplit('.')[:-1])
+                print "Sending %s job signal for: %s" % (when, app_name)
+            if when == 'hourly':
+                signals.run_hourly_jobs.send(sender=app, app=app)
+            elif when == 'daily':
+                signals.run_daily_jobs.send(sender=app, app=app)
+            elif when == 'weekly':
+                signals.run_weekly_jobs.send(sender=app, app=app)
+            elif when == 'monthly':
+                signals.run_monthly_jobs.send(sender=app, app=app)
+
+    def handle(self, *args, **options):
+        when = None
+        if len(args)>1:
+            self.usage_msg()
+            return
+        elif len(args)==1:
+            if not args[0] in ['hourly', 'daily', 'weekly', 'monthly']:
+                self.usage_msg()
+                return
+            else:
+                when = args[0]
+        if options.get('list_jobs'):
+            print_jobs(when, only_scheduled=True, show_when=True, show_appname=True)
+        else:
+            if not when:
+                self.usage_msg()
+                return
+            self.runjobs(when, options)
+            self.runjobs_by_signals(when, options)
+
+# Backwards compatibility for Django r9110
+if not [opt for opt in Command.option_list if opt.dest=='verbosity']:
+    Command.option_list += (
+        make_option('--verbosity', '-v', action="store", dest="verbosity",
+            default='1', type='choice', choices=['0', '1', '2'],
+            help="Verbosity level; 0=minimal output, 1=normal output, 2=all output"),
+    )