usbinstall/usbinstall.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python

from gui import * 	# Importa la parte grafica

import gettext      

from subprocess import Popen
# La parte di dbus
import gobject
import dbus
import dbus.service
from dbus.mainloop.glib import DBusGMainLoop
if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
        import dbus.glib
        
gettext.install('usbinstall', '/usr/share/locale', unicode=1)

# Variabili globali
passo = 1
p = "" # Processo del file .sh
ext 	  = ""	# Vedi apri_file()
path_Boot = ""	# Percorso file.cpio.gz
path_Iso  = ""	# Percorso file.iso
sel_usb_file= ""# Chiavetta per l'installazione (file)
sel_usb_nome= ""# Modello
usb_file  = []	# Lista chiavette disponibili
usb_nome  = []	# Rispettivi modelli


## Assegna gli eventi agli oggetti del form
def assegna_eventi():
	app.connect(ui.b_avanti, QtCore.SIGNAL('clicked()'), avanti)
	app.connect(ui.b_indietro, QtCore.SIGNAL('clicked()'), indietro)
	app.connect(ui.b_apri_file, QtCore.SIGNAL('clicked()'), apri_file)
	app.connect(ui.b_help, QtCore.SIGNAL('clicked()'), gestisci_nuvola)
	app.connect(ui.n_chiudi, QtCore.SIGNAL('clicked()'), chiudi_nuvola)
	app.connect(ui.b_refresh, QtCore.SIGNAL('clicked()'), lista_usb)
	app.connect(ui.l_usb, QtCore.SIGNAL("activated(int)"), seleziona_usb)


## Gestione del bottone b_avanti
def avanti():
	# Prima controlla che tutto sia a posto per passare al passo successivo
	global passo, path_Boot, path_Iso, sel_usb_file, sel_usb_nome
	
	if passo == 1:
		if path_Boot == "":
			msg_warn(_("A file .cpio is required"))
			return
		else:
			passo += 1
			passo2()
			
	elif passo == 2:
		if path_Iso == "":
			msg_warn(_("An openmamba's iso is required"))
			return
		else:
			passo += 1
			passo3()
	
	elif passo == 3:
		if sel_usb_file == "":
			msg_warn(_("You must choose a usb key"))
			return
		else:
			msg = QtGui.QMessageBox.question(None, _("Attention"), _("All data on the key ") + sel_usb_nome + _(" will be lost. \n")+_("Are you sure to continue?"), 
				QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.Yes)
			# Continua solo se l'utente preme il bottone Yes
			if msg == QtGui.QMessageBox.Yes:
				passo += 1
				passo4()
			else:
				return

	
	
## Gestione del bottone b_indietro		
def indietro():
	global passo
	passo -= 1
	
	if passo == 0:
		app.exit(1)
		print _("Application terminated by pressing \"cancel\" button")
		
	elif passo == 1:
		passo1()
		
	elif passo == 2:
		passo2()
		
	elif passo == 3:
		passo3()
	
	
## Dialogo in cui si scegle i file per l'installazione
## Inoltre setta le variabili path_Boot e path_Iso e aggiorna la casella di testo con il percorso
def apri_file():
	global path_Boot, path_Iso, ext
	if ext == "boot":
		fileType = QtCore.QString("cpio *.cpio.gz")
		nfileName = QtGui.QFileDialog.getOpenFileName(None, _("Choose the boot file"), "~", "File .cpio.gz (*.cpio.gz)",  fileType)
		path_Boot = nfileName
		ui.t_file.setText(QtGui.QApplication.translate("Dialog", nfileName, None, QtGui.QApplication.UnicodeUTF8))
	elif ext == "iso":
		fileType = QtCore.QString("iso *.iso")
		nfileName = QtGui.QFileDialog.getOpenFileName(None, _("Choose the iso file"), "~", "File .iso (*.iso)",  fileType) #26/10 path_Boot -> "~"
		path_Iso = nfileName
		ui.t_file.setText(QtGui.QApplication.translate("Dialog", nfileName, None, QtGui.QApplication.UnicodeUTF8))


