-
Notifications
You must be signed in to change notification settings - Fork 0
/
MainWindow.py
132 lines (116 loc) · 5.31 KB
/
MainWindow.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
from functools import partial
from forms.mainwindow import Ui_MainMenu
from forms.loadingscreen import Ui_LoadScreenWindow
from PyQt6 import QtCore
from PyQt6.QtWidgets import QMainWindow, QLabel, QPushButton, QFileDialog, QMessageBox, QWidget
from PyQt6.QtCore import QThread, Qt, pyqtSlot
from BiPackage import BiPackage
from PackageCreator import PackageCreator
from controllers.FileController import FileController
from controllers.FolderController import FolderController
from controllers.PackageController import PackageController
from Elements import Element
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.msg = LoadingScreen()
self.ui = Ui_MainMenu()
self.ui.setupUi(self)
self.packageElement = None
self.biPackage = None
self.creator = None
self.ui.addButton.clicked.connect(self.addElement)
self.ui.createPackageButton.clicked.connect(self.createPackage)
self.ui.saveChangesButton.clicked.connect(self.saveChanges)
self.ui.cancelButton.clicked.connect(self.cancel)
self.ui.elementType.activated.connect(self.typeSelected)
self.worker = QThread()
def typeSelected(self):
curOption = self.ui.elementType.currentText()
if curOption == 'Select type':
self.ui.addButton.setEnabled(False)
else:
self.ui.addButton.setEnabled(True)
def addElement(self):
curOption = self.ui.elementType.currentText()
if curOption == 'File':
self.packageElement = FileController()
if curOption == 'Folder':
self.packageElement = FolderController()
if curOption == 'Package':
self.packageElement = PackageController()
self.packageElement.newElement.connect(self.addElementGui)
self.packageElement.show()
def createPackage(self):
self.biPackage = BiPackage()
self.ui.elementType.setEnabled(True)
self.ui.saveChangesButton.setEnabled(True)
self.ui.cancelButton.setEnabled(True)
self.ui.createPackageButton.setEnabled(False)
self.ui.biPackageName.setEnabled(True)
self.ui.biPackageVersion.setEnabled(True)
def saveChanges(self):
name = self.ui.biPackageName.text()
version = self.ui.biPackageVersion.text()
if len(self.biPackage.getElements()) == 0:
QMessageBox().critical(self, "Error", "Suitcase cannot be empty")
elif not name or name.isspace():
QMessageBox().critical(self, "Error", "Suitcase should have a name")
elif not version or version.isspace():
QMessageBox().critical(self, "Error", "Suitcase should have a version")
else:
path = QFileDialog.getSaveFileName(self, 'Save File', name, '.bi')[0]
if path != '':
self.biPackage.setName(name)
self.biPackage.setVersion(version)
self.creator = PackageCreator(self.biPackage, path)
self.worker.started.connect(self.msg.show)
self.worker.finished.connect(self.creator.deleteLater)
self.worker.started.connect(self.creator.start)
self.creator.finished.connect(self.worker.quit)
self.worker.finished.connect(self.msg.close)
self.worker.finished.connect(self.creator.deleteLater)
self.worker.finished.connect(self.clear)
self.creator.moveToThread(self.worker)
self.worker.start()
def cancel(self):
self.clear()
def clear(self):
self.ui.elementType.setEnabled(False)
self.ui.addButton.setEnabled(False)
self.ui.saveChangesButton.setEnabled(False)
self.ui.cancelButton.setEnabled(False)
self.ui.createPackageButton.setEnabled(True)
self.ui.biPackageName.setEnabled(False)
self.ui.biPackageVersion.setEnabled(False)
self.ui.elementType.setCurrentIndex(0)
for i in reversed(range(self.ui.elements.count())):
self.ui.elements.itemAt(i).widget().deleteLater()
self.biPackage.deleteAllElements()
self.ui.biPackageVersion.clear()
self.ui.biPackageName.clear()
def uploadPackageToGui(self):
for i in reversed(range(self.ui.elements.count())):
self.ui.elements.itemAt(i).widget().deleteLater()
elements = self.biPackage.getElements()
for i in range(len(elements)):
(elementType, element), = elements[i].items()
self.ui.elements.addWidget(QLabel(element.description), i, 0)
btn = QPushButton('Remove')
btn.clicked.connect(partial(self.biPackage.deleteElement, i))
btn.clicked.connect(self.uploadPackageToGui)
self.ui.elements.addWidget(btn, i, 1)
@pyqtSlot(Element)
def addElementGui(self, element):
self.biPackage.addElement(element)
self.uploadPackageToGui()
self.packageElement.newElement.disconnect()
self.ui.elementType.setCurrentIndex(0)
class LoadingScreen(QWidget):
def __init__(self):
super().__init__()
self.ui = Ui_LoadScreenWindow()
self.ui.setupUi(self)
self.setWindowFlags(Qt.WindowType.WindowStaysOnTopHint |
Qt.WindowType.CustomizeWindowHint |
Qt.WindowType.FramelessWindowHint)