示例#1
0
	def __init__(self):
		"""Creates a Main() instance, and setup the mainWindow options

		@param	self		A Main() instance
		"""
		QtGui.QMainWindow.__init__(self)

		# Setup main UI
		self.ui = Ui_ThinClient()
		self.ui.setupUi(self)
		self.hide()

#		self.ui.toolBar.addWidget(QtGui.QSpacerItem(40, 20,
#			QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
		self.ui.toolBar.addWidget(self.ui.nameEdit)

		self.ui.topBar.close()
		
		# Check SSH key
		if not os.path.isfile(os.path.expanduser("~/.ssh/id_rsa")) and\
			QtGui.QMessageBox.question(self, self.tr("SSH private key not found"),
			self.tr('SSH RSA Private Key File "~/.ssh/id_rsa" not found.\n'+
			'Do you want to generate a RSA private/public keys pair now?'),
			QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.Yes)\
			== QtGui.QMessageBox.Yes:
			self.keyGen()

		# Init users and profiles package
		self.moduleparser = LTCModuleParser()
		self.configparser = LTCConfigParser(self.moduleparser)

		if os.path.isfile(configfile):
			self.configparser.readConfigFile(configfile)
		else:
			self.newConfigFile(first=True)

		# Set config file name
		self.ui.nameEdit.setText(self.configparser.getName())

		# Users widget configuration routines
		self.Users = UsersPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Users)
		#self.Users.hide()

		# Edit widget configuration routines
		self.Edit = EditPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Edit)
		self.Edit.hide()

		# Export widget configuration routines
		self.Export = ExportPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Export)
		self.Export.hide()

		summaryAction = self.Users.ui.summaryDock.toggleViewAction()
		self.ui.menuView.addAction(summaryAction)

		self.current = self.Users
		self.ui.listWidget.item(0).setSelected(True)
		self.show()
示例#2
0
class Main(QtGui.QMainWindow):
	def __init__(self):
		"""Creates a Main() instance, and setup the mainWindow options

		@param	self		A Main() instance
		"""
		QtGui.QMainWindow.__init__(self)

		# Setup main UI
		self.ui = Ui_ThinClient()
		self.ui.setupUi(self)
		self.hide()

