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
133
134
135
136
137
138
139
140
141
142
143
144
|
import pickle
import sys
from fbs_runtime.application_context.PyQt5 import ApplicationContext
from PyQt5.QtCore import QObject, Qt, pyqtSignal
from PyQt5.QtGui import QBrush, QColor, QImage, QPainter, QPalette
from PyQt5.QtWidgets import (QComboBox, QFileDialog, QFormLayout, QVBoxLayout,
QHBoxLayout, QLabel, QMainWindow, QMenu,
QPushButton, QWidget, QMdiArea)
from utils.canvas import canvas
from utils.fileWindow import fileWindow
from utils.sizes import ppiList, sheetDimensionList
from utils import dialogs
class appWindow(QMainWindow):
"""
Application entry point, subclasses QMainWindow and implements the main widget,
sets necessary window behaviour etc.
"""
def __init__(self, parent=None):
super(appWindow, self).__init__(parent)
self.mainWidget = QWidget(self) #create new widget
#create the menu bar
titleMenu = self.menuBar() #fetch reference to current menu bar
self.mainWidget.setObjectName("Main Widget")
self.menuFile = titleMenu.addMenu('File') #File Menu
self.menuFile.addAction("New", self.newProject)
self.menuFile.addAction("Open", self.openProject)
self.menuFile.addAction("Save", self.saveProject)
self.menuGenerate = titleMenu.addMenu('Generate') #Generate menu
self.menuGenerate.addAction("Image", self.saveImage)
self.menuGenerate.addAction("Report", self.generateReport)
# create new layout for the main widget
mainLayout = QHBoxLayout()
mainLayout.setObjectName("Main Layout")
self.mdi = QMdiArea(self) #create area for files to be displayed
self.mdi.setObjectName('mdi area')
#create toolbar and add the toolbar plus mdi to layout
self.createToolbar()
mainLayout.addWidget(self.toolbar)
mainLayout.addWidget(self.mdi)
self.mdi.setOption(QMdiArea.DontMaximizeSubWindowOnActivation, True) #set flag so that window doesnt look weird
self.mdi.setTabsClosable(True)
self.mdi.setTabsMovable(True)
self.mdi.setDocumentMode(True)
#declare main window layout
self.mainWidget.setLayout(mainLayout)
self.setCentralWidget(self.mainWidget)
self.resize(1280, 720) #set collapse dim
self.setWindowState(Qt.WindowMaximized) #launch maximized
def createToolbar(self):
#place holder for toolbar with fixed width, layout may change
self.toolbar = QWidget(self.mainWidget)
self.toolbar.setObjectName("Toolbar")
self.toolbar.setFixedWidth(200)
toolbarLayout = QFormLayout(self.toolbar)
self.toolbar.setLayout(toolbarLayout)
def newProject(self):
#call to create a new file inside mdi area
project = fileWindow(self.mdi)
project.setObjectName("New Project")
self.mdi.addSubWindow(project)
if not project.tabList: # important when unpickling a file instead
project.newDiagram() #create a new tab in the new file
project.resizeHandler()
project.fileCloseEvent.connect(self.fileClosed) #closed file signal to switch to sub window view
if self.count > 1: #switch to tab view if needed
self.mdi.setViewMode(QMdiArea.TabbedView)
project.show()
def openProject(self):
#show the open file dialog to open a saved file, then unpickle it.
name = QFileDialog.getOpenFileNames(self, 'Open File(s)', '', 'Process Flow Diagram (*pfd)')
if name:
for files in name[0]:
with open(files,'rb') as file:
project = pickle.load(file)
self.mdi.addSubWindow(project)
project.show()
project.resizeHandler()
project.fileCloseEvent.connect(self.fileClosed)
if self.count > 1:
# self.tabSpace.setVisible(True)
self.mdi.setViewMode(QMdiArea.TabbedView)
def saveProject(self):
#pickle all files in mdi area
for j, i in enumerate(self.mdi.activeFiles): #get list of all windows with atleast one tab
if i.tabCount:
name = QFileDialog.getSaveFileName(self, 'Save File', f'New Diagram {j}', 'Process Flow Diagram (*.pfd)')
i.saveProject(name)
else:
return False
return True
def saveImage(self):
#place holder for future implementaion
pass
def generateReport(self):
#place holder for future implementaion
pass
def resizeEvent(self, event):
#overload resize to also handle resize on file windows inside
if self.mdi.activeSubWindow():
self.mdi.activeSubWindow().resizeHandler()
super(appWindow, self).resizeEvent(event)
def closeEvent(self, event):
#save alert on window close
if len(self.activeFiles) and not dialogs.saveEvent(self):
event.ignore()
else:
event.accept()
def fileClosed(self, index):
if self.count <= 2 :
self.mdi.setViewMode(QMdiArea.SubWindowView)
@property
def activeFiles(self):
return [i for i in self.mdi.subWindowList() if i.tabCount]
@property
def count(self):
return len(self.mdi.subWindowList())
if __name__ == '__main__':
app = ApplicationContext() # 1. Instantiate ApplicationContext
test = appWindow()
test.show()
exit_code = app.app.exec_() # 2. Invoke appctxt.app.exec_()
sys.exit(exit_code)
|
