Пример #1
0
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()
Пример #2
0
	def startExport(self):
		self.dialog = SSHExport([self.configparser.configfile], self.ui.privKeyPath.text(), 
			list(self.targets), self)
		self.dialog.execThreads()
		self.dialog.show()