class ModelEditor:
"""Model editor main class"""
def __init__(self):
# load config
cfg.init(None)
# main window
self._app = QtWidgets.QApplication(sys.argv)
self._app.setWindowIcon(icon.get_app_icon("me-geomap"))
self.mainwindow = MainWindow(self)
cfg.main_window = self.mainwindow
# set default values
self._update_document_name()
# show
self.mainwindow.show()
def new_file(self):
"""new file menu action"""
if not self.save_old_file():
return
cfg.new_file()
self.mainwindow.reload()
self.mainwindow.update_recent_files(0)
self._update_document_name()
self.mainwindow.info.update_from_data({'record_id': cfg.root_input_type['id']})
self.mainwindow.show_status_message("New file is opened")
def open_file(self):
"""open file menu action"""
if not self.save_old_file():
return
yaml_file = QtWidgets.QFileDialog.getOpenFileName(
self.mainwindow, "Choose Yaml Model File",
cfg.config.last_data_dir, "Yaml Files (*.yaml)")
if yaml_file[0]:
cfg.open_file(yaml_file[0])
self.mainwindow.reload()
self.mainwindow.update_recent_files()
self._update_document_name()
self.mainwindow.show_status_message("File '" + yaml_file[0] + "' is opened")
def import_file(self):
"""import con file menu action"""
if not self.save_old_file():
return
con_file = QtWidgets.QFileDialog.getOpenFileName(
self.mainwindow, "Choose Con Model File",
cfg.config.last_data_dir, "Con Files (*.con)")
if con_file[0]:
cfg.import_file(con_file[0])
self.mainwindow.reload()
self.mainwindow.update_recent_files()
self._update_document_name()
self.mainwindow.show_status_message("File '" + con_file[0] + "' is imported")
def open_recent(self, action):
"""open recent file menu action"""
if action.text() == cfg.curr_file:
return
if not self.save_old_file():
return
cfg.open_recent_file(action.text())
self.mainwindow.reload()
self.mainwindow.update_recent_files()
self._update_document_name()
self.mainwindow.show_status_message("File '" + action.text() + "' is opened")
def save_file(self):
"""save file menu action"""
if cfg.curr_file is None:
return self.save_as()
cfg.update_yaml_file(self.mainwindow.editor.text())
cfg.save_file()
self.mainwindow.show_status_message("File is saved")
def save_as(self):
"""save file menu action"""
cfg.update_yaml_file(self.mainwindow.editor.text())
if cfg.curr_file is None:
if cfg.imported_file_name is not None:
new_file = cfg.imported_file_name
else:
new_file = cfg.config.last_data_dir + os.path.sep + "NewFile.yaml"
else:
new_file = cfg.curr_file
dialog = QtWidgets.QFileDialog(self.mainwindow, 'Save as YAML File', new_file,
"YAML Files (*.yaml)")
dialog.setDefaultSuffix('.yaml')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setViewMode(QtWidgets.QFileDialog.Detail)
if dialog.exec_():
file_name = dialog.selectedFiles()[0]
cfg.save_as(file_name)
self.mainwindow.update_recent_files()
self._update_document_name()
self.mainwindow.show_status_message("File is saved")
return True
return False
def transform(self, file):
"""Run transformation according rules in set file"""
cfg.update_yaml_file(self.mainwindow.editor.text())
cfg.transform(file)
# synchronize cfg document with editor text
self.mainwindow.editor.setText(cfg.document, keep_history=True)
self.mainwindow.reload()
def edit_transformation_file(self, file):
"""edit transformation rules in file"""
text = cfg.get_transformation_text(file)
if text is not None:
dlg = JsonEditorDlg(cfg.transformation_dir, file,
"Transformation rules:", text, self.mainwindow)
dlg.exec_()
def select_format(self, filename):
"""Selects format file by filename."""
if cfg.curr_format_file == filename:
return
cfg.curr_format_file = filename
cfg.update_format()
self.mainwindow.reload()
self.mainwindow.show_status_message("Format of file is changed")
def edit_format(self):
"""Open selected format file in Json Editor"""
text = cfg.get_curr_format_text()
if text is not None:
dlg = JsonEditorDlg(cfg.format_dir, cfg.curr_format_file,
"Format", text, self.mainwindow)
dlg.exec_()
def _update_document_name(self):
"""Update document title (add file name)"""
title = "GeoMop Model Editor"
if cfg.curr_file is None:
title += " - New File"
else:
title += " - " + cfg.curr_file
self.mainwindow.setWindowTitle(title)
def save_old_file(self):
"""
If file not saved, display confirmation dialeg and if is needed, do it
return: False if action have to be aborted
"""
cfg.update_yaml_file(self.mainwindow.editor.text())
if cfg.changed:
msg_box = QMessageBox()
msg_box.setWindowTitle("Confirmation")
msg_box.setIcon(QMessageBox.Question)
msg_box.setStandardButtons(QMessageBox.Yes | QMessageBox.No | QMessageBox.Abort)
msg_box.setDefaultButton(QMessageBox.Abort)
msg_box.setText("The document has unsaved changes, do you want to save it?")
reply = msg_box.exec_()
if reply == QtWidgets.QMessageBox.Abort:
return False
if reply == QtWidgets.QMessageBox.Yes:
if cfg.curr_file is None:
return self.save_as()
else:
self.save_file()
return True
def main(self):
"""go"""
self._app.exec_()
