nmreval/src/gui_qt/editors/script_editor.py

from __future__ import annotations

from pathlib import Path

from ..Qt import QtWidgets, QtCore, QtGui
from .codeeditor import EditorWidget


class QEditor(QtWidgets.QMainWindow):
    runSignal = QtCore.pyqtSignal(str)

    def __init__(self, path: str | Path = None, parent=None):
        super().__init__(parent=parent)

        self._init_gui()

        self.fname = None
        self._dir = None
        if path is not None:
            self.read_file(path)

    def _init_gui(self):
        self.centralwidget = QtWidgets.QWidget(self)

        layout = QtWidgets.QVBoxLayout(self.centralwidget)
        layout.setContentsMargins(3, 3, 3, 3)
        layout.setSpacing(3)

        self.edit_field = EditorWidget(self.centralwidget)
        font = QtGui.QFont('default')
        font.setStyleHint(font.Monospace)
        font.setPointSize(10)
        self.edit_field.setFont(font)

        layout.addWidget(self.edit_field)
        self.setCentralWidget(self.centralwidget)

        self.statusbar = QtWidgets.QStatusBar(self)
        self.setStatusBar(self.statusbar)

        self.menubar = self.menuBar()

        self.menuFile = QtWidgets.QMenu('File', self.menubar)
        self.menubar.addMenu(self.menuFile)

        self.menuFile.addAction('Open...', self.open_file, QtGui.QKeySequence.Open)
        self.menuFile.addAction('Save', self.overwrite_file, QtGui.QKeySequence.Save)
        self.menuFile.addAction('Save as...', self.save_file, QtGui.QKeySequence('Ctrl+Shift+S'))
        self.menuFile.addSeparator()
        self.menuFile.addAction('Close', self.close, QtGui.QKeySequence.Quit)

        self.resize(800, 600)
        self.setGeometry(
            QtWidgets.QStyle.alignedRect(
                QtCore.Qt.LayoutDirection.LeftToRight,
                QtCore.Qt.AlignmentFlag.AlignCenter,
                self.size(),
                QtWidgets.qApp.desktop().availableGeometry()
            )
        )

    def is_modified(self):
        return self.edit_field.editor.document().isModified()

    def set_modified(self, val: bool):
        self.edit_field.editor.document().setModified(val)

    @QtCore.pyqtSlot()
    def open_file(self):
        overwrite = self.changes_saved

        if overwrite:
            fname, _ = QtWidgets.QFileDialog.getOpenFileName(directory=str(self._dir))
            if fname:
                self.read_file(fname)

    def read_file(self, fname: str | Path):
        self.set_fname_opts(fname)

        if self.fname is not None:
            with self.fname.open('r') as f:
                self.edit_field.setPlainText(f.read())

    def set_fname_opts(self, fname: str | Path):
        fname = Path(fname)
        if fname.is_file():
            self.fname = Path(fname)
            self._dir = self.fname.parent
        elif fname.is_dir():
            self._dir = fname
        self.setWindowTitle('Edit ' + str(fname))

    def changes_saved(self) -> bool:
        if not self.is_modified():
            return True

        ret = QtWidgets.QMessageBox.question(self, 'Time to think',
                                             '<h4><p>The document was modified.</p>\n'
                                             '<p>Do you want to save changes?</p></h4>',
                                             QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No |
                                             QtWidgets.QMessageBox.Cancel)
        if ret == QtWidgets.QMessageBox.Yes:
            self.save_file()

        if ret == QtWidgets.QMessageBox.No:
            self.set_modified(False)

        return not self.is_modified()

    @QtCore.pyqtSlot()
    def save_file(self):
        outfile, _ = QtWidgets.QFileDialog().getSaveFileName(parent=self, caption='Save file', directory=str(self._dir))

        if outfile:
            with open(outfile, 'w') as f:
                f.write(self.edit_field.toPlainText())

            self.set_fname_opts(outfile)

            self.set_modified(False)

        return self.is_modified()

    @QtCore.pyqtSlot()
    def overwrite_file(self):
        if self.fname is not None:
            with self.fname.open('w') as f:
                f.write(self.edit_field.toPlainText())

        self.modelsChanged.emit()

        self.set_modified(False)

    def closeEvent(self, evt: QtGui.QCloseEvent):
        self.runSignal.emit(self.edit_field.toPlainText())

        super().closeEvent(evt)