Refactor

Scifabric · Apr 14, 2015 · df756da · df756da
1 parent d6377aa
commit df756da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pybossa/forms/forms.py b/pybossa/forms/forms.py
@@ -103,10 +103,9 @@ class TaskSchedulerForm(Form):
     sched = SelectField(lazy_gettext('Task Scheduler'), choices=_choices)
 
     @classmethod
-    def refresh_sched_options(cls):
-        from pybossa.sched import sched_variants
+    def update_sched_options(cls, new_options):
         _translate_names = lambda variant: (variant[0], lazy_gettext(variant[1]))
-        _choices = map(_translate_names, sched_variants())
+        _choices = map(_translate_names, new_options)
         cls.sched.kwargs['choices'] = _choices
 
 

diff --git a/pybossa/plugins/random_scheduler/__init__.py b/pybossa/plugins/random_scheduler/__init__.py
@@ -40,4 +40,4 @@ class RandomScheduler(Plugin):
     def setup(self):
         sched.new_task = with_random_scheduler(sched.new_task)
         sched.sched_variants = variants_with_random_scheduler(sched.sched_variants)
-        TaskSchedulerForm.refresh_sched_options()
+        TaskSchedulerForm.update_sched_options(sched.sched_variants())