## Visualizza l'aiuto
def gestisci_nuvola():
	global passo
	# Aiuto in base al passo corrente
	if passo == 1:
		aiuto = _("This archive is designed to make the usb bootable.") + _(" You can find this file <a style=\"color: green; \"href=\"http://www.openmamba.org/distribution/media/bootusb.html?lang=en\">here</a>") 
		# Cambiata anche la lingua della pagina (lang=en) 
	elif passo == 2:
		aiuto = _("The file .iso contains the operating system. You can download it from <a style=\"color: green; \"href=\"http://www.openmamba.org/distribution/download.html?lang=en\">here</a>")
	elif passo == 3:
		aiuto = _("Choose an usb key for the installation.                    <span style=\"color:red;\">All data will be lost</span>")
	ui.n_testo.setText(QtGui.QApplication.translate("Dialog", aiuto, None, QtGui.QApplication.UnicodeUTF8))
	ui.n_img.show() # Immagine della nuvola
	ui.n_testo.show() # Testo della nuvola
	ui.n_chiudi.show()

## Nasconde la nuvola
def chiudi_nuvola():
	ui.n_img.hide() 
	ui.n_testo.hide()
	ui.n_chiudi.hide()

## Lista periferiche usb e le aggiuge alla lista (widget)
def lista_usb():
	import dbus
	global usb_file, usb_nome
	
	# Cancella i valori precedenti
	usb_file  = []	
	usb_nome  = []	
	ui.l_usb.clear()
	i = 0
	
	# Riceve da dbus le chiavette collegate al pc
	bus = dbus.SystemBus()
	ud_manager_obj = bus.get_object('org.freedesktop.UDisks', '/org/freedesktop/UDisks')
	ud_manager = dbus.Interface(ud_manager_obj, 'org.freedesktop.UDisks')
	for dev in ud_manager.EnumerateDevices():
		device_obj = bus.get_object('org.freedesktop.UDisks', dev)
		device_props = dbus.Interface(device_obj, dbus.PROPERTIES_IFACE)
		# Se lo spazio e' minore di 1GB salta l'usb corrente
		if(device_props.Get('org.freedesktop.UDisks.Device', 'PartitionSize')) > 1073741823:
			# Se e' un device rimuovibile (== chiavetta usb)
			if(device_props.Get('org.freedesktop.UDisks.Device', 'DeviceIsRemovable')):
				usb_file.append(device_props.Get('org.freedesktop.UDisks.Device', 'DeviceFile'))
				usb_nome.append(device_props.Get('org.freedesktop.UDisks.Device', 'DriveModel'))
				ui.l_usb.addItem(usb_file[i] + " " + usb_nome[i])
				i += 1
				
	# Imposta il primo come quello di default
	if len(usb_file) != 0:			
		seleziona_usb(0)
	else:
		print _("No usb available > 1 Gb")
			
			
## Abbreviazione di un warning
def msg_warn(testo):
	msg = QtGui.QMessageBox.warning(None, _("Attention"), testo)	

## Aggiorna gli oggetti al cambio del passo
def aggiorna_oggetti(immagine, titolo, descrizione):
	# Funzioni che servono solo qua dentro
	def nascondi_oggetti():
		ui.l_usb.hide()
		ui.b_refresh.hide()
		ui.t_file.hide()
		ui.pach1.hide()
		ui.b_apri_file.hide()
		
	def visualizza_lista_usb():
		ui.l_usb.show()
		ui.b_refresh.show()
		
	def visualizza_scegli_file():
		ui.t_file.show()
		ui.pach1.show()
		ui.b_apri_file.show()
	
	global passo, ext, path_Boot, path_Iso
	
	nascondi_oggetti()
	if passo == 1:
		ext   = "boot"
		path_Boot = ""
		ui.b_indietro.setText(QtGui.QApplication.translate("Dialog", _("Cancel"), None, QtGui.QApplication.UnicodeUTF8))
		# Visualizzo gli oggetti del passo 1
		visualizza_scegli_file()
		# Chiudi la nuovola dell'help al cambio di passo
		chiudi_nuvola()		
	elif passo == 2:
		ext   = "iso"
		path_Iso = ""
		ui.b_indietro.setText(QtGui.QApplication.translate("Dialog", _("Go back"), None, QtGui.QApplication.UnicodeUTF8))
		visualizza_scegli_file()
		# Chiudi la nuovola dell'help al cambio di passo
		chiudi_nuvola()
	elif passo == 3:
		visualizza_lista_usb()
		lista_usb()
		# Chiudi la nuovola dell'help al cambio di passo
		chiudi_nuvola()
	elif passo == 4:
		chiudi_nuvola()
	ui.t_file.setText(QtGui.QApplication.translate("Dialog", "", None, QtGui.QApplication.UnicodeUTF8))  		
	ui.i_back.setPixmap(QtGui.QPixmap(immagine))
	Dialog.setWindowTitle(QtGui.QApplication.translate("Dialog", titolo, None, QtGui.QApplication.UnicodeUTF8))
	ui.l_descrizione.setText(QtGui.QApplication.translate("Dialog", descrizione, None, QtGui.QApplication.UnicodeUTF8))
	
	

