class ConfigDialog(QDialog):
def __init__(self, parent=None):
super(ConfigDialog, self).__init__(parent)
self.msg_box = MessageBox()
self.ejabberd = EjabberdPage()
self.lider_page = LiderPage()
self.contentsWidget = QListWidget()
self.contentsWidget.setViewMode(QListView.IconMode)
self.contentsWidget.setIconSize(QSize(64, 64))
self.contentsWidget.setMovement(QListView.Static)
self.contentsWidget.setMaximumWidth(128)
self.contentsWidget.setMinimumWidth(128)
self.contentsWidget.setMinimumHeight(700)
self.contentsWidget.setSpacing(12)
self.pagesWidget = QStackedWidget()
self.pagesWidget.setMinimumHeight(800)
self.pagesWidget.setMinimumWidth(1024)
self.pagesWidget.addWidget(SettingsPage())
# self.pagesWidget.addWidget(DatabasePage())
# self.pagesWidget.addWidget(OpenLdapPage())
# self.pagesWidget.addWidget(EjabberdPage())
# self.pagesWidget.addWidget(LiderPage())
# self.pagesWidget.addWidget(LiderConsolePage())
# self.pagesWidget.addWidget(AhenkPage())
# self.pagesWidget.addWidget(WatchLog())
closeButton = QPushButton("Kapat")
aboutButton = QPushButton("Hakkında")
# self.createIcons()
self.contentsWidget.setCurrentRow(0)
closeButton.clicked.connect(self.close_page)
aboutButton.clicked.connect(self.about_lider_installer)
horizontalLayout = QHBoxLayout()
# horizontalLayout.addWidget(self.contentsWidget)
horizontalLayout.addWidget(self.pagesWidget, 1)
buttonsLayout = QHBoxLayout()
buttonsLayout.addStretch(0)
buttonsLayout.addWidget(aboutButton)
buttonsLayout.addStretch(1)
buttonsLayout.addWidget(closeButton)
mainLayout = QVBoxLayout()
mainLayout.addLayout(horizontalLayout)
mainLayout.addStretch(1)
mainLayout.addSpacing(12)
mainLayout.addLayout(buttonsLayout)
self.setLayout(mainLayout)
self.setWindowTitle("Lider Ahenk Kurulum Uygulaması")
self.setWindowIcon(QIcon(":/images/liderahenk-32.png"))
def close_page(self):
for proc in psutil.process_iter():
# check whether the process name matches
if proc.name() == "xterm":
proc.kill()
self.close()
def about_lider_installer(self):
self.msg_box.about(
"Lider Ahenk Merkezi Yönetim Sistemi Kurulum Uygulaması\nDaha fazla bilgi için...\nwww.liderahenk.org\nVersiyon: 2.0"
)
def changePage(self, current, previous):
if not current:
current = previous
self.pagesWidget.setCurrentIndex(self.contentsWidget.row(current))
def createIcons(self):
vagrantButton = QListWidgetItem(self.contentsWidget)
vagrantButton.setIcon(QIcon(':/images/settings.png'))
vagrantButton.setText("Sunucu\nAyarları")
vagrantButton.setTextAlignment(Qt.AlignHCenter)
vagrantButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
# dbButton = QListWidgetItem(self.contentsWidget)
# dbButton.setIcon(QIcon(':/images/database.png'))
# dbButton.setText("Veritabanı")
# dbButton.setTextAlignment(Qt.AlignHCenter)
# dbButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# ldapButton = QListWidgetItem(self.contentsWidget)
# ldapButton.setIcon(QIcon(':/images/ldap.png'))
# ldapButton.setText("OpenLDAP")
# ldapButton.setTextAlignment(Qt.AlignHCenter)
# ldapButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# xmppButton = QListWidgetItem(self.contentsWidget)
# xmppButton.setIcon(QIcon(':/images/ejabberd.png'))
# xmppButton.setText("Ejabberd")
# xmppButton.setTextAlignment(Qt.AlignHCenter)
# xmppButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
# liderButton = QListWidgetItem(self.contentsWidget)
# liderButton.setIcon(QIcon(':/images/liderahenk.png'))
# liderButton.setText("Lider Kur")
# liderButton.setTextAlignment(Qt.AlignHCenter)
# liderButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
# consoleButton = QListWidgetItem(self.contentsWidget)
# consoleButton.setIcon(QIcon(':/images/lider_console.png'))
# consoleButton.setText("Lider\nArayüz")
# consoleButton.setTextAlignment(Qt.AlignHCenter)
# consoleButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
# ahenkButton = QListWidgetItem(self.contentsWidget)
# ahenkButton.setIcon(QIcon(':/images/ahenk.png'))
# ahenkButton.setText("Ahenk Kur")
# ahenkButton.setTextAlignment(Qt.AlignHCenter)
# ahenkButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# logButton = QListWidgetItem(self.contentsWidget)
# logButton.setIcon(QIcon(':/images/log.png'))
# logButton.setText("Log")
# logButton.setTextAlignment(Qt.AlignHCenter)
# logButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
self.contentsWidget.currentItemChanged.connect(self.changePage)
class ConfigDialog(QDialog):
def __init__(self, parent=None):
super(ConfigDialog, self).__init__(parent)
self.msg_box = MessageBox()
self.ejabberd = EjabberdPage()
self.contentsWidget = QListWidget()
self.contentsWidget.setViewMode(QListView.IconMode)
self.contentsWidget.setIconSize(QSize(64, 64))
self.contentsWidget.setMovement(QListView.Static)
self.contentsWidget.setMaximumWidth(128)
self.contentsWidget.setMinimumWidth(128)
self.contentsWidget.setMinimumHeight(700)
self.contentsWidget.setSpacing(12)
self.pagesWidget = QStackedWidget()
self.pagesWidget.setMinimumHeight(700)
self.pagesWidget.setMinimumWidth(720)
self.pagesWidget.addWidget(SettingsPage())
# self.pagesWidget.addWidget(DatabasePage())
# self.pagesWidget.addWidget(OpenLdapPage())
# self.pagesWidget.addWidget(EjabberdPage())
self.pagesWidget.addWidget(LiderPage())
self.pagesWidget.addWidget(LiderConsolePage())
self.pagesWidget.addWidget(AhenkPage())
self.pagesWidget.addWidget(WatchLog())
closeButton = QPushButton("Kapat")
self.createIcons()
self.contentsWidget.setCurrentRow(0)
closeButton.clicked.connect(self.close)
horizontalLayout = QHBoxLayout()
horizontalLayout.addWidget(self.contentsWidget)
horizontalLayout.addWidget(self.pagesWidget, 1)
buttonsLayout = QHBoxLayout()
buttonsLayout.addStretch(1)
buttonsLayout.addWidget(closeButton)
mainLayout = QVBoxLayout()
mainLayout.addLayout(horizontalLayout)
mainLayout.addStretch(1)
mainLayout.addSpacing(12)
mainLayout.addLayout(buttonsLayout)
self.setLayout(mainLayout)
self.setWindowTitle("Lider Ahenk Kurulum Uygulaması")
self.setWindowIcon(QIcon(":/images/liderahenk-32.png"))
def changePage(self, current, previous):
if not current:
current = previous
self.pagesWidget.setCurrentIndex(self.contentsWidget.row(current))
def createIcons(self):
# asdButton = QListWidgetItem(self.contentsWidget)
vagrantButton = QListWidgetItem(self.contentsWidget)
vagrantButton.setIcon(QIcon(':/images/settings.png'))
vagrantButton.setText("Ayarlar")
vagrantButton.setTextAlignment(Qt.AlignHCenter)
vagrantButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# dbButton = QListWidgetItem(self.contentsWidget)
# dbButton.setIcon(QIcon(':/images/database.png'))
# dbButton.setText("Veritabanı")
# dbButton.setTextAlignment(Qt.AlignHCenter)
# dbButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# ldapButton = QListWidgetItem(self.contentsWidget)
# ldapButton.setIcon(QIcon(':/images/ldap.png'))
# ldapButton.setText("OpenLDAP")
# ldapButton.setTextAlignment(Qt.AlignHCenter)
# ldapButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
#
# xmppButton = QListWidgetItem(self.contentsWidget)
# xmppButton.setIcon(QIcon(':/images/ejabberd.png'))
# xmppButton.setText("Ejabberd")
# xmppButton.setTextAlignment(Qt.AlignHCenter)
# xmppButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
liderButton = QListWidgetItem(self.contentsWidget)
liderButton.setIcon(QIcon(':/images/liderahenk.png'))
liderButton.setText("Lider")
liderButton.setTextAlignment(Qt.AlignHCenter)
liderButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
consoleButton = QListWidgetItem(self.contentsWidget)
consoleButton.setIcon(QIcon(':/images/lider_console.png'))
consoleButton.setText("Lider\nArayüz")
consoleButton.setTextAlignment(Qt.AlignHCenter)
consoleButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
ahenkButton = QListWidgetItem(self.contentsWidget)
ahenkButton.setIcon(QIcon(':/images/ahenk.png'))
ahenkButton.setText("Ahenk")
ahenkButton.setTextAlignment(Qt.AlignHCenter)
ahenkButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
logButton = QListWidgetItem(self.contentsWidget)
logButton.setIcon(QIcon(':/images/log.png'))
logButton.setText("Log")
logButton.setTextAlignment(Qt.AlignHCenter)
logButton.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
self.contentsWidget.currentItemChanged.connect(self.changePage)
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(800, 600)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 21))
self.menubar.setObjectName("menubar")
self.menu_File = QtWidgets.QMenu(self.menubar)
self.menu_File.setObjectName("menu_File")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.action_Exit = QtWidgets.QAction(MainWindow)
self.action_Exit.setObjectName("action_Exit")
self.action_Exit.setStatusTip('Exit')
self.actionOpenFile = QtWidgets.QAction(MainWindow)
self.actionOpenFile.setObjectName("actionOpenFile")
self.actionOpenFile.setStatusTip('Open new File')
self.actionOpenDir = QtWidgets.QAction(MainWindow)
self.actionOpenDir.setObjectName("actionOpenDir")
self.actionOpenDir.setStatusTip('Open Directory')
self.menu_File.addAction(self.actionOpenFile)
self.menu_File.addAction(self.actionOpenDir)
self.menu_File.addSeparator()
self.menu_File.addAction(self.action_Exit)
self.menubar.addAction(self.menu_File.menuAction())
self.contentsWidget = QListWidget()
self.contentsWidget.setViewMode(QListView.ViewMode.IconMode)
self.contentsWidget.setIconSize(QSize(140,140)) # add icon 2/6
self.contentsWidget.setMovement(QListView.Movement.Static) # add icon
# self.contentsWidget.insertItem(0, 'Linked/Unlinked Bags') # add icon
# self.contentsWidget.insertItem(1, 'Simulation Tools') # add icon
self.contentsWidget.setMinimumWidth(160)
self.contentsWidget.setMaximumWidth(160)
self.contentsWidget.setSpacing(6) # add icon
# self.contentsWidget.setMaximumWidth(480)
self.createIcons() # add icon
self.stack1 = QWidget()
self.stack2 = QWidget()
self.stack1UI()
self.stack2UI()
self.st = QStackedWidget(self.centralwidget)
self.st.addWidget(self.stack1)
self.st.addWidget(self.stack2)
self.st.setMinimumWidth(640)
self.st.setMinimumHeight(480)
# print(self.size().width()) # tried to get mainwindow size
hbox = QHBoxLayout(self.centralwidget)
hbox.addWidget(self.contentsWidget)
hbox.addWidget(self.st)
# hbox.setStretch(1, 100) # need to check if this line matters 2/12 10:35
self.centralwidget.setLayout(hbox)
# self.statusbar = QMainWindow.statusBar(MainWindow)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
self.contentsWidget.currentRowChanged.connect(self.display)
self.actionOpenFile.triggered.connect(self.openFile)
self.actionOpenDir.triggered.connect(self.openDir)
self.action_Exit.triggered.connect(self.exit)
# self.createBar()
def createIcons(self): # add icon
histIcon = QListWidgetItem(self.contentsWidget)
histIcon.setIcon(QIcon("icon/hist-icon.jpg"))
histIcon.setText("Histogram")
histIcon.setTextAlignment(Qt.AlignHCenter)
histIcon.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
simuIcon = QListWidgetItem(self.contentsWidget)
simuIcon.setIcon(QIcon("icon/simul-icon.png"))
simuIcon.setText("Simulation")
simuIcon.setTextAlignment(Qt.AlignHCenter)
simuIcon.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
def createBar(self):
print("in create bar")
min_num, max_num = 0, 100
max_count = 0
linked_bag_list = []
try:
df = self.linked['Beam Diff'].dropna()
linked_bag_list = df.values.tolist()
min_num = int(min(linked_bag_list))
if min_num > 0: # check if greater than 0, set to 0
min_num = 0
max_num = int(max(linked_bag_list))
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = linked_bag_list
count = [0] * (max_num + 1) # choose the largest num as length of count
for num in linked_bag_list:
count[int(num)] += 1 # update every number's count
max_count = max(count)
setBar = QBarSet('Beam Difference Occurrence')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Linked Bags Histogram')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num+20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def createUnlinkedBar(self):
max_count = 0
unlinked_bag_list = []
try:
df = self.unlinked['Beam Diff'].dropna()
unlinked_bag_list = df.values.tolist()
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = [0] * 4
for num in unlinked_bag_list:
if -1000 <= num and num <= -51:
count[0] += 1
elif -50 <= num and num <= -1:
count[1] += 1
elif 151 <= num and num <= 200:
count[2] += 1
elif 201 <= num:
count[3] += 1
# print(count)
max_count = max(count)
setBar = QBarSet('Beam Difference Occurrence')
setBar.append(count)
series = QBarSeries()
series.append(setBar)
brush = QBrush(QColor(0xfdb157))
pen = QPen(QColor(0xfdb157))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Unlinked Bags Summary')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
labels = ['Not useful(-50 to -1000)', 'Pushed by operator(-1 to -50)', 'Slipping on belt(151 to 200)', 'Not useful 201+']
axisX = QBarCategoryAxis()
axisX.append(labels)
# chart.setAxisX(axisX, series)
chart.addAxis(axisX, Qt.AlignBottom)
# chart.ChartAreas[0].AxisX.LabelAutoFitStyle = LabelAutoFitStyle.WrodWrap
series.attachAxis(axisX)
axisY = QValueAxis()
axisY.setRange(0, max_count+1)
# chart.setAxisY(axisY, series)
chart.addAxis(axisY, Qt.AlignLeft)
series.attachAxis(axisY)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def createSummaryBar(self):
# self.dp.dfinal
series = QPieSeries()
labelfont = QFont()
labelfont.setPixelSize(11)
try:
series.append("Linked", len(self.linked.index))
series.append("Unlinked", len(self.unlinked.index))
# slices = QPieSlice()
slices1 = series.slices()[1]
slices1.setExploded()
slices1.setLabelVisible()
slices1.setPen(QPen(QColor(0x57B1FD), 2))
slices1.setBrush(QBrush(QColor(0xfdb157)))
slices1.setLabelPosition(QPieSlice.LabelOutside)
slices1.setLabel(("{0} {1:.2f}%").format("Unlinked", 100*slices1.percentage()))
slices1.setLabelFont(labelfont)
# slices.setPen(QPen(Qt.darkGreen, 2))
# slices.setBrush(QBrush(QColor(0xfdb157)))
except AttributeError:
self.statusbar.showMessage('Data not ready')
series.append("Linked Bags VS Unlinked Bags", 1)
slices = series.slices()[0]
slices.setBrush(QBrush(QColor(0x57B1FD)))
# slices.setLabel(("%1%").format(100*slices.percentage(), 0, 'f', 1))
# slices.setLabelPosition(QPieSlice.LabelInsideHorizontal)
if len(series.slices()) == 2: # display "linked" label only at data was processed
slices.setLabelVisible()
slices.setLabel(("{0} {1:.2f}%").format("Linked", 100*slices.percentage()))
slices.setLabelFont(labelfont)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
chart.setTitle("Total")
chart.legend().hide()
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
return chartView
def getCertainRangeList(self, total_time_lst):
self.ready_to_plot_lst = []
self.range0To2 = []
self.range2To3 = []
self.range3To6 = []
self.range6To9 = []
self.range9To20 = []
for item in total_time_lst:
if 0 < item and item <= 2:
# print("in 0-3: ", item)
self.ready_to_plot_lst.append(int(item))
self.range0To2.append(item)
elif 2 < item and item <= 3:
self.ready_to_plot_lst.append(int(item))
self.range2To3.append(item)
elif 3 < item and item <= 6:
# print("in 3-6: ", item)
self.ready_to_plot_lst.append(int(item))
self.range3To6.append(item)
elif 6 < item and item <= 9:
# print("in 6-9: ", item)
self.range6To9.append(item)
self.ready_to_plot_lst.append(int(item))
elif 9 < item and item <= 20:
self.range9To20.append(item)
def onClicked(self, b): # try to add histogram radio button 2/11 12:02
# self.removeWidgets()
# self.layout.addWidget(self.createBar())
if b.text() == "Histogram":
print("histogram checkd!")
elif b.text() == "RangeHist":
print("range histogram")
elif b.text() == "Pie Chart":
print("pie chart")
def stack1UI(self):
self.layout = QVBoxLayout()
combo = QComboBox(self.stack1)
combo.addItem("Linked Bags")
combo.addItem("Unlinked Bags")
combo.addItem("Summary")
self.image = QtWidgets.QLabel(self.stack1) # try to do stackwidget here 2/10 10:52
self.image.setObjectName("imageLabel") # try to do stackwidget here 2/10 10:52
self.layout.addWidget(combo)
self.layout.addWidget(self.image) # add stackwidget 2/10 10:52
self.stack1.setLayout(self.layout)
combo.activated[str].connect(self.onActivated)
def stack2UI(self):
self.vlayout = QVBoxLayout()
hlayout = QHBoxLayout()
combo1 = QComboBox(self.stack2)
combo1.addItem("Distribution")
combo1.addItem("Occurrence")
combo1.addItem("Summary")
label = QLabel()
label.setText("Delay: ")
self.text = QLineEdit()
button = QPushButton("Go")
hlayout.addWidget(combo1)
hlayout.addWidget(label)
hlayout.addWidget(self.text)
hlayout.addWidget(button)
self.slayout = QHBoxLayout()
self.labelDelay = QtWidgets.QLabel(self.stack2)
self.labelafterDelay = QtWidgets.QLabel(self.stack2) # try to add a paralel widget (before / after)
self.labelDelay.setObjectName("result")
self.labelafterDelay.setObjectName("delayResult") # try to add a paralel widget (before / after)
# self.labelDelay.setText("before")
# self.labelafterDelay.setText("after") # try to add a paralel widget (before / after)
self.slayout.addWidget(self.labelDelay)
self.slayout.addWidget(self.labelafterDelay)
self.vlayout.addLayout(hlayout)
self.vlayout.addLayout(self.slayout)
self.stack2.setLayout(self.vlayout)
button.clicked.connect(self.simulate)
combo1.activated[str].connect(self.onActivated1)
def afterDelayPieChart(self):
series = QPieSeries()
labelfont = QFont()
labelfont.setPixelSize(11)
total_running_after_delay, total_stopped_after_delay = [], []
try:
total_running_after_delay = self.tm.total_running_after_delay
total_stopped_after_delay = self.tm.total_stopped_after_delay
# slices.setPen(QPen(Qt.darkGreen, 2))
# slices.setBrush(QBrush(QColor(0xfdb157)))
except AttributeError:
self.statusbar.showMessage('Data not ready')
series.append("Run", sum(total_running_after_delay))
series.append("Stop", sum(total_stopped_after_delay))
# slices = QPieSlice()
slices = series.slices()[0] # Run time slice
slices.setBrush(QBrush(QColor(0x57B1FD))) # Blue
slices.setLabelVisible() # Set label visible
slices.setLabel(("{0} {1:.2f}%").format("Run time", 100*slices.percentage())) # Set percentage
slices.setLabelFont(labelfont) # Set label font
slices1 = series.slices()[1] # Stop time slice
slices1.setExploded() # Set stop slice exploded
slices1.setLabelVisible() # Set label visible
slices1.setPen(QPen(QColor(0x57B1FD), 2)) # Blue
slices1.setBrush(QBrush(QColor(0xA6E22E))) # Orange
# slices1.setLabelPosition(QPieSlice.LabelOutside) # Set label outside
slices1.setLabel(("{0} {1:.2f}%").format("Stop time", 100*slices1.percentage())) # Set percentage
slices1.setLabelFont(labelfont) # Set label font
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
chart.setTitle("Total Time (after)")
chart.legend().hide()
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
return chartView
def afterDelay(self):
# print("in after delay bar")
min_num, max_num = 0, 100
max_count = 0
total_stopped_after_delay = []
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
try:
total_stopped_after_delay = self.tm.total_stopped_after_delay
max_num = max(total_stopped_after_delay)
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
for num in total_stopped_after_delay:
count[int(num)] += 1 # update every number's count
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('Stop Time Occurrence')
setBar.append(count)
brush = QBrush(QColor(0xA6E22E)) # Green
pen = QPen(QColor(0xA6E22E)) # Green
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Occurrence (after)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num+20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def afterDelayDistribution(self):
min_num, max_num = 0, 100
max_count = 0
total_stopped_after_delay = []
count = [0]
try:
total_stopped_after_delay = self.tm.total_stopped_after_delay
max_num = len(total_stopped_after_delay) # change max() to len(), now it's correct
count = total_stopped_after_delay
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('stop time')
setBar.append(count)
brush = QBrush(QColor(0xA6E22E)) # Green
pen = QPen(QColor(0xA6E22E)) # Green
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Distribution (after)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def beforeDelayPieChart(self):
series = QPieSeries()
labelfont = QFont()
labelfont.setPixelSize(11)
total_running_time, total_stopped_time = [], []
try:
total_running_time = self.tm.total_running_time
total_stopped_time = self.tm.total_stopped_time
# slices.setPen(QPen(Qt.darkGreen, 2))
# slices.setBrush(QBrush(QColor(0xfdb157)))
except AttributeError:
self.statusbar.showMessage('Data not ready')
series.append("Run", sum(total_running_time))
series.append("Stop", sum(total_stopped_time))
slices = series.slices()[0]
slices.setBrush(QBrush(QColor(0x57B1FD)))
slices.setLabelVisible()
slices.setLabel(("{0} {1:.2f}%").format("Run Time", 100*slices.percentage()))
slices.setLabelFont(labelfont)
slices1 = series.slices()[1]
slices1.setExploded()
slices1.setLabelVisible()
slices1.setPen(QPen(QColor(0x57B1FD), 2))
slices1.setBrush(QBrush(QColor(0xfdb157)))
slices1.setLabel(("{0} {1:.2f}%").format("Stop Time", 100*slices1.percentage()))
slices1.setLabelFont(labelfont)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
chart.setTitle("Total Time (before)")
chart.legend().hide()
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
return chartView
def beforeDelay(self):
# print("in before delay bar")
min_num, max_num = 0, 100
max_count = 0
total_stopped_time = []
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
try:
total_stopped_time = self.tm.total_stopped_time
max_num = max(total_stopped_time)
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
for num in total_stopped_time:
count[int(num)] += 1 # update every number's count
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('Stop Time Occurrence')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Occurrence (before)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num+20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def beforeDelayDistribution(self):
min_num, max_num = 0, 100
max_count = 0
total_stopped_time = []
count = [0]
try:
total_stopped_time = self.tm.total_stopped_time
max_num = len(total_stopped_time) # change from max() to len() 2/12 11:11
count = total_stopped_time
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('stop time')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Distribution (before)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def display(self, i):
self.st.setCurrentIndex(i)
# mainMenu components
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
self.menu_File.setTitle(_translate("MainWindow", "&File"))
self.action_Exit.setText(_translate("MainWindow", "&Exit"))
self.actionOpenFile.setText(_translate("MainWindow", "&Open File/Files"))
self.actionOpenDir.setText(_translate("MainWindow", "&Open Folder"))
def removeWidgets(self):
for i in reversed(range(0,self.layout.count())): # change 2 to 3 if add another widget(radio)
self.layout.itemAt(i).widget().setParent(None)
def removeWidgets1(self):
for i in reversed(range(1,self.slayout.count())):
self.slayout.itemAt(i).widget().setParent(None)
def onActivated1(self, text):
if text == 'Distribution':
self.removeWidgets1()
self.slayout.addWidget(self.beforeDelayDistribution(), QtCore.Qt.AlignCenter)
self.slayout.addWidget(self.afterDelayDistribution(), QtCore.Qt.AlignCenter)
elif text == 'Occurrence':
self.removeWidgets1()
self.slayout.addWidget(self.beforeDelay(), QtCore.Qt.AlignCenter)
self.slayout.addWidget(self.afterDelay(), QtCore.Qt.AlignCenter)
elif text == 'Summary':
self.removeWidgets1()
self.slayout.addWidget(self.beforeDelayPieChart(), QtCore.Qt.AlignCenter)
self.slayout.addWidget(self.afterDelayPieChart(), QtCore.Qt.AlignCenter)
# combo box items
def onActivated(self, text):
# print(text)
if text == 'Linked Bags':
self.removeWidgets()
# self.layout.addWidget(self.createBar())
elif text == 'Unlinked Bags':
self.removeWidgets()
self.layout.addWidget(self.createUnlinkedBar())
elif text == 'Summary':
self.removeWidgets()
self.layout.addWidget(self.createSummaryBar())
# self.s1layout = QHBoxLayout()
# self.sumlabel = QtWidgets.QLabel()
# self.summaryLabel()
# # label.setText("tettddddddddddddddddddtt")
# self.s1layout.addWidget(self.createSummaryBar())
# self.s1layout.addWidget(self.sumlabel)
# self.layout.addLayout(self.s1layout)
def summaryLabel(self):
try:
total = len(self.dfinal.index) # total number of scanned bags
linked = len(self.linked.index) # total number of linked bags
unlinked = total - linked # total number of unlinked bags
cut_oversized = len(self.cut_oversized.index) # cut or oversized bags.
co_percentage = (cut_oversized/total) * 100 # cut or oversized percentage
percentage = (unlinked/(total)) * 100 # percentage for unlinked bags
self.sumlabel.setText('Total bins: {0}\nTotal linked bags: {1}\nTotal unlinked bags: {2}\nPercentage of unlinked bags: {3:.2f} % \nPercentage of cut/oversized bags: {4:.2f} %'.format(total, \
linked, unlinked, percentage, co_percentage))
except AttributeError:
self.statusbar.showMessage("Data not processed")
def simulate(self):
# print(self.text.text())
if bool(re.match('^[0-9]+$',self.text.text())):
try:
self.tm = TimeMeasure('runtime_.txt', int(self.text.text())) # change variables to global value 2/11 2:06
self.tm.getDuration()
self.tm.runList("")
self.tm.stopList("")
self.tm.runListDelay("")
self.tm.stopListDelay("")
self.total_time = self.tm.org_finish_time - self.tm.org_start_time
except FileNotFoundError:
self.statusbar.showMessage("Data not ready")
# self.labelDelay.setText("Simulation Result:\n")
# self.labelDelay.setText(self.labelDelay.text() + "\n---------------Before delay---------------\n\n")
# self.labelDelay.setText(self.labelDelay.text() + "Run Time total: {0:d} Sum: {1:f} seconds Percentage: {2:.2f}%\n".format(len(self.tm.total_running_time), sum(self.tm.total_running_time), sum(self.tm.total_running_time)/self.total_time * 100))
# self.labelDelay.setText(self.labelDelay.text() + "Stop Time total: {0:d} Sum: {1:f} seconds Percentage: {2:.2f}%\n".format(len(self.tm.total_stopped_time), sum(self.tm.total_stopped_time), sum(self.tm.total_stopped_time)/self.total_time * 100))
# self.labelDelay.setText(self.labelDelay.text() + "Total: {0:f} seconds Percentage: {1:.2f}%\n".format(self.total_time, sum(self.tm.total_running_time)/self.total_time * 100 + sum(self.tm.total_stopped_time)/self.total_time * 100))
# # print("run:",len(tm.total_running_time), "sum:", sum(tm.total_running_time),"percentage:",sum(tm.total_running_time)/total_time * 100)
# # print("stop:",len(tm.total_stopped_time), "sum:", sum(tm.total_stopped_time),"percentage:",sum(tm.total_stopped_time)/total_time * 100)
# # print("total:",total_time, sum(tm.total_running_time)/total_time * 100 + sum(tm.total_stopped_time)/total_time * 100)
# # print("\n---------------Add delay 2 seconds---------------\n")
# self.labelDelay.setText(self.labelDelay.text() + "\n---------------Add delay 2 seconds---------------\n\n")
# self.total_time_after_delay = self.tm.delay_finish_time - self.tm.delay_start_time
# self.labelDelay.setText(self.labelDelay.text() + "Run time after delay: {0:d} Sum: {1:f} seconds Percentage: {2:.2f}%\n".format(len(self.tm.total_running_after_delay), sum(self.tm.total_running_after_delay), sum(self.tm.total_running_after_delay) / self.total_time_after_delay * 100))
# self.labelDelay.setText(self.labelDelay.text() + "Stop time after delay: {0:d} Sum: {1:f} seconds Percentage: {2:.2f}%\n".format(len(self.tm.total_stopped_after_delay), sum(self.tm.total_stopped_after_delay), sum(self.tm.total_stopped_after_delay) / self.total_time_after_delay * 100))
# self.labelDelay.setText(self.labelDelay.text() + "Total: {0:f} seconds Percentage: {1:.2f}%\n".format(self.total_time_after_delay, sum(self.tm.total_running_after_delay) / self.total_time_after_delay * 100 + sum(self.tm.total_stopped_after_delay) / self.total_time_after_delay * 100))
# # print("run after delay:", len(tm.total_running_after_delay), "sum:", sum(tm.total_running_after_delay), "percentage:",sum(tm.total_running_after_delay) / total_time_after_delay * 100)
# # print("stop after delay:",len(tm.total_stopped_after_delay), "sum:", sum(tm.total_stopped_after_delay), "percentage:",sum(tm.total_stopped_after_delay) / total_time_after_delay * 100)
# # print("total:",total_time_after_delay, sum(tm.total_running_after_delay) / total_time_after_delay * 100 + sum(tm.total_stopped_after_delay) / total_time_after_delay * 100)
# self.tm.fakestopList()
# self.delaylist = self.tm.getCertainRangeList(self.tm.fake_total_stopped_time)
# # print("\nSummary:")
# # print("Before delay XRAY_MIN / total: 100% ----> After delay XRAY_MIN / total: {0:.2f}%".format((len(tm.fake_total_stopped_time)-len(delaylist))/len(tm.fake_total_stopped_time)*100))
# # print("Before delay total time: 100% ----> After delay total time: {0:.2f}%".format(total_time_after_delay/total_time*100))
# self.labelDelay.setText(self.labelDelay.text() + "\nSummary:")
# self.labelDelay.setText(self.labelDelay.text() + "Before delay total time: 100% ----> After delay total time: {0:.2f}%\n".format(self.total_time_after_delay/self.total_time*100))
# self.labelDelay.setText(self.labelDelay.text() + "Before delay XRAY_MIN / total: 100% ----> After delay XRAY_MIN / total: {0:.2f}%".format((len(self.tm.fake_total_stopped_time)-len(self.delaylist))/len(self.tm.fake_total_stopped_time)*100))
# self.labelDelay.setFont(QtGui.QFont("Times", 12, QtGui.QFont.Bold))
# self.labelDelay.setAlignment(Qt.AlignTop)#HCenter)
self.removeWidgets1()
self.slayout.addWidget(self.beforeDelayDistribution(), QtCore.Qt.AlignCenter)
self.slayout.addWidget(self.afterDelayDistribution(), QtCore.Qt.AlignCenter)
# display summary label
def setSummary(self):
self.image.setPixmap(QtGui.QPixmap('his.png'))
# display linked bags histogram
def setLinkedHistogram(self):
# pixmap = pixmap.tmaled(self.label.width(), self.label.height(), QtCore.Qt.KeepAspectRatio, QtCore.Qt.FastTransformation) # make image unclear
self.image.setPixmap(QtGui.QPixmap('link.png'))
# display unlinked bags histogram
def setUnlinkedHistogram(self):
self.image.setPixmap(QtGui.QPixmap('unlink.png'))
# display time stopped histogram
def setTimeStoppedHistogram(self):
self.image.setPixmap(QtGui.QPixmap('stop_time_distribution.png'))
# display time running histogram
def setTimeRunningHistogram(self):
self.image.setPixmap(QtGui.QPixmap('running_time_distribution'))
# open file dialog
def openFile(self):
filenames, _ = QtWidgets.QFileDialog.getOpenFileNames(None,"QFileDialog.getOpenFileNames()", "","All Files (*);;Log Files (*.log);;")
if filenames:
print("inside:",filenames)
print("fdsfs", filenames)
illegal = False
suf = 'log' # test for single .txt first, need to modify for final version
if len(filenames) != 0:
for f in filenames: # check all files are illegal in here
suffix = f[f.rfind('.')+1:]
if suffix != suf:
print("Illegal selection")
illegal = True
print(illegal)
if illegal:
self.showdialog(illegal, True);
else:
# self.statusbar.showMessage('Processing Files...')
start_time = timeit.default_timer()
pp = PreProc(filenames) # pass files into PreProc class
elapsed = timeit.default_timer() - start_time
print("time cost preProc:", elapsed)
# self.statusbar.showMessage('Analyzing datas...')
start_time1 = timeit.default_timer()
dp = DataProc() # pass the prepared data into DataProc class
elapsed1 = timeit.default_timer() - start_time1
print("time cost dataProc:", elapsed1)
self.linked = dp.linked
self.unlinked = dp.unlinked
self.dfinal = dp.dfinal
self.cut_oversized = dp.cut_oversized
self.statusbar.showMessage('Done')
self.removeWidgets()
self.layout.addWidget(self.createBar())
# open directory dialog
def openDir(self):
dialog = QtWidgets.QFileDialog()
dialog.setFileMode(QFileDialog.DirectoryOnly)
dic_file = []
if dialog.exec_(): # == QtGui.QDialog.Accepted:
dic_file = dialog.selectedFiles()
print("openDir: (dic_file) ", dic_file)
log_file = []
has_log = False
suf = 'log'
if dic_file:
for f in os.listdir(dic_file[0]):
suffix = f[f.rfind('.')+1:]
name = f[:f.rfind('.')]
print(name)
if suffix == suf:
has_log = True
if "AnalogicStandaloneType" in name:# if match AnalogicStandaloneType*.log, append to log_file
log_file += dic_file[0]+"/"+f,
print(has_log, log_file)
if not has_log:
self.showdialog(False, has_log)
if len(log_file) == 0 and has_log:
self.showdialog(False, False)
if len(log_file) != 0:
start_time = timeit.default_timer()
pp = PreProc(log_file) # pass files into PreProc class
elapsed = timeit.default_timer() - start_time
print("time cost preProc:", elapsed)
start_time1 = timeit.default_timer()
dp = DataProc() # pass the prepared data into DataProc class
elapsed1 = timeit.default_timer() - start_time1
print("time cost dataProc:", elapsed1)
self.linked = dp.linked
self.unlinked = dp.unlinked
self.dfinal = dp.dfinal
self.cut_oversized = dp.cut_oversized
self.statusbar.showMessage('Done')
self.removeWidgets()
self.layout.addWidget(self.createBar())
# pop up warning window
def showdialog(self, illegal_file, has_log):
msg = QMessageBox()
msg.setIcon(QMessageBox.Critical)
if not has_log:
msg.setText("No log files found.")
if illegal_file:
print("in illegal")
msg.setText("Invalid log files detected.")
msg.setWindowTitle("Something's wrong")
# msg.setGeometry(200, 200, 300, 400) # need to change if need this line
msg.setStandardButtons(QMessageBox.Ok)
retval = msg.exec_()
def exit(self):
sys.exit(app.exec_())
class FunctionWidget(QWidget):
begin_save_signal = pyqtSignal(str, int)
saved_signal = pyqtSignal(str, int)
def __init__(self, parent=None):
super(FunctionWidget, self).__init__(parent=parent)
h_layout = QHBoxLayout(self)
splitter = QSplitter(Qt.Horizontal)
# 功能树
self.tree = QTreeWidget(self)
self.tree.setColumnCount(1)
self.tree.setFixedWidth(160)
self.tree.setHeaderLabel("功能选项")
# 交互——树节点
react = QTreeWidgetItem(self.tree)
react.setText(0, "交互")
new_order = QTreeWidgetItem(react)
new_order.setText(0, "新指令")
age_sex_identification = QTreeWidgetItem(react)
age_sex_identification.setText(0, "年龄、性别识别")
emotion_analysis = QTreeWidgetItem(react)
emotion_analysis.setText(0, "情感分析")
saved_order = QTreeWidgetItem(react)
saved_order.setText(0, "已存指令")
history_order = QTreeWidgetItem(react)
history_order.setText(0, "历史记录")
txt2v = QTreeWidgetItem(react)
txt2v.setText(0, "文字转语音")
v2txt = QTreeWidgetItem(react)
v2txt.setText(0, "语音转文字")
# 设置——树节点
setting = QTreeWidgetItem(self.tree)
setting.setText(0, "设置")
token = QTreeWidgetItem(setting)
token.setText(0, "讯飞Token")
turing_robot_token = QTreeWidgetItem(setting)
turing_robot_token.setText(0, "图灵机器人Token")
# 树操作
self.tree.expandAll()
self.tree.currentItemChanged.connect(self.on_clicked_tree)
# 功能界面栈
self.stack = QStackedWidget()
self.stack.setMinimumWidth(380)
# 功能界面
stack_new_order = NewOrder(title_text='新指令')
stack_age_sex_identification = NewOrder(title_text='年龄、性别识别')
stack_emotion_analysis = NewOrder(title_text='情感分析')
stack_saved_order = SavedOrder()
stack_saved_order.begin_save_tree_signal.connect(
self.handle_begin_save)
stack_saved_order.tree_saved_signal.connect(self.handle_saved)
stack_history_order = HistoryOrder()
stack_history_order.begin_save_tree_signal.connect(
self.handle_begin_save)
stack_history_order.tree_saved_signal.connect(self.handle_saved)
stack_new_order.begin_record_order_signal.connect(
stack_history_order.history_recorder)
stack_saved_order.begin_record_order_signal.connect(
stack_history_order.history_recorder)
stack_txt2v = TXT2V()
stack_v2txt = V2TXT()
stack_token = Token()
stack_token.begin_save_token_signal.connect(self.handle_begin_save)
stack_token.token_saved_signal.connect(self.handle_saved)
stack_turing_robot_token = TuringRobotToken()
stack_turing_robot_token.begin_save_token_signal.connect(
self.handle_begin_save)
stack_turing_robot_token.token_saved_signal.connect(self.handle_saved)
self.stack.addWidget(stack_new_order)
self.stack.addWidget(stack_age_sex_identification)
self.stack.addWidget(stack_emotion_analysis)
self.stack.addWidget(stack_saved_order)
self.stack.addWidget(stack_history_order)
self.stack.addWidget(stack_txt2v)
self.stack.addWidget(stack_v2txt)
self.stack.addWidget(stack_token)
self.stack.addWidget(stack_turing_robot_token)
splitter.addWidget(self.tree)
splitter.addWidget(self.stack)
splitter.setStretchFactor(0, 1)
splitter.setStretchFactor(1, 6)
h_layout.addWidget(splitter)
self.setLayout(h_layout)
# 切换功能界面栈的界面
def on_clicked_tree(self, item: QTreeWidgetItem):
text = item.text(0)
if text == "新指令" or text == "交互":
self.stack.setCurrentIndex(0)
elif text == "年龄、性别识别":
self.stack.setCurrentIndex(1)
elif text == "情感分析":
self.stack.setCurrentIndex(2)
elif text == "已存指令":
self.stack.setCurrentIndex(3)
elif text == "历史记录":
self.stack.setCurrentIndex(4)
elif text == "文字转语音":
self.stack.setCurrentIndex(5)
elif text == "语音转文字":
self.stack.setCurrentIndex(6)
elif text == "讯飞Token" or text == "设置":
self.stack.setCurrentIndex(7)
elif text == "图灵机器人Token":
self.stack.setCurrentIndex(8)
# 已存指令树：开始保存
def handle_begin_save(self, s: str, duration: int):
self.begin_save_signal.emit(s, duration)
# 已存指令树：已保存
def handle_saved(self, s: str, duration: int):
self.saved_signal.emit(s, duration)
class MainWindowUi(QMainWindow):
"""This class manages the system tray icon of the main program, post-login.
"""
vpn_connect_section = QVBoxLayout()
def __init__(self):
super(QMainWindow, self).__init__()
# Using a StackedWidget to be able to replace login window
# https://stackoverflow.com/questions/13550076
self.cw = QStackedWidget()
self.setCentralWidget(self.cw)
# Set minimum width of Main Window to 500 pixels
self.cw.setMinimumWidth(500)
self.setWindowTitle('MerLink - VPN Client for Meraki firewalls')
self.link_style = "font-family: verdana, sans-serif; font-style:" \
" normal; font-size: 13px; color: #1795E6;"
# Required for Login. Elements should be object variables if they
# Could be called by other modules
self.username_textfield = QLineEdit()
self.password_textfield = QLineEdit()
self.guest_user_chkbox = QCheckBox("Use guest account instead")
self.password_textfield.setEchoMode(QLineEdit.Password)
self.org_dropdown = QComboBox()
self.create_vpn_btn = QPushButton("Create VPN Interface")
self.create_vpn_tabs = QTabWidget()
self.tab_dashboard = QWidget()
self.tab_manual = QWidget()
self.vpn_opts_layout = QVBoxLayout()
def setup_pyqt_slots(self):
"""Initiate all of the triggers and connects."""
def setup_window(self):
"""Setup various sections that will be combined."""
self.create_vpn_tabs.addTab(self.tab_dashboard, "Dashboard Setup")
self.create_vpn_tabs.addTab(self.tab_manual, "Manual Setup")
self.vpn_opts_setup()
self.setup_manual_tab()
self.setup_dashboard_tab()
self.vpn_connect_setup()
self.decorate_sections()
self.combine_sections_dashboard()
# Set the layout once we are done adding elements
self.main_window_set_admin_layout()
def setup_dashboard_tab(self):
"""Provide input fields for dashboard-gathered data."""
self.tab_dashboard.layout = QVBoxLayout(self)
# Guest users should manually enter their information
# Only difference between Guest and Dashboard for UI should be that
# Email/pass is prepopulated for dashboard admins (inactivated)
email_pass_layout = QVBoxLayout()
username_label = QLabel("Email")
password_label = QLabel("Password")
self.disable_email_pass(True)
self.add_all_to_layout(
email_pass_layout,
[
username_label,
self.username_textfield,
password_label,
self.password_textfield
]
)
self.tab_dashboard.layout.addWidget(self.guest_user_chkbox)
self.tab_dashboard.layout.addLayout(email_pass_layout)
self.tab_dashboard.layout.addLayout(self.vpn_opts_layout)
self.tab_dashboard.setLayout(self.tab_dashboard.layout)
def disable_email_pass(self, change_to_disabled):
"""Disable username/password if user should not edit them."""
disable_lineedit(self.username_textfield, change_to_disabled)
disable_lineedit(self.password_textfield, change_to_disabled)
def vpn_opts_setup(self):
"""Set up the vpn vars UI region."""
# Create org and network dropdowns so the user can select the firewall
# they would like to connect to.
org_dropdown = QComboBox()
org_dropdown.addItem('-- Select an Organization --')
network_dropdown = QComboBox()
network_dropdown.setEnabled(False)
# Allow the user to change the VPN name
vpn_name_layout = QHBoxLayout()
vpn_name_label = QLabel("VPN Name:")
vpn_name_textfield = QLineEdit()
vpn_name_layout.addWidget(vpn_name_label)
vpn_name_layout.addWidget(vpn_name_textfield)
# Ask the user for int/str values if they want to enter them
idle_disconnect_layout = QHBoxLayout()
idle_disconnect_chkbox = QCheckBox("Idle disconnect seconds?")
idle_disconnect_spinbox = QSpinBox()
idle_disconnect_spinbox.setValue(0)
# Negative seconds aren't useful here
idle_disconnect_spinbox.setMinimum(0)
idle_disconnect_layout.addWidget(
idle_disconnect_chkbox)
idle_disconnect_layout.addWidget(
idle_disconnect_spinbox)
dns_suffix_layout = QHBoxLayout()
dns_suffix_chkbox = QCheckBox("DNS Suffix?")
dns_suffix_txtbox = QLineEdit('-')
dns_suffix_layout.addWidget(dns_suffix_chkbox)
dns_suffix_layout.addWidget(dns_suffix_txtbox)
# Boolean asks of the user
split_tunneled_chkbox = QCheckBox("Split-Tunnel?")
remember_credential_chkbox = QCheckBox("Remember Credentials?")
use_winlogon_chkbox = QCheckBox("Use Windows Logon Credentials?")
self.add_all_to_layout(
self.vpn_opts_layout,
[
org_dropdown,
network_dropdown,
vpn_name_layout,
# Add layouts for specialized params
idle_disconnect_layout,
dns_suffix_layout,
# Add checkboxes
split_tunneled_chkbox,
remember_credential_chkbox,
use_winlogon_chkbox,
# Ensure that button is at bottom of pane by adding space
QSpacerItem(0, 0, QSizePolicy.Minimum, QSizePolicy.Expanding)
]
)
def setup_manual_tab(self):
"""Gray out options and provide input fields to manually enter data."""
self.tab_manual.layout = QVBoxLayout(self)
# User should be able to change email/pass as it's required
self.disable_email_pass(False)
username_textfield = QLineEdit()
password_textfield = QLineEdit()
password_textfield.setEchoMode(QLineEdit.Password)
username_label = QLabel("Email")
password_label = QLabel("Password")
vpn_name_label = QLabel("VPN Name")
vpn_name_textfield = QLineEdit()
server_name_label = QLabel("Server name/IP")
server_name_textfield = QLineEdit()
shared_secret_label = QLabel("Shared Secret")
shared_secret_textfield = QLineEdit()
shared_secret_textfield.setEchoMode(QLineEdit.Password)
self.add_all_to_layout(
self.tab_manual.layout,
[
username_label,
username_textfield,
password_label,
password_textfield,
vpn_name_label,
vpn_name_textfield,
server_name_label,
server_name_textfield,
shared_secret_label,
shared_secret_textfield,
QSpacerItem(0, 0, QSizePolicy.Minimum, QSizePolicy.Expanding)
]
)
self.tab_manual.setLayout(self.tab_manual.layout)
def vpn_connect_setup(self):
"""Setup the GUI componentst of the right pane."""
vpn_list = QListWidget()
ipsum_vpn_interfaces = ['eth', 'wifi']
vpn_list.addItems(ipsum_vpn_interfaces)
check_for_probs_cb = QCheckBox(
"Check for issues before connecting (recommended)")
check_for_probs_cb.setChecked(True)
probs_list = QListWidget()
problems = ["Forget the milk", "My hovercraft is full of eels"]
probs_list.addItems(problems)
connect_vpn_btn = QPushButton("Connect")
self.add_all_to_layout(
self.vpn_connect_section,
[
vpn_list,
check_for_probs_cb,
probs_list,
connect_vpn_btn
]
)
def decorate_sections(self):
"""Add a box around each section for readability."""
# Set GroupBox CSS manually because otherwise margins are huge
self.setStyleSheet(".QGroupBox { border: 1px solid #ccc;}")
def combine_sections_dashboard(self):
"""Combine left and right panes into a final layout."""
# Create a horizontal line above the status bar to highlight it
self.hline = QFrame()
self.hline.setFrameShape(QFrame.HLine)
self.hline.setFrameShadow(QFrame.Sunken)
self.vline = QFrame()
self.vline.setFrameShape(QFrame.VLine)
self.vline.setFrameShadow(QFrame.Sunken)
# Status bar be at bottom and inform user of what the program is doing.
self.status = QStatusBar()
self.status.showMessage("Status: -")
self.status.setStyleSheet("Background:#fff")
main_widget = QWidget()
main_layout = QVBoxLayout(main_widget)
left_pane = QVBoxLayout()
left_pane.addWidget(self.create_vpn_tabs)
left_pane.addWidget(self.create_vpn_btn)
two_pane_layout = QHBoxLayout()
two_pane_layout.addLayout(left_pane)
two_pane_layout.addWidget(self.vline)
two_pane_layout.addLayout(self.vpn_connect_section)
main_layout.addLayout(two_pane_layout)
main_layout.addWidget(self.hline)
main_layout.addWidget(self.status)
self.cw.addWidget(main_widget)
def main_window_set_admin_layout(self):
"""Set the dashboard user layout.
Hides guest user layout as we will only be connecting with one user.
The user will see the username/obfuscated password they entered.
"""
self.username_textfield.setText(
self.login_dict['username'])
self.username_textfield.setReadOnly(True)
self.password_textfield.setText(
self.login_dict['password'])
self.password_textfield.setReadOnly(True)
def main_window_set_guest_layout(self):
"""Set the guest user layout.
Hides dashboard user layout as we will only be connecting with one user.
The user will see blank user/pass text fields where they can enter
information for a guest user.
"""
self.radio_username_textfield.clear()
self.radio_username_textfield.setReadOnly(False)
self.radio_password_textfield.clear()
self.radio_password_textfield.setReadOnly(False)
@staticmethod
def add_all_to_layout(layout, element_list):
"""Add all of the elements to the layout."""
for elem in element_list:
if isinstance(elem, QWidget):
layout.addWidget(elem)
elif is_layout(elem):
layout.addLayout(elem)
elif isinstance(elem, QSpacerItem):
layout.addItem(elem)
else:
print("ERROR: Trying to add illegal element to UI!")
exit(1)