157 lines
5.4 KiB
Python
157 lines
5.4 KiB
Python
import json
|
|
import time
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
from PyQt6.QtCore import pyqtSignal
|
|
from PyQt6.QtWidgets import QFrame, QVBoxLayout, QHBoxLayout, QPushButton
|
|
|
|
from source.survey.base import BaseSurvey
|
|
from source.survey import Empty, survey_get
|
|
|
|
|
|
result_path = Path("./result/")
|
|
result_path.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
|
class FrameSurvey(QFrame):
|
|
signal_abandon = pyqtSignal()
|
|
signal_skip = pyqtSignal()
|
|
signal_success = pyqtSignal()
|
|
|
|
def __init__(self, survey_path: Path | str):
|
|
# TODO: translation support
|
|
super().__init__()
|
|
|
|
# signals
|
|
self.signal_abandon.connect(self._on_signal_abandon) # NOQA: connect exist
|
|
self.signal_skip.connect(self._on_signal_skip) # NOQA: connect exist
|
|
self.signal_success.connect(self._on_signal_success) # NOQA: connect exist
|
|
|
|
# prepare the survey collected data
|
|
self.collected_datas: dict[str, dict] = {"time": time.time(), "surveys": {}}
|
|
self.survey_screens: list[tuple[str, BaseSurvey]] = []
|
|
self.current_survey_index = 0
|
|
|
|
# set the layout
|
|
self._layout = QVBoxLayout()
|
|
self.setLayout(self._layout)
|
|
|
|
# prepare the frame for the survey elements
|
|
self.frame_survey: BaseSurvey = Empty()
|
|
self._layout.addWidget(self.frame_survey)
|
|
|
|
# navigations actions
|
|
self.frame_navigation = QFrame()
|
|
self._layout.addWidget(self.frame_navigation)
|
|
|
|
self._layout_navigation = QHBoxLayout()
|
|
self.frame_navigation.setLayout(self._layout_navigation)
|
|
|
|
self._layout_navigation.addStretch(0) # add a stretch to put the buttons on the right
|
|
|
|
self.button_abandon = QPushButton()
|
|
self._layout_navigation.addWidget(self.button_abandon)
|
|
self.button_abandon.setText("Abandonner")
|
|
self.button_abandon.setStyleSheet("QPushButton { color : red; }")
|
|
self.button_abandon.clicked.connect(self.quit) # NOQA: connect exist
|
|
|
|
self.button_skip = QPushButton()
|
|
self._layout_navigation.addWidget(self.button_skip)
|
|
self.button_skip.setText("Passer")
|
|
self.button_skip.clicked.connect(self.next_survey) # NOQA: connect exist
|
|
|
|
self.button_forward = QPushButton()
|
|
self._layout_navigation.addWidget(self.button_forward)
|
|
self.button_forward.setText("Suivant")
|
|
self.button_forward.clicked.connect(self.next_survey) # NOQA: connect exist
|
|
|
|
# load the survey configuration file
|
|
self.load_file(survey_path)
|
|
|
|
# finalize the initialisation
|
|
self.update_survey()
|
|
|
|
def _on_signal_abandon(self):
|
|
# on success, show the button to give up
|
|
self.button_abandon.show()
|
|
|
|
def _on_signal_skip(self):
|
|
# on success, show the button to skip
|
|
self.button_skip.show()
|
|
|
|
def _on_signal_success(self):
|
|
# on success, show the button to go forward
|
|
self.button_forward.show()
|
|
|
|
def load_file(self, survey_path: Path | str):
|
|
# load the surveys screens
|
|
with open(survey_path, encoding="utf-8") as file:
|
|
surveys_data = json.load(file)
|
|
|
|
self.survey_screens = [
|
|
(
|
|
survey_id,
|
|
survey_get(
|
|
survey_data,
|
|
signals={
|
|
"abandon": self.signal_abandon,
|
|
"skip": self.signal_skip,
|
|
"success": self.signal_success,
|
|
}
|
|
)
|
|
)
|
|
for survey_id, survey_data in surveys_data["surveys"].items()
|
|
]
|
|
self.current_survey_index = 0
|
|
|
|
def next_survey(self):
|
|
# get the collected data from the survey
|
|
collected_data = self.frame_survey.get_collected_data()
|
|
if collected_data is not None:
|
|
# if there is data, get the current survey id
|
|
survey_id, survey = self.survey_screens[self.current_survey_index]
|
|
# save the response in the data
|
|
self.collected_datas["surveys"][survey_id] = collected_data
|
|
|
|
self.current_survey_index += 1
|
|
|
|
if self.current_survey_index < len(self.survey_screens):
|
|
self.update_survey()
|
|
else:
|
|
self.finish_survey()
|
|
|
|
def update_survey(self):
|
|
# disable the buttons
|
|
self.button_abandon.hide()
|
|
self.button_skip.hide()
|
|
self.button_forward.hide()
|
|
|
|
# mark the actual survey as the old one
|
|
old_frame_survey = self.frame_survey
|
|
# call the old survey event
|
|
old_frame_survey.on_hide()
|
|
# get the currently selected survey
|
|
survey_id, survey = self.survey_screens[self.current_survey_index]
|
|
# update it to the new one
|
|
self.frame_survey = survey
|
|
# change the widget on the layout
|
|
self._layout.replaceWidget(old_frame_survey, self.frame_survey)
|
|
# adjust the size of the widgets
|
|
self.window().adjustSize()
|
|
# call the new survey event
|
|
survey.on_show()
|
|
# delete the old frame
|
|
old_frame_survey.deleteLater()
|
|
|
|
def finish_survey(self):
|
|
# save the result in a json file
|
|
with open(result_path / f"{uuid.uuid4()}.json", "w", encoding="utf-8") as file:
|
|
json.dump(self.collected_datas, file, ensure_ascii=False)
|
|
|
|
self.quit()
|
|
|
|
def quit(self):
|
|
# quit the application by closing and deleting the window
|
|
self.window().close()
|
|
self.window().deleteLater()
|