mirror of
https://gitlab.crans.org/bde/nk20
synced 2025-02-06 15:23:00 +00:00
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
|
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
|
||
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
|
||
|
|
from django import forms
|
||
|
|
|
||
|
|
from .base import WEISurvey, WEISurveyInformation, WEISurveyAlgorithm
|
||
|
|
from ...models import Bus
|
||
|
|
|
||
|
|
|
||
|
|
class WEISurveyForm2020(forms.Form):
|
||
|
|
bus = forms.ModelChoiceField(
|
||
|
|
Bus.objects,
|
||
|
|
)
|
||
|
|
|
||
|
|
def set_registration(self, registration):
|
||
|
|
self.fields["bus"].queryset = Bus.objects.filter(wei=registration.wei)
|
||
|
|
|
||
|
|
|
||
|
|
class WEISurveyInformation2020(WEISurveyInformation):
|
||
|
|
chosen_bus_pk = None
|
||
|
|
|
||
|
|
|
||
|
|
class WEISurvey2020(WEISurvey):
|
||
|
|
year = 2020
|
||
|
|
|
||
|
|
def get_survey_information_class(self):
|
||
|
|
return WEISurveyInformation2020
|
||
|
|
|
||
|
|
def get_form_class(self):
|
||
|
|
return WEISurveyForm2020
|
||
|
|
|
||
|
|
def update_form(self, form):
|
||
|
|
form.set_registration(self.registration)
|
||
|
|
|
||
|
|
def form_valid(self, form):
|
||
|
|
self.information.chosen_bus_pk = form.cleaned_data["bus"].pk
|
||
|
|
self.save()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_algorithm_class():
|
||
|
|
return WEISurveyAlgorithm2020
|
||
|
|
|
||
|
|
|
||
|
|
class WEISurveyAlgorithm2020(WEISurveyAlgorithm):
|
||
|
|
def get_survey_class(self):
|
||
|
|
return WEISurvey2020
|
||
|
|
|
||
|
|
def run_algorithm(self):
|
||
|
|
for registration in self.get_registrations():
|
||
|
|
survey = self.get_survey_class()(registration)
|
||
|
|
survey.select_bus(survey.information.chosen_bus_pk)
|
||
|
|
survey.save()
|