Add clear_all class method to LocalScheduler

2012-08-01 04:15:43 -04:00
parent f43f496241
commit dd00296c0a
2 changed files with 10 additions and 1 deletions
--- a/apps/scheduler/api.py
+++ b/apps/scheduler/api.py
@@ -72,6 +72,11 @@ class LocalScheduler(object):
    def lockdown(cls):
        cls._lockdown = True

+    @classmethod
+    def clear_all(cls):
+        for scheduler in cls.scheduler_registry.values():
+            scheduler.clear()
+
    def __init__(self, name, label=None):
        self.scheduled_jobs = {}
        self._scheduler = None
@@ -108,7 +113,9 @@ class LocalScheduler(object):
            self.stop_job(job)

    def stop_job(self, job):
-        self._scheduler.unschedule_job(job._job)
+        if self.running:
+            self._scheduler.unschedule_job(job._job)
+
        del(self.scheduled_jobs[job.name])
        job.scheduler = None