class ExportPage(QtGui.QWidget):
"""Creates the main Export page"""
def __init__(self, configparser, moduleparser, leftList, parent=None):
"""Instantiate a ExportPage object
@param self A ExportPage instance
@param configparser A LTCConfigParser instance
@param leftList The leftMenu QListWidget, to create the tab
@param parent A QtGui.QWidget parent object
"""
self.moduleparser = moduleparser
self.configparser = configparser
self.leftList = leftList
self.parent = parent
self.targets = {}
self.threads = {}
QtGui.QWidget.__init__(self, parent)
self.ui = Ui_ExportWidget()
self.ui.setupUi(self)
self.tab = LeftMenuItem(leftList, self.tr("Export"),
QtGui.QIcon(":/export_icon/fork.png"))
self.fscompleter = QtGui.QCompleter(self)
self.fscompleter.setModel(QtGui.QDirModel(self.fscompleter))
self.ui.privKeyPath.setCompleter(self.fscompleter)
self.checkPrivKeyFile()
def browsePrivKeyFile(self):
"""Browse files and set ui.pathLineEdit with given path
@param self A ExportPage instance
"""
filepath = QtGui.QFileDialog.getOpenFileName(self, self.tr("Import File"),
os.path.expanduser('~root/.ssh'))
if filepath:
self.ui.privKeyPath.setText(filepath)
def checkPrivKeyFile(self, path=''):
"""Verify if path on self.ui.pathLineEdit exists.
Also check if has targets, and set lineedit text color as red
and apply button disabled if not
@param self A ExportPage instance
@param path A filepath string
"""
if not path:
path = self.ui.privKeyPath.text()
if os.path.isfile(path):
# Set lineedit text color to default
self.ui.privKeyPath.setStyleSheet("")
else:
# Set lineedit text color as red
self.ui.privKeyPath.setStyleSheet("color: red;")
self.checkConfigs()
def checkConfigs(self):
if os.path.isfile(self.ui.privKeyPath.text()) and self.targets:
for b in self.ui.buttonBox.buttons():
if self.ui.buttonBox.standardButton(b)\
== QtGui.QDialogButtonBox.Apply:
b.setEnabled(True)
else:
for b in self.ui.buttonBox.buttons():
if self.ui.buttonBox.standardButton(b)\
== QtGui.QDialogButtonBox.Apply:
b.setEnabled(False)
def addTargetsClicked(self):
"""Launch dialog to add targets to scan list
@param self A ExportPage instance
"""
addTargetsDialog = AddTargets(self)
scanTargetsDialog = ScanTargets(self)
addTargetsDialog.show()
addTargetsDialog.ipList.connect(scanTargetsDialog.show)
# update self.targets with targets dict from scanTargetsDialog
scanTargetsDialog.ipDict.connect(self.targets.update)
scanTargetsDialog.ipDict.connect(self.refreshTargets)
def removeTargetsClicked(self):
"""Remove selected targets from targets list
@param self A ExportPage instance
"""
for i in self.ui.treeWidget.selectedItems():
if i.isSelected() and i.text(0) in self.targets:
self.targets.pop(i.text(0))
#except Exception as e: print("This must not has happened:", e)
self.refreshTargets()
def refreshTargets(self):
print("__refreshed")
D = []
for i in self.targets:
if i not in self.threads:
self.threads[i] = ThreadedScan(i, self.targets[i],
parent=self.ui.treeWidget)
for i in self.threads:
if i not in self.targets:
D.append(i)
for i in D:
I = self.threads.pop(i)
# Should be a better way to delete a tree item
index = self.ui.treeWidget.indexOfTopLevelItem(I.treeItem)
self.ui.treeWidget.takeTopLevelItem(index)
self.checkPrivKeyFile()
self.checkConfigs()
def rescan(self):
print("__rescanning")
for i in self.threads:
self.threads[i].start()
def buttonBoxClicked(self, button):
if self.ui.buttonBox.standardButton(button) == QtGui.QDialogButtonBox.Reset:
self.targets = {}
self.ui.privKeyPath.setText("/root/.ssh/id_rsa")
self.refreshTargets()
self.checkPrivKeyFile()
elif self.ui.buttonBox.standardButton(button) == QtGui.QDialogButtonBox.Apply:
self.startExport()
def startExport(self):
self.dialog = SSHExport([self.configparser.configfile], self.ui.privKeyPath.text(),
list(self.targets), self)
self.dialog.execThreads()
self.dialog.show()
def startExport(self): self.dialog = SSHExport([self.configparser.configfile], self.ui.privKeyPath.text(), list(self.targets), self) self.dialog.execThreads() self.dialog.show()