#		self.ui.toolBar.addWidget(QtGui.QSpacerItem(40, 20,
#			QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum))
		self.ui.toolBar.addWidget(self.ui.nameEdit)

		self.ui.topBar.close()
		
		# Check SSH key
		if not os.path.isfile(os.path.expanduser("~/.ssh/id_rsa")) and\
			QtGui.QMessageBox.question(self, self.tr("SSH private key not found"),
			self.tr('SSH RSA Private Key File "~/.ssh/id_rsa" not found.\n'+
			'Do you want to generate a RSA private/public keys pair now?'),
			QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.Yes)\
			== QtGui.QMessageBox.Yes:
			self.keyGen()

		# Init users and profiles package
		self.moduleparser = LTCModuleParser()
		self.configparser = LTCConfigParser(self.moduleparser)

		if os.path.isfile(configfile):
			self.configparser.readConfigFile(configfile)
		else:
			self.newConfigFile(first=True)

		# Set config file name
		self.ui.nameEdit.setText(self.configparser.getName())

		# Users widget configuration routines
		self.Users = UsersPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Users)
		#self.Users.hide()

		# Edit widget configuration routines
		self.Edit = EditPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Edit)
		self.Edit.hide()

		# Export widget configuration routines
		self.Export = ExportPage(self.configparser, self.moduleparser,
			self.ui.listWidget, self)
		self.ui.horizontalLayout.addWidget(self.Export)
		self.Export.hide()

		summaryAction = self.Users.ui.summaryDock.toggleViewAction()
		self.ui.menuView.addAction(summaryAction)

		self.current = self.Users
		self.ui.listWidget.item(0).setSelected(True)
		self.show()

	def activateTab(self, listItem):
		"""Show the widget of selected Tab

		@param	self		A Main() instance
		@param	listItem	A QtGui.QListWidgetItem object
		"""
		self.current.hide()
		if listItem == self.Users.tab:
			self.Users.show()
			self.current = self.Users
		elif listItem == self.Edit.tab:
			self.Edit.show()
			self.current = self.Edit
		elif listItem == self.Export.tab:
			self.Export.show()
			self.current = self.Export

	def configNameChanged(self, text):
		"""When config file name was changed, set it to configfile

		Qt Slot, called by UI
		@param	self		A Main window instance
		"""
		self.configparser.setName(text)

	def closeEvent(self, event):
		"""Reimplementation of closeEvent Qt event

		Exec writeConfigFile in self.configparser before close main window
		"""
		if not self.configparser.modified():
			event.accept()
		else:
			ret = QtGui.QMessageBox.question(self, self.tr("LTMT"),
				self.tr("The configuration file has been modified.")+
				"\n"+self.tr("Do you want to save your changes?"),
				QtGui.QMessageBox.Save | QtGui.QMessageBox.Discard |\
				QtGui.QMessageBox.Cancel, QtGui.QMessageBox.Save)
			if ret == QtGui.QMessageBox.Save:
				self.saveConfigFile()
				event.accept()
			elif ret == QtGui.QMessageBox.Discard:
				os.remove(self.configparser.backupfile)
				self.newConfigFile()
				event.accept()
			else:
				event.ignore()

	def newConfigFile(self, first=False):
		"""Creates a new config file

		@param	self		A Main window instance
		@param	first		Bool. True if the first operation in
							configparser, when there is no configfile set on it
		"""
		if not first:
			event = QtCore.QEvent(QtCore.QEvent.User)
			self.closeEvent(event)
			if not event.isAccepted(): return

		self.configparser.newConfigFile()
		if not first:
			self.Users.updateLists()
			self.Edit.updateLists()

	def openConfigFile(self):
		"""Open file dialog to select and open a configuration

		If there is a backup file, ask if admin want to recover it
		@param	self		A Main window instance
		"""
		event = QtCore.QEvent(QtCore.QEvent.User)
		self.closeEvent(event)
		if not event.isAccepted(): return

		file = QtGui.QFileDialog.getOpenFileName(self,
			self.tr("Open Config File"), os.path.abspath("thinclient.conf"))
		if file and self.configparser.modified(file)==1:
			reply = QtGui.QMessageBox.question(self, self.tr("Backup found"),
            self.tr("A backup file of <b>{0}</b> was found.\n"+
				"Do you want to restore it?")\
			.format(file), QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
			QtGui.QMessageBox.No)
			# If yes, write backup to file
			if reply == QtGui.QMessageBox.Yes:
				with open(file, 'w') as orig, open(file + '~', 'r') as bkp:
					orig.write(bkp.read())
			# finally, remove bkp
			try: os.remove(file + '~')
			except: pass

		if file:
			self.configparser.readConfigFile(file)
			# If file outdated, ask for update
			if self.configparser.getOptionsList() != self.moduleparser.getModulesList():
				if QtGui.QMessageBox.question(self,
				self.tr("Outdated Config File"), self.tr("This config file "+
				"has absent and/or hasn't present modules.\n"+
				"You want to update it (required to proceed)?\n"+
				"P.S.: New modules will be added to configurations "+
				"but absent, present in configuration, won't be touched "+
				"or shown."), QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
				QtGui.QMessageBox.No) == QtGui.QMessageBox.Yes:
					self.configparser.updateConfigFile()
				else:
					self.newConfigFile()
			self.ui.nameEdit.setText(self.configparser.getName())
			self.ui.statusbar.showMessage(self.tr("Editing file: {0}")\
				.format(file), 0)
			self.Users.updateLists()
			self.Edit.updateLists()

	def saveConfigFile(self):
		"""Save current opened config file

		@param	self		 A Main window instance
		"""
		if self.configparser.modified()==1:
			self.configparser.writeConfigFile()
			self.ui.statusbar.showMessage(self.tr("File \"{0}\" saved!").format(
				self.configparser.configfile), 5000)
		elif self.configparser.modified()==2:
			self.saveAsConfigFile()

	def saveAsConfigFile(self):
		"""Open Save as dialog to choose where to save current config file

		@param	self		 A Main window instance
		"""
		# TODO: change work dir of save dialog to /etc
		file = QtGui.QFileDialog.getSaveFileName(self, self.tr("Save Config File as"),
			"thinclient.conf", self.tr("Conf files (*.conf) (*.conf);; "+
			"XML files (*.xml) (*.xml);; All files (*) (*)"))
		if file:
			self.configparser.writeConfigFile(file)
			self.ui.statusbar.showMessage(self.tr("File \"{0}\" saved!").format(
				self.configparser.configfile), 5000)

	def editKeys(self):
		"""Open EditKeys dialog

		@param	self		A Main window instance
		"""
		EditKeys(self.configparser, self).exec_()
	
	def keyGen(self):
		"""Open Key Generation dialog
		
		@param	self		A Main window instance
		"""
		KeyGen(self).exec_()

	def aboutLTMT(self):
		"""Show a LTMT 'About' dialog

		@param	self		A Main window instance
		"""
		helpText = self.tr("Librix ThinClient Management Tool (LTMT) is a program "+
			"developed by <a href='http://librixdev.las.ic.unicamp.br'>"+
			"Librix Dev Team</a>, also responsible by Librix Linux gentoo-based "+
			"distribution, and it's embedded tools, in association with "+
			"<a href='http://www.itautec.com.br'>Itautec</a> and "+
			"<a href='http://www.las.ic.unicamp.br'>Administration and "+
			"Security Laboratory of Unicamp</a>.")
		QtGui.QMessageBox.about(self,  self.tr("Librix ThinClient Management Tool"),
			helpText)

	def aboutQt(self):
		"""Show a Qt 'About' dialog

		@param	self		A Main window instance
		"""
		QtGui.QMessageBox.aboutQt(self)