mirror of https://gitlab.crans.org/bde/nk20
[WEI] Run algorithm only on valid surveys
Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
parent
881cd88f48
commit
ba9ef0371a
|
@ -170,6 +170,7 @@ class WEISurveyAlgorithm2021(WEISurveyAlgorithm):
|
||||||
We modify it to allow buses to have multiple "weddings".
|
We modify it to allow buses to have multiple "weddings".
|
||||||
"""
|
"""
|
||||||
surveys = list(self.get_survey_class()(r) for r in self.get_registrations()) # All surveys
|
surveys = list(self.get_survey_class()(r) for r in self.get_registrations()) # All surveys
|
||||||
|
surveys = [s for s in surveys if s.is_complete()]
|
||||||
free_surveys = [s for s in surveys if not s.information.valid] # Remaining surveys
|
free_surveys = [s for s in surveys if not s.information.valid] # Remaining surveys
|
||||||
while free_surveys: # Some students are not affected
|
while free_surveys: # Some students are not affected
|
||||||
survey = free_surveys[0]
|
survey = free_surveys[0]
|
||||||
|
|
|
@ -28,7 +28,8 @@ class Command(BaseCommand):
|
||||||
|
|
||||||
output = options['output']
|
output = options['output']
|
||||||
registrations = algorithm.get_registrations()
|
registrations = algorithm.get_registrations()
|
||||||
per_bus = {bus: [r for r in registrations if r.information['selected_bus_pk'] == bus.pk]
|
per_bus = {bus: [r for r in registrations if 'selected_bus_pk' in r.information
|
||||||
|
and r.information['selected_bus_pk'] == bus.pk]
|
||||||
for bus in algorithm.get_buses()}
|
for bus in algorithm.get_buses()}
|
||||||
for bus, members in per_bus.items():
|
for bus, members in per_bus.items():
|
||||||
output.write(bus.name + "\n")
|
output.write(bus.name + "\n")
|
||||||
|
|
Loading…
Reference in New Issue