From 3f99ec70da0fcaad1aa7cf63d3ee3fca8d3794b1 Mon Sep 17 00:00:00 2001 From: Douglas Paz Date: Wed, 26 Sep 2018 15:04:24 -0300 Subject: [PATCH] Fix ready_for_train updates --- bothub/common/models.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/bothub/common/models.py b/bothub/common/models.py index 483341bf..56b46e3c 100644 --- a/bothub/common/models.py +++ b/bothub/common/models.py @@ -168,25 +168,31 @@ def languages_status(self): settings.SUPPORTED_LANGUAGES.keys(), )) + @property + def current_updates(self): + return map( + lambda lang: self.current_update(lang), + self.available_languages) + @property def requirements_to_train(self): return dict(filter( lambda l: l[1], map( lambda u: (u.language, u.requirements_to_train,), - self.updates.filter(training_started_at__isnull=True)))) + self.current_updates))) @property def languages_ready_for_train(self): return dict(map( lambda u: (u.language, u.ready_for_train,), - self.updates.filter(training_started_at__isnull=True))) + self.current_updates)) @property def ready_for_train(self): return reduce( lambda current, u: u.ready_for_train or current, - self.updates.filter(training_started_at__isnull=True), + self.current_updates, False) @property