Files
SaveWizard/module_config_editor/script.py
Lev Rusanov 6330a0dfa8 Update
* Implemented show_progress_bar and update_progress_bar to interact with progress bars
* Added 3 progress bars to second window
* Imports little optimization

Signed-off-by: Lev Rusanov <30170278+JDM170@users.noreply.github.com>
2024-06-17 08:26:19 +07:00

112 lines
4.4 KiB
Python

#!/usr/bin/python3
# -*- coding: utf-8 -*-
from PyQt5.QtGui import QResizeEvent, QCloseEvent, QClipboard
from PyQt5.QtWidgets import QMainWindow, QFileDialog, QMessageBox, QApplication
from ast import literal_eval
from .form import Ui_MainWindow
from dataIO import dataIO
from util import util
class EditorWindow(QMainWindow, Ui_MainWindow):
def __init__(self, parent=None):
QMainWindow.__init__(self, parent, flags=Qt.Window)
Ui_MainWindow.__init__(self)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.file_path = ""
self.win_title = self.tr("{}".format(self.windowTitle()))
self.ui.textEdit.document().contentsChanged.connect(self.content_changed)
for action, method in {
self.ui.actionOpen: self.open_file,
self.ui.actionSave: self.save_file,
self.ui.actionSave_As: self.save_as,
self.ui.actionMD5: self.copy_hash,
self.ui.actionCloseFile: self.close_file,
self.ui.actionExit: self.exit
}.items():
action.triggered.connect(method)
def resizeEvent(self, event: QResizeEvent):
window = self.ui.centralwidget.geometry().getCoords()
edit = self.ui.textEdit.geometry().getCoords()
self.ui.textEdit.setGeometry(edit[0], edit[1],
window[2], window[3]-(self.ui.menubar.size().height()-1))
def closeEvent(self, event: QCloseEvent):
if self.maybe_save():
event.accept()
else:
event.ignore()
def content_changed(self):
self.setWindowModified(self.ui.textEdit.document().isModified())
def open_file(self):
file_path, file_name = QFileDialog.getOpenFileName(parent=self,
caption=self.tr("Choose config..."),
filter=self.tr("*.json"))
if file_path != "":
self.file_path = file_path
with open(file_path) as f:
self.ui.textEdit.setPlainText(f.read())
self.setWindowTitle(self.tr("[*]{} - {}".format(file_path, self.win_title)))
self.ui.textEdit.document().setModified(False)
self.setWindowModified(False)
def save_file(self):
if self.file_path != "":
data = literal_eval(self.ui.textEdit.toPlainText())
ret = dataIO.save_json(self.file_path, data)
if ret:
self.ui.textEdit.document().setModified(False)
self.setWindowModified(False)
return ret
def save_as(self):
file_path, file_name = QFileDialog.getSaveFileName(parent=self,
caption=self.tr("Select place to save file"),
filter=self.tr("*.json"))
file_data = literal_eval(self.ui.textEdit.toPlainText())
ret = dataIO.save_json(file_path, file_data)
if ret:
self.file_path = file_path
self.setWindowTitle(self.tr("[*]{} - {}".format(file_path, self.win_title)))
self.ui.textEdit.document().setModified(False)
self.setWindowModified(False)
def copy_hash(self):
if self.maybe_save():
result = util.generate_md5(self.file_path)
clip = QApplication.clipboard()
QClipboard.setText(clip, result)
QMessageBox.information(self, "Information", "Hash successfully copied into your clipboard.")
def close_file(self):
if self.maybe_save():
self.setWindowTitle(self.win_title)
self.ui.textEdit.clear()
self.file_path = ""
def exit(self):
self.close()
def maybe_save(self):
if not self.ui.textEdit.document().isModified():
return True
box = QMessageBox.warning(self,
self.tr("SaveWizard config editor"),
self.tr("The document has been modified.\nDo you want to save your changes?"),
QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel)
if box == QMessageBox.Save:
return self.save_file()
elif box == QMessageBox.Discard:
return True
elif box == QMessageBox.Cancel:
return False
return True