def outputContextMenu(self, listview: QListView, pos: QPoint):
current_model = listview.model()
menu = QMenu()
edit_action = QAction(
"Edit clicked" if len(listview.selectedIndexes()) > 1 else "Edit",
menu)
edit_action.triggered.connect(self.editCurrentTarget)
menu.addAction(edit_action)
delete_action = QAction("Remove", menu)
delete_action.triggered.connect(lambda: current_model.__delitem__(
[i.row() for i in listview.selectedIndexes()]))
menu.addAction(delete_action)
selected_istemp = current_model[
listview.currentIndex().row()].temporary
mark_temp_action = QAction("Mark temp", menu)
mark_temp_action.setCheckable(True)
mark_temp_action.setChecked(selected_istemp)
mark_temp_action.triggered.connect(lambda: [
current_model[i.row()].switch_temporary(not selected_istemp)
for i in listview.selectedIndexes()
])
menu.addAction(mark_temp_action)
self.addTargetActions(menu)
menu.exec_(listview.mapToGlobal(pos))
class Window(QDialog):
def __init__(self, parent=None):
super(Window, self).__init__(parent)
self.createTrayIcon()
self.createProgramsList()
self.createCodeEditPage()
self.logsPage = QTextBrowser()
self.documentation = QTextBrowser()
self.tabWidget = QTabWidget()
self.tabWidget.setIconSize(QSize(64, 64))
self.tabWidget.addTab(self.programsListPage, QIcon(":/images/Adventure-Map-icon.png"), "Programs")
self.tabWidget.addTab(self.codeEditPage, QIcon(":/images/Sword-icon.png"), "Edit Program")
self.tabWidget.addTab(self.logsPage, QIcon(":/images/Spell-Scroll-icon.png"), "Logs")
self.tabWidget.addTab(self.documentation, QIcon(":/images/Spell-Book-icon.png"), "Documentation")
self.mainLayout = QVBoxLayout()
self.mainLayout.addWidget(self.tabWidget)
self.setLayout(self.mainLayout)
self.setWindowTitle(APP_NAME)
self.setWindowFlags(Qt.WindowStaysOnTopHint | Qt.Dialog)
self.resize(800, 600)
self.systrayHintMsgShowed = False
self.firstShow = True
self.fromQuit = False
def createProgramsList(self):
self.programsListModel = QStandardItemModel(0, 1, self)
self.programsList = QListView()
self.programsList.setModel(self.programsListModel)
self.programsListPage = QWidget()
self.programsListLayout = QVBoxLayout()
self.programsListButtons = QHBoxLayout()
self.programsListButtonNew = QPushButton("New")
self.programsListButtonDelete = QPushButton("Delete")
self.programsListButtonEdit = QPushButton("Edit")
self.programsListButtonSet = QPushButton("Set")
self.programsListButtons.addWidget(self.programsListButtonNew)
self.programsListButtons.addWidget(self.programsListButtonSet)
self.programsListButtons.addWidget(self.programsListButtonEdit)
self.programsListButtons.addWidget(self.programsListButtonDelete)
self.programsListLayout.addLayout(self.programsListButtons)
self.programsListLayout.addWidget(self.programsList)
self.programsListPage.setLayout(self.programsListLayout)
self.programsListButtonNew.clicked.connect(self.newProgram)
self.programsListButtonEdit.clicked.connect(self.editProgram)
self.programsListButtonDelete.clicked.connect(self.deleteProgram)
self.programsListButtonSet.clicked.connect(self.setProgram)
def createCodeEditPage(self):
self.codeEditPage = QWidget()
self.codeEditLayout = QVBoxLayout()
self.codeEditNameBox = QHBoxLayout()
self.codeEditNameBoxNameLabel = QLabel("Name:")
self.codeEditNameBoxNameInput = QLineEdit()
self.codeEditNameBoxSaveButton = QPushButton("Save")
self.codeEditNameBoxCancelButton = QPushButton("Cancel")
self.codeEditNameBox.addWidget(self.codeEditNameBoxNameLabel)
self.codeEditNameBox.addWidget(self.codeEditNameBoxNameInput)
self.codeEditNameBox.addWidget(self.codeEditNameBoxSaveButton)
self.codeEditNameBox.addWidget(self.codeEditNameBoxCancelButton)
self.codeEdit = QTextEdit()
self.codeEditLayout.addLayout(self.codeEditNameBox)
self.codeEditLayout.addWidget(self.codeEdit)
self.codeEditPage.setLayout(self.codeEditLayout)
self.codeEditLastCode = ''
self.codeEditNameBoxSaveButton.clicked.connect(self.saveEditProgram)
self.codeEditNameBoxCancelButton.clicked.connect(self.cancelEditProgram)
def showEvent(self, event):
super().showEvent(event)
if self.firstShow:
self.firstShow = False
self.createWaitDialog()
self.findKeyboard()
def closeWaitDialog(self):
time.sleep(1)
self.waitDialog.close()
@Slot()
def newProgram(self):
self.showNormal()
self.tabWidget.setCurrentWidget(self.codeEditPage)
self.codeEdit.setPlainText("")
self.codeEditLastCode = ''
self.codeEditNameBoxNameInput.setText("")
@Slot()
def editProgram(self):
selected = self.programsList.selectedIndexes()
if not selected:
return
selected = selected[0].data()
loadProgram = LoadProgram(self.cmdSocket, selected)
loadProgram.loaded.connect(self.programLoaded)
loadProgram.start()
self.showWaitDialog("Loading program ...")
@Slot()
def saveEditProgram(self):
name = self.codeEditNameBoxNameInput.text()
program = self.codeEdit.toPlainText()
if not name:
return
editProgramWorker = EditProgram(self.cmdSocket, name, program)
editProgramWorker.edited.connect(self.programSaved)
editProgramWorker.start()
self.showWaitDialog("Saving program ...")
@Slot()
def cancelEditProgram(self):
self.codeEdit.setPlainText(self.codeEditLastCode)
@Slot()
def deleteProgram(self):
selected = self.programsList.selectedIndexes()
if not selected:
return
selected = selected[0].data()
deleteWorker = DeleteProgram(self.cmdSocket, selected)
deleteWorker.deleted.connect(self.programDeleted)
deleteWorker.start()
self.showWaitDialog("Deleting program ...")
@Slot()
def setProgram(self):
selected = self.programsList.selectedIndexes()
if not selected:
return
selected = selected[0].data()
setWorker = SetProgram(self.cmdSocket, selected)
setWorker.setDone.connect(self.programSet)
setWorker.start()
self.showWaitDialog("Setting program ...")
@Slot(str, str)
def programLoaded(self, name, program):
self.closeWaitDialog()
self.tabWidget.setCurrentWidget(self.codeEditPage)
self.codeEdit.setPlainText(program)
self.codeEditLastCode = program
self.codeEditNameBoxNameInput.setText(name)
@Slot(str)
def programDeleted(self, name):
self.updateProgramsList()
@Slot(str)
def programSet(self, name):
self.updateProgramsList()
@Slot(str)
def programSaved(self, name):
self.updateProgramsList()
self.tabWidget.setCurrentWidget(self.programsListPage)
def createWaitDialog(self):
self.waitDialog = QDialog(self)
self.waitDialogLayout = QHBoxLayout()
self.waitDialogLabel = QLabel()
self.waitDialogLayout.addWidget(self.waitDialogLabel)
self.waitDialog.setLayout(self.waitDialogLayout)
self.waitDialog.setWindowFlags(Qt.Dialog | Qt.FramelessWindowHint)
def showWaitDialog(self, text='Please wait...'):
self.waitDialogLabel.setText(text)
self.waitDialog.exec_()
def findKeyboard(self):
find = ConnectKeyboard()
find.connected.connect(self.keyboardConnected)
find.start()
self.showWaitDialog('Finding and connecting Fruit2Pi Keyboard ...')
@Slot(BluetoothSocket)
def keyboardConnected(self, socket):
self.cmdSocket = socket
self.updateProgramsList()
def updateProgramsList(self):
self.listProgram = ListProgram(self.cmdSocket)
self.listProgram.listed.connect(self.programListUpdated)
self.listProgram.start()
@Slot(list, str)
def programListUpdated(self, programs, current_program):
print(programs)
print(current_program)
self.closeWaitDialog()
self.programsListModel.clear()
for p in programs:
item = QStandardItem(p)
if p == current_program:
item.setForeground(QBrush(QColor(0, 0, 255, 127)))
self.programsListModel.appendRow(item)
def setVisible(self, visible):
super().setVisible(visible)
def closeEvent(self, event):
if self.fromQuit:
return
if not event.spontaneous() or not self.isVisible():
return
if not self.systrayHintMsgShowed:
self.systrayHintMsgShowed = True
icon = QIcon(":/images/yammi-banana-icon.png")
self.trayIcon.showMessage(APP_NAME,
"Running on background"
"To quit, choose <b>Quit</b> in the icon menu",
icon,
5000
)
self.hide()
event.ignore()
@Slot(str)
def iconActivated(self, reason):
print(reason)
if reason == QSystemTrayIcon.Trigger:
self.showNormal()
if reason == QSystemTrayIcon.DoubleClick:
self.showNormal()
@Slot()
def showProgramsPage(self):
self.showNormal()
self.tabWidget.setCurrentWidget(self.programsListPage)
@Slot()
def showLogsPage(self):
self.showNormal()
self.tabWidget.setCurrentWidget(self.logsPage)
@Slot()
def showDocumentation(self):
self.showNormal()
self.tabWidget.setCurrentWidget(self.documentation)
@Slot()
def quit(self):
self.fromQuit = True
qApp.quit()
def createTrayIcon(self):
self.showProgramsAction = QAction("Programs", self)
self.showProgramsAction.triggered.connect(self.showProgramsPage)
self.showNewProgramAction = QAction("New Program", self)
self.showNewProgramAction.triggered.connect(self.newProgram)
self.showSetProgramAction = QAction("Logs", self)
self.showSetProgramAction.triggered.connect(self.showLogsPage)
self.showDocumentationAction = QAction("Documentation", self)
self.showDocumentationAction.triggered.connect(self.showDocumentation)
self.quitAction = QAction("Quit", self)
self.quitAction.triggered.connect(self.quit)
self.trayIconMenu = QMenu(self)
self.trayIconMenu.addAction(self.showProgramsAction)
self.trayIconMenu.addAction(self.showSetProgramAction)
self.trayIconMenu.addAction(self.showNewProgramAction)
self.trayIconMenu.addAction(self.showDocumentationAction)
self.trayIconMenu.addSeparator()
self.trayIconMenu.addAction(self.quitAction)
self.trayIcon = QSystemTrayIcon(self)
self.trayIcon.setContextMenu(self.trayIconMenu)
self.trayIcon.activated.connect(self.iconActivated)
self.trayIcon.setIcon(QIcon(":/images/yammi-banana-icon.png"))
self.trayIcon.show()