[WEI] Reset cache after running algorithm

Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>

apps/wei/forms/surveys/base.py

@@ -223,3 +223,15 @@ class WEISurvey:
         self.information.selected_bus_pk = None
         self.information.selected_bus_name = None
         self.information.valid = False
+
+    @classmethod
+    def clear_cache(cls):
+        """
+        Clear stored information.
+        """
+        if hasattr(cls, '_wei'):
+            del cls._wei
+        if hasattr(cls.get_algorithm_class(), '_registrations'):
+            del cls.get_algorithm_class()._registrations
+        if hasattr(cls.get_algorithm_class(), '_buses'):
+            del cls.get_algorithm_class()._buses

apps/wei/forms/surveys/wei2021.py

@@ -169,6 +169,11 @@ class WEISurvey2021(WEISurvey):
         values.sort(key=lambda item: -item[1])
         return values
 
+    @classmethod
+    def clear_cache(cls):
+        cls.word_mean.cache_clear()
+        return super().clear_cache()
+
 
 class WEISurveyAlgorithm2021(WEISurveyAlgorithm):
     """
@@ -240,6 +245,9 @@ class WEISurveyAlgorithm2021(WEISurveyAlgorithm):
         if display_tqdm:
             tqdm_obj.close()
 
+        # Clear cache information after running algorithm
+        WEISurvey2021.clear_cache()
+
     def make_repartition(self, surveys, quotas=None, tqdm_obj=None):
         free_surveys = surveys.copy()  # Remaining surveys
         while free_surveys:  # Some students are not affected