mirror of
https://github.com/aidygus/LinVAM.git
synced 2024-11-27 02:58:06 +11:00
222 lines
8.8 KiB
Python
222 lines
8.8 KiB
Python
|
from PyQt5.QtCore import *
|
||
|
|
from PyQt5.QtGui import *
|
||
|
|
from PyQt5.QtWidgets import *
|
||
|
|
from ui_profileeditwnd import Ui_ProfileEditDialog
|
||
|
|
from commandeditwnd import CommandEditWnd
|
||
|
|
import json
|
||
|
|
from profileexecutor import *
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class ProfileEditWnd(QDialog):
|
||
|
|
def __init__(self, p_profile, p_parent = None):
|
||
|
|
super().__init__(p_parent)
|
||
|
|
self.m_profiles = []
|
||
|
|
self.ui = Ui_ProfileEditDialog()
|
||
|
|
self.ui.setupUi(self)
|
||
|
|
self.m_profile = {}
|
||
|
|
|
||
|
|
self.ui.cmdTable.setHorizontalHeaderLabels(('Spoken command', 'Actions'))
|
||
|
|
self.ui.cmdTable.setSelectionBehavior(QAbstractItemView.SelectRows)
|
||
|
|
self.ui.cmdTable.setSelectionMode(QAbstractItemView.SingleSelection)
|
||
|
|
|
||
|
|
self.ui.cmdTable.horizontalHeader().setSectionResizeMode(0, QHeaderView.Interactive)
|
||
|
|
self.ui.cmdTable.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)
|
||
|
|
|
||
|
|
# define actions here
|
||
|
|
self.ui.newCmd.clicked.connect(self.slotNewCmd)
|
||
|
|
self.ui.editCmd.clicked.connect(self.slotEditCmd)
|
||
|
|
self.ui.cmdTable.doubleClicked.connect(self.slotEditCmd)
|
||
|
|
self.ui.deleteCmd.clicked.connect(self.slotDeleteCmd)
|
||
|
|
self.ui.ok.clicked.connect(self.slotOK)
|
||
|
|
self.ui.cancel.clicked.connect(self.slotCancel)
|
||
|
|
|
||
|
|
if p_profile == None or p_profile == {}:
|
||
|
|
self.ui.cmdTable.setRowCount(0)
|
||
|
|
return
|
||
|
|
|
||
|
|
self.ui.profileNameEdit.setText(p_profile['name'])
|
||
|
|
w_commands = p_profile['commands']
|
||
|
|
self.ui.cmdTable.setRowCount(len(w_commands))
|
||
|
|
i = 0
|
||
|
|
for w_command in w_commands:
|
||
|
|
self.ui.cmdTable.setItem(i, 0, QTableWidgetItem(w_command['name']))
|
||
|
|
w_text = json.dumps(w_command)
|
||
|
|
w_item = QTableWidgetItem(w_text)
|
||
|
|
w_item.setData(Qt.UserRole, json.dumps(w_command))
|
||
|
|
self.ui.cmdTable.setItem(i, 1, w_item)
|
||
|
|
i = i + 1
|
||
|
|
|
||
|
|
QTimer.singleShot(100, self.ui.cmdTable.resizeRowsToContents)
|
||
|
|
|
||
|
|
# def activateProfile(self, p_profile):
|
||
|
|
# if p_profile == None or p_profile == {}:
|
||
|
|
# self.ui.cmdTable.setRowCount(0)
|
||
|
|
# return
|
||
|
|
#
|
||
|
|
# w_commands = p_profile['commands']
|
||
|
|
# self.ui.cmdTable.setRowCount(len(w_commands))
|
||
|
|
# i = 0
|
||
|
|
# for w_command in w_commands:
|
||
|
|
# self.ui.cmdTable.setItem(i, 0, QTableWidgetItem(w_command['name']))
|
||
|
|
# w_text = json.dumps(w_command)
|
||
|
|
# w_item = QTableWidgetItem(w_text)
|
||
|
|
# w_item.setData(Qt.UserRole, json.dumps(w_command))
|
||
|
|
# self.ui.cmdTable.setItem(i, 1, w_item)
|
||
|
|
# i = i + 1
|
||
|
|
# self.ui.cmdTable.resizeRowsToContents()
|
||
|
|
|
||
|
|
# def updateActiveProfile(self):
|
||
|
|
# w_idx = self.m_curProfileIdx
|
||
|
|
# if w_idx < 0:
|
||
|
|
# return
|
||
|
|
#
|
||
|
|
# w_profile = {}
|
||
|
|
# w_profile['name'] = self.ui.profileCbx.itemText(w_idx)
|
||
|
|
# w_commands = []
|
||
|
|
#
|
||
|
|
# w_commandCnt = self.ui.cmdTable.rowCount()
|
||
|
|
# for w_i in range(w_commandCnt):
|
||
|
|
# w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
|
||
|
|
# w_command = json.loads(w_jsonCommand)
|
||
|
|
# w_commands.append(w_command)
|
||
|
|
# w_profile['commands'] = w_commands
|
||
|
|
# w_jsonProfile = json.dumps(w_profile)
|
||
|
|
# self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)
|
||
|
|
|
||
|
|
# def slotSelChanged(self, p_idx):
|
||
|
|
# self.updateActiveProfile()
|
||
|
|
# w_jsonProfile = self.ui.profileCbx.itemData(p_idx)
|
||
|
|
#
|
||
|
|
# w_profile = {}
|
||
|
|
# if w_jsonProfile != None:
|
||
|
|
# w_profile = json.loads(w_jsonProfile)
|
||
|
|
#
|
||
|
|
# self.activateProfile(w_profile)
|
||
|
|
#
|
||
|
|
# self.m_curProfileIdx = p_idx
|
||
|
|
|
||
|
|
# def importProfile(self, p_profile, p_update = True):
|
||
|
|
# if p_profile == None or p_profile == {}:
|
||
|
|
# return
|
||
|
|
#
|
||
|
|
# w_profileCnt = self.ui.profileCbx.count()
|
||
|
|
# for w_idx in range(w_profileCnt):
|
||
|
|
# w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
|
||
|
|
# if w_jsonProfile == None:
|
||
|
|
# continue
|
||
|
|
#
|
||
|
|
# w_profile = json.loads(w_jsonProfile)
|
||
|
|
# if p_profile['name'] == w_profile['name']:
|
||
|
|
# if p_update:
|
||
|
|
# w_jsonProfile = json.dumps(p_profile)
|
||
|
|
# self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)
|
||
|
|
# return True
|
||
|
|
# return False
|
||
|
|
#
|
||
|
|
# self.ui.profileCbx.addItem(p_profile['name'])
|
||
|
|
# w_jsonProfile = json.dumps(p_profile)
|
||
|
|
# self.ui.profileCbx.setItemData(w_profileCnt, w_jsonProfile)
|
||
|
|
# return True
|
||
|
|
|
||
|
|
# def slotAddNewProfile(self):
|
||
|
|
# text, okPressed = QInputDialog.getText(self, "Get Profile Name", "Profile name:", QLineEdit.Normal, "")
|
||
|
|
# if okPressed and text != '':
|
||
|
|
# w_profile = {}
|
||
|
|
# w_profile['name'] = text
|
||
|
|
# w_profile['commands'] = []
|
||
|
|
# if self.importProfile(w_profile, False) == False:
|
||
|
|
# QMessageBox.critical(None, 'Error', 'Adding a new profile was failed')
|
||
|
|
# return
|
||
|
|
# self.ui.profileCbx.setCurrentIndex(self.ui.profileCbx.count()-1)
|
||
|
|
|
||
|
|
# def slotRemoveProfile(self):
|
||
|
|
# w_curIdx = self.ui.profileCbx.currentIndex()
|
||
|
|
# if w_curIdx >= 0:
|
||
|
|
# self.ui.cmdTable.setRowCount(0)
|
||
|
|
# self.m_curProfileIdx = -1
|
||
|
|
# self.ui.profileCbx.removeItem(w_curIdx)
|
||
|
|
#
|
||
|
|
# def slotRenameProfile(self):
|
||
|
|
# w_curIdx = self.ui.profileCbx.currentIndex()
|
||
|
|
# if w_curIdx >= 0:
|
||
|
|
# text, okPressed = QInputDialog.getText(self, "Input Dialog", "Profile name:",
|
||
|
|
# QLineEdit.Normal, self.ui.profileCbx.itemText(w_curIdx))
|
||
|
|
# if okPressed and text != '':
|
||
|
|
# self.ui.profileCbx.setItemText(w_curIdx, text)
|
||
|
|
# w_jsonProfile = self.ui.profileCbx.itemData(w_curIdx)
|
||
|
|
# w_profile = json.loads(w_jsonProfile)
|
||
|
|
# w_profile['name'] = text
|
||
|
|
# w_jsonProfile = json.dumps(w_profile)
|
||
|
|
# self.ui.profileCbx.setItemData(w_curIdx, w_jsonProfile)
|
||
|
|
|
||
|
|
def importCommand(self, p_command, p_update):
|
||
|
|
w_commandCnt = self.ui.cmdTable.rowCount()
|
||
|
|
for w_i in range(w_commandCnt):
|
||
|
|
w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
|
||
|
|
w_command = json.loads(w_jsonCommand)
|
||
|
|
if w_command['name'] == p_command['name']:
|
||
|
|
if p_update:
|
||
|
|
w_text = json.dumps(p_command)
|
||
|
|
w_item = QTableWidgetItem(w_text)
|
||
|
|
w_item.setData(Qt.UserRole, json.dumps(p_command))
|
||
|
|
self.ui.cmdTable.setItem(w_i, 1, w_item)
|
||
|
|
self.ui.cmdTable.resizeRowsToContents()
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
w_rowCnt = self.ui.cmdTable.rowCount()
|
||
|
|
self.ui.cmdTable.setRowCount(w_rowCnt + 1)
|
||
|
|
self.ui.cmdTable.setItem(w_rowCnt, 0, QTableWidgetItem(p_command['name']))
|
||
|
|
w_text = json.dumps(p_command)
|
||
|
|
w_item = QTableWidgetItem(w_text)
|
||
|
|
w_item.setData(Qt.UserRole, json.dumps(p_command))
|
||
|
|
self.ui.cmdTable.setItem(w_rowCnt, 1, w_item)
|
||
|
|
self.ui.cmdTable.resizeRowsToContents()
|
||
|
|
|
||
|
|
def slotNewCmd(self):
|
||
|
|
w_cmdEditWnd = CommandEditWnd(None, self)
|
||
|
|
if w_cmdEditWnd.exec_() == QDialog.Accepted:
|
||
|
|
if self.importCommand(w_cmdEditWnd.m_command, False) == False:
|
||
|
|
QMessageBox.critical(None, 'Error', 'Adding a new command was failed')
|
||
|
|
return
|
||
|
|
self.ui.cmdTable.selectRow(self.ui.cmdTable.rowCount() - 1)
|
||
|
|
self.ui.cmdTable.setFocus()
|
||
|
|
|
||
|
|
def slotEditCmd(self):
|
||
|
|
w_modelIdxs = self.ui.cmdTable.selectionModel().selectedRows()
|
||
|
|
if len(w_modelIdxs) == 0:
|
||
|
|
return
|
||
|
|
w_modelIdx = w_modelIdxs[0]
|
||
|
|
w_jsonCommand = self.ui.cmdTable.item(w_modelIdx.row(), 1).data(Qt.UserRole)
|
||
|
|
w_command = json.loads(w_jsonCommand)
|
||
|
|
|
||
|
|
w_cmdEditWnd = CommandEditWnd(w_command, self)
|
||
|
|
if w_cmdEditWnd.exec_() == QDialog.Accepted:
|
||
|
|
self.importCommand(w_cmdEditWnd.m_command, True)
|
||
|
|
self.ui.cmdTable.resizeRowsToContents()
|
||
|
|
|
||
|
|
def slotDeleteCmd(self):
|
||
|
|
w_modelIdxs = self.ui.cmdTable.selectionModel().selectedRows()
|
||
|
|
i = 0
|
||
|
|
for w_modelIdx in sorted(w_modelIdxs):
|
||
|
|
self.ui.cmdTable.removeRow(w_modelIdx.row() - i)
|
||
|
|
i = i + 1
|
||
|
|
self.ui.cmdTable.setFocus()
|
||
|
|
|
||
|
|
def slotOK(self):
|
||
|
|
self.m_profile = {}
|
||
|
|
self.m_profile['name'] = self.ui.profileNameEdit.text()
|
||
|
|
w_commands = []
|
||
|
|
|
||
|
|
w_commandCnt = self.ui.cmdTable.rowCount()
|
||
|
|
for w_i in range(w_commandCnt):
|
||
|
|
w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
|
||
|
|
w_command = json.loads(w_jsonCommand)
|
||
|
|
w_commands.append(w_command)
|
||
|
|
self.m_profile['commands'] = w_commands
|
||
|
|
|
||
|
|
super().accept()
|
||
|
|
|
||
|
|
def slotCancel(self):
|
||
|
|
super().reject()
|