# -*- coding: utf-8 -*-

from model.Study import Study
from view.ASubWindow import ASubWindow

class NewStudyWindow(ASubWindow):
    def __init__(self, title="New Study", parent=None):
        super(NewStudyWindow, self).__init__(name=title, ui="NewStudy", parent=parent)
        self.ui.setWindowTitle(title)

        self.parent = parent

    def accept(self):
        name = self.get_line_edit_text("lineEdit_name")
        description = self.get_text_edit_text("textEdit_description")
        if self.get_radio_button("radioButton_mage"):
            solver = "mage"
        else:
            solver = "rubarbe"

        self.parent.set_model(Study.new(name, description, solver))
        self.close()