## Setta la variabili globali sel_usb_* con l'usb scelto
def seleziona_usb(i):
	global sel_usb_file, usb_file, sel_usb_nome, usb_nome
	sel_usb_file = usb_file[i]
	sel_usb_nome = usb_nome[i]
	


### Inizio dei vari passi


def passo1():
	aggiorna_oggetti("img/back1.png", _("Step 1/4 - choice the boot file"), _("Choose the .cpio file"))	
	
def passo2():
	aggiorna_oggetti("img/back2.png", _("Step 2/4 - choice the iso file"), _("Choose the openmamba .iso file"))	
	
def passo3():
	aggiorna_oggetti("img/back3.png", _("Step 3/4 - choice of usb key"), _("Choose the usb key"))	
	

	
def passo4():
	global p
	aggiorna_oggetti("img/back4.png", _("Step 4/4 - installation of openmamba to usb"), _("Installation in progress..."))	
	
	# Nasconde gli oggetti superflui:
	ui.b_help.hide()
	ui.line.hide()
	ui.b_indietro.hide()
	ui.b_avanti.hide()
	# Mostra la progressBar
	ui.p_installazione.show()
	init_ascolto()					
	p = Popen(["./usbinstall.sh", "start", sel_usb_file, path_Boot, path_Iso])
	print _("Installation process started")

	
def init_ascolto():
	session_bus = dbus.SessionBus()
	bus_name = dbus.service.BusName('org.openmamba.usbinstall', bus=session_bus)
	dbus_loop = setta_dbus(bus_name)
	bus = dbus.SessionBus(mainloop=dbus_loop)



class setta_dbus(dbus.service.Object):
    def __init__(self, bus_name, object_path='/org/openmamba/usbinstall'):
        dbus.service.Object.__init__(self, bus_name, object_path)
    
    @dbus.service.method('org.openmamba.usbinstall')
    def Pbar_aggiorna(self, p):
	ui.p_installazione.setProperty("value", int(p))
        if p == 100:
		installazione_terminata()
	return 0

    @dbus.service.method('org.openmamba.usbinstall')
    def Descrizione(self, desc):
	ui.l_descrizione.setText(QtGui.QApplication.translate("Dialog",desc, None, QtGui.QApplication.UnicodeUTF8))
        return 0

    @dbus.service.method('org.openmamba.usbinstall')
    def Esci(self):
	mainloop.quit()
	app.exit(0)

# Termina lo scipt e dbus insieme alla finestra
def closeEvent(form, event):
	installazione_terminata()


def installazione_terminata():
	global p
	msg = QtGui.QMessageBox.information(None, _("Information"), _("Installation process completed"))
	p.terminate()
	mainloop.quit()
	app.exit(0)       

# Inizializza la parte grafica
app = QtGui.QApplication([])
Dialog = QtGui.QDialog()
ui = Ui_MainWindow()
ui.setupUi(Dialog)
mainloop = gobject.MainLoop()

def run():
	assegna_eventi()
	passo1()
	ui.b_avanti.setText(QtGui.QApplication.translate("Dialog", _("Go on"), None, QtGui.QApplication.UnicodeUTF8))
	Dialog.show() # Mostra il dialog precedentemente creato
	
	return app.exec_()
	
if __name__ == "__main__":
	run()