#!/bin/env python

 # -*- coding: utf-8 -*-

 # ------------------------------------------------------------

 # format_drive_dialog.pyw

 # 

 # letting the user format a drive attached to a certain VM

 #

 # Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>

 #

 # Copyright (C) 2014 AIT Austrian Institute of Technology

 # AIT Austrian Institute of Technology GmbH

 # Donau-City-Strasse 1 | 1220 Vienna | Austria

 # http://www.ait.ac.at

 #

 # This program is free software; you can redistribute it and/or

 # modify it under the terms of the GNU General Public License

 # as published by the Free Software Foundation version 2.

 # 

 # This program is distributed in the hope that it will be useful,

 # but WITHOUT ANY WARRANTY; without even the implied warranty of

 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 # GNU General Public License for more details.

 # 

 # You should have received a copy of the GNU General Public License

 # along with this program; if not, write to the Free Software

 # Foundation, Inc., 51 Franklin Street, Fifth Floor, 

 # Boston, MA  02110-1301, USA.

 # ------------------------------------------------------------

 # ------------------------------------------------------------

 # imports

 import base64

 import sys

 import urllib

 import urllib2

 from PyQt4 import QtCore

 from PyQt4 import QtGui

 from ui_FormatDriveDialog import Ui_FormatDriveDialog 

 from about_dialog import AboutDialog

 # ------------------------------------------------------------

 # code

 class FormatDriveDialog(QtGui.QDialog):

     """A dialog for letting the user pass on a password/keyfile combo"""

     def __init__(self, ip):

         QtGui.QDialog.__init__(self)

         user_text = """

 <b>Attention!</b><br/>

 You are going to wipe all data stored<br/>

 at device attached to IP <b>%s</b>.<br/>

 <br/>

 <b>This is irreversible.</b><br/>

 <br/>

 Please provide an appropriate password or keyfile to proceed:

 """ % ip

         # setup the user interface

         self.ui = Ui_FormatDriveDialog()

         self.ui.setupUi(self)

         # local members

         self._about_dialog = AboutDialog()

         self._open_file_dialog = QtGui.QFileDialog(self, 'Open Keyfile', QtCore.QDir.homePath(), 'All Files (*.*);;')

         self._open_file_dialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)

         self._open_file_dialog.setFileMode(QtGui.QFileDialog.ExistingFile)

         # connectors

         self.ui.lblText.setText(user_text)

         self.ui.btnAbout.clicked.connect(self.clicked_about)

         self.ui.btnCancel.clicked.connect(self.reject)

         self.ui.btnOk.clicked.connect(self.clicked_ok)

         self.ui.btnBrowse.clicked.connect(self.clicked_browse)

         # positionate ourself central

         screen = QtGui.QDesktopWidget().screenGeometry()

         size = self.geometry()

         self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2)

     def clicked_about(self):

         """About button has been clicked."""

         self._about_dialog.show()

     def clicked_browse(self):

         """Browse button has been clicked."""

         if self._open_file_dialog.exec_() == QtGui.QDialog.Accepted:

             self.ui.edtKeyfile.setText(self._open_file_dialog.selectedFiles()[0])

     def clicked_ok(self):

         """Ok button has been clicked."""

         init_data = dict()

         # pick the password

         init_data['password'] = str(self.ui.edtPassword.text())

         if len(init_data['password']) == 0:

             QtGui.QMessageBox.critical(self, 'Format error', 'Please specify a password.')

             return

         # read the content of the keyfile

         keyfile_content = ''

         try:

             if len(self.ui.edtKeyfile.text()) > 0:

                 keyfile = open(self.ui.edtKeyfile.text(), 'r')

                 keyfile_content = keyfile.read()

         except Exception as e:

             sys.stderr.write('failed to read keyfile content:\n' + str(e) + '\n')

             QtGui.QMessageBox.critical(self, 'Failed to read keyfile', str(e))

             return

         # turn into Base64

         if len(keyfile_content) > 0:

             keyfile_content_base64 = base64.b64encode(keyfile_content)

             init_data['keyfile'] = keyfile_content_base64

         res = ""

         try:

             req_data = urllib.urlencode(init_data)

             req = urllib2.Request('http://' + ip + ':58081/init', req_data)

             res = urllib2.urlopen(req)

         except:

             print('EXCEPTION ' + res)

             pass

         self.accept()

     def set_user_text(self, user_text):

         """Set a text to explain which password we need."""

         self.ui.lblText.setText(user_text)

 if __name__ == "__main__":

     print(sys.argv)

     ip = None

     try:

         ip = sys.argv[-1:][0]

     except:

         pass

     a = QtGui.QApplication(sys.argv)

     d = FormatDriveDialog(ip)

     d.show()

     sys.exit(a.exec_())