Backup restore dialog.

qubesmanager/restore.py

@@ -0,0 +1,129 @@
+#!/usr/bin/python2.6
+#
+# The Qubes OS Project, http://www.qubes-os.org
+#
+# Copyright (C) 2012  Agnieszka Kostrzewa <agnieszka.kostrzewa@gmail.com>
+# Copyright (C) 2012  Marek Marczykowski <marmarek@mimuw.edu.pl>
+#
+# 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; either version 2
+# of the License, or (at your option) any later version.
+#
+# 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.
+#
+#
+
+import sys
+import os
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+
+from qubes.qubes import QubesVmCollection
+from qubes.qubes import QubesException
+from qubes.qubes import QubesDaemonPidfile
+from qubes.qubes import QubesHost
+
+import qubesmanager.resources_rc
+
+from pyinotify import WatchManager, Notifier, ThreadedNotifier, EventsCodes, ProcessEvent
+
+import subprocess
+import time
+import threading
+from operator import itemgetter
+
+from ui_restoredlg import *
+from multiselectwidget import *
+
+
+
+class RestoreVMsWindow(Ui_Restore, QWizard):
+
+    def __init__(self, parent=None):
+        super(RestoreVMsWindow, self).__init__(parent)
+
+        self.setupUi(self)
+
+        self.selectVMsWidget = MultiSelectWidget(self)
+        self.verticalLayout.insertWidget(1, self.selectVMsWidget)
+
+        self.selectVMsWidget.available_list.addItem("netVM1")
+        self.selectVMsWidget.available_list.addItem("appVM1")
+        self.selectVMsWidget.available_list.addItem("appVM2")
+        self.selectVMsWidget.available_list.addItem("templateVM1")
+        
+       
+    def reject(self):
+        self.done(0)
+
+    def save_and_apply(self):
+        pass
+
+    @pyqtSlot(name='on_selectPathButton_clicked')
+    def selectPathButton_clicked(self):
+        self.path = self.pathLineEdit.text()
+        newPath = QFileDialog.getExistingDirectory(self, 'Select backup directory.')
+        if newPath:
+            self.pathLineEdit.setText(newPath)
+            self.path = newPath
+
+# Bases on the original code by:
+# Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
+
+def handle_exception( exc_type, exc_value, exc_traceback ):
+    import sys
+    import os.path
+    import traceback
+
+    filename, line, dummy, dummy = traceback.extract_tb( exc_traceback ).pop()
+    filename = os.path.basename( filename )
+    error    = "%s: %s" % ( exc_type.__name__, exc_value )
+
+    QMessageBox.critical(None, "Houston, we have a problem...",
+                         "Whoops. A critical error has occured. This is most likely a bug "
+                         "in Qubes Restore VMs application.<br><br>"
+                         "<b><i>%s</i></b>" % error +
+                         "at <b>line %d</b> of file <b>%s</b>.<br/><br/>"
+                         % ( line, filename ))
+
+
+
+
+def main():
+
+    global qubes_host
+    qubes_host = QubesHost()
+
+    global app
+    app = QApplication(sys.argv)
+    app.setOrganizationName("The Qubes Project")
+    app.setOrganizationDomain("http://qubes-os.org")
+    app.setApplicationName("Qubes Restore VMs")
+
+    sys.excepthook = handle_exception
+
+    qvm_collection = QubesVmCollection()
+    qvm_collection.lock_db_for_reading()
+    qvm_collection.load()
+    qvm_collection.unlock_db()
+
+    global restore_window
+    restore_window = RestoreVMsWindow()
+
+    restore_window.show()
+
+    app.exec_()
+    app.exit()
+
+
+
+if __name__ == "__main__":
+    main()

qubesmanager/settings.py

@@ -0,0 +1,149 @@
+#!/usr/bin/python2.6
+#
+# The Qubes OS Project, http://www.qubes-os.org
+#
+# Copyright (C) 2012  Agnieszka Kostrzewa <agnieszka.kostrzewa@gmail.com>
+# Copyright (C) 2012  Marek Marczykowski <marmarek@mimuw.edu.pl>
+#
+# 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; either version 2
+# of the License, or (at your option) any later version.
+#
+# 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.
+#
+#
+
+import sys
+import os
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+
+from qubes.qubes import QubesVmCollection
+from qubes.qubes import QubesException
+from qubes.qubes import qubes_appmenu_create_cmd
+from qubes.qubes import qubes_appmenu_remove_cmd
+from qubes.qubes import QubesDaemonPidfile
+from qubes.qubes import QubesHost
+from qubes.qubes import qrexec_client_path
+
+import qubesmanager.resources_rc
+
+from pyinotify import WatchManager, Notifier, ThreadedNotifier, EventsCodes, ProcessEvent
+
+import subprocess
+import time
+import threading
+from operator import itemgetter
+
+from ui_settingsdlg import *
+from multiselectwidget import *
+
+
+
+class VMSettingsWindow(Ui_SettingsDialog, QDialog):
+
+    def __init__(self, vm, parent=None):
+        super(VMSettingsWindow, self).__init__(parent)
+
+        self.setupUi(self)
+
+        self.connect(self.buttonBox, SIGNAL("accepted()"), self.save_and_apply)
+        self.connect(self.buttonBox, SIGNAL("rejected()"), self.reject)
+
+        self.app_list = MultiSelectWidget(self)
+        self.dev_list = MultiSelectWidget(self)
+        
+        self.apps_layout.addWidget(self.app_list)
+        self.devices_layout.addWidget(self.dev_list)
+
+        self.vm = vm
+        if self.vm.template_vm:
+            self.source_vm = self.vm.template_vm
+        else:
+            self.source_vm = self.vm
+
+        
+        #self.fill_apps_list()
+        #self.fill_devices_list()
+
+    def reject(self):
+        self.done(0)
+
+    def save_and_apply(self):
+        pass
+
+# Bases on the original code by:
+# Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
+
+def handle_exception( exc_type, exc_value, exc_traceback ):
+    import sys
+    import os.path
+    import traceback
+
+    filename, line, dummy, dummy = traceback.extract_tb( exc_traceback ).pop()
+    filename = os.path.basename( filename )
+    error    = "%s: %s" % ( exc_type.__name__, exc_value )
+
+    QMessageBox.critical(None, "Houston, we have a problem...",
+                         "Whoops. A critical error has occured. This is most likely a bug "
+                         "in Qubes VM Settings application.<br><br>"
+                         "<b><i>%s</i></b>" % error +
+                         "at <b>line %d</b> of file <b>%s</b>.<br/><br/>"
+                         % ( line, filename ))
+
+
+
+
+def main():
+
+    global qubes_host
+    qubes_host = QubesHost()
+
+    global app
+    app = QApplication(sys.argv)
+    app.setOrganizationName("The Qubes Project")
+    app.setOrganizationDomain("http://qubes-os.org")
+    app.setApplicationName("Qubes VM Settings")
+
+    sys.excepthook = handle_exception
+
+    qvm_collection = QubesVmCollection()
+    qvm_collection.lock_db_for_reading()
+    qvm_collection.load()
+    qvm_collection.unlock_db()
+
+    vm = None
+
+    if len(sys.argv) > 1:
+        vm = qvm_collection.get_vm_by_name(sys.argv[1])
+        if vm is None or vm.qid not in qvm_collection:
+            QMessageBox.critical(None, "Qubes VM Settings Error",
+                    "A VM with the name '{0}' does not exist in the system.".format(sys.argv[1]))
+            sys.exit(1)
+    else:
+        vms_list = [vm.name for vm in qvm_collection.values() if (vm.is_appvm() or vm.is_template())]
+        vmname = QInputDialog.getItem(None, "Select VM", "Select VM:", vms_list, editable = False)
+        if not vmname[1]:
+            sys.exit(1)
+        vm = qvm_collection.get_vm_by_name(vmname[0])
+
+    global settings_window
+    settings_window = VMSettingsWindow(vm)
+
+    settings_window.show()
+
+    app.exec_()
+    app.exit()
+
+
+
+if __name__ == "__main__":
+    main()

restoredlg.ui

@@ -0,0 +1,240 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>Restore</class>
+ <widget class="QWizard" name="Restore">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>700</width>
+    <height>399</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Qubes Restore VMs</string>
+  </property>
+  <property name="locale">
+   <locale language="English" country="UnitedStates"/>
+  </property>
+  <widget class="QWizardPage" name="wizardPage1">
+   <layout class="QGridLayout" name="gridLayout">
+    <item row="1" column="0">
+     <widget class="QLabel" name="label">
+      <property name="text">
+       <string>Device</string>
+      </property>
+     </widget>
+    </item>
+    <item row="1" column="1" colspan="2">
+     <widget class="QComboBox" name="comboBox">
+      <property name="sizePolicy">
+       <sizepolicy hsizetype="Maximum" vsizetype="Fixed">
+        <horstretch>0</horstretch>
+        <verstretch>0</verstretch>
+       </sizepolicy>
+      </property>
+      <item>
+       <property name="text">
+        <string>dev1</string>
+       </property>
+      </item>
+      <item>
+       <property name="text">
+        <string>longdeviceblablabla</string>
+       </property>
+      </item>
+      <item>
+       <property name="text">
+        <string>dev2</string>
+       </property>
+      </item>
+      <item>
+       <property name="text">
+        <string>dev3</string>
+       </property>
+      </item>
+     </widget>
+    </item>
+    <item row="2" column="0">
+     <widget class="QLabel" name="label_2">
+      <property name="text">
+       <string>Backup directory:</string>
+      </property>
+     </widget>
+    </item>
+    <item row="2" column="1">
+     <widget class="QLineEdit" name="pathLineEdit"/>
+    </item>
+    <item row="2" column="2">
+     <widget class="QToolButton" name="selectPathButton">
+      <property name="text">
+       <string>...</string>
+      </property>
+     </widget>
+    </item>
+    <item row="0" column="1">
+     <widget class="QLabel" name="label_3">
+      <property name="font">
+       <font>
+        <pointsize>12</pointsize>
+        <weight>75</weight>
+        <italic>false</italic>
+        <bold>true</bold>
+        <underline>false</underline>
+       </font>
+      </property>
+      <property name="text">
+       <string>Select backup source location:</string>
+      </property>
+     </widget>
+    </item>
+   </layout>
+  </widget>
+  <widget class="QWizardPage" name="wizardPage">
+   <layout class="QVBoxLayout" name="verticalLayout">
+    <item>
+     <widget class="QLabel" name="label_4">
+      <property name="font">
+       <font>
+        <pointsize>12</pointsize>
+        <weight>75</weight>
+        <italic>false</italic>
+        <bold>true</bold>
+        <underline>false</underline>
+       </font>
+      </property>
+      <property name="text">
+       <string>Select VMs to restore:</string>
+      </property>
+     </widget>
+    </item>
+    <item>
+     <widget class="QWidget" name="selectVMsWidget" native="true"/>
+    </item>
+    <item>
+     <widget class="QLabel" name="label_5">
+      <property name="font">
+       <font>
+        <pointsize>12</pointsize>
+        <weight>75</weight>
+        <italic>false</italic>
+        <bold>true</bold>
+        <underline>false</underline>
+       </font>
+      </property>
+      <property name="text">
+       <string>Restore options:</string>
+      </property>
+     </widget>
+    </item>
+    <item>
+     <layout class="QGridLayout" name="gridLayout_2">
+      <item row="0" column="0">
+       <widget class="QCheckBox" name="skip_broken">
+        <property name="toolTip">
+         <string>Do not restore VMs that have missing templates or netvms.</string>
+        </property>
+        <property name="text">
+         <string>skip broken</string>
+        </property>
+       </widget>
+      </item>
+      <item row="0" column="1">
+       <widget class="QCheckBox" name="skip_conflicting">
+        <property name="toolTip">
+         <string>Do not restore VMs that are already present on the host.</string>
+        </property>
+        <property name="text">
+         <string>skip conflicting</string>
+        </property>
+       </widget>
+      </item>
+      <item row="0" column="2">
+       <widget class="QCheckBox" name="ignore_uname_mismatch">
+        <property name="toolTip">
+         <string>Ignore dom0 username mismatch while restoring homedir.</string>
+        </property>
+        <property name="text">
+         <string>ignore username mismatch</string>
+        </property>
+       </widget>
+      </item>
+      <item row="1" column="0">
+       <widget class="QCheckBox" name="ignore_missing">
+        <property name="toolTip">
+         <string>Ignore missing templates or netvms, restore VMs anyway.</string>
+        </property>
+        <property name="text">
+         <string>ignore missing</string>
+        </property>
+       </widget>
+      </item>
+      <item row="1" column="1">
+       <widget class="QCheckBox" name="force_root">
+        <property name="toolTip">
+         <string>Force to run, even with root privileges.</string>
+        </property>
+        <property name="text">
+         <string>force root</string>
+        </property>
+       </widget>
+      </item>
+     </layout>
+    </item>
+   </layout>
+  </widget>
+  <widget class="QWizardPage" name="wizardPage2">
+   <layout class="QVBoxLayout" name="verticalLayout_2">
+    <item>
+     <widget class="QLabel" name="label_6">
+      <property name="font">
+       <font>
+        <pointsize>12</pointsize>
+        <weight>75</weight>
+        <italic>false</italic>
+        <bold>true</bold>
+        <underline>false</underline>
+       </font>
+      </property>
+      <property name="text">
+       <string>You're about to perform the following actions:</string>
+      </property>
+     </widget>
+    </item>
+    <item>
+     <widget class="QTextEdit" name="textEdit">
+      <property name="html">
+       <string>&lt;!DOCTYPE HTML PUBLIC &quot;-//W3C//DTD HTML 4.0//EN&quot; &quot;http://www.w3.org/TR/REC-html40/strict.dtd&quot;&gt;
+&lt;html&gt;&lt;head&gt;&lt;meta name=&quot;qrichtext&quot; content=&quot;1&quot; /&gt;&lt;style type=&quot;text/css&quot;&gt;
+p, li { white-space: pre-wrap; }
+&lt;/style&gt;&lt;/head&gt;&lt;body style=&quot; font-family:'Sans Serif'; font-size:9pt; font-weight:400; font-style:normal;&quot;&gt;
+&lt;p style=&quot; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;&quot;&gt;A lot of info&lt;br /&gt;A lot of info&lt;br /&gt;A lot of info&lt;br /&gt;A lot of info&lt;br /&gt;A lot of info&lt;br /&gt;A lot of info&lt;br /&gt;A lot of info A lot of info A lot of info A lot of info A lot of info A lot of info&lt;/p&gt;
+&lt;p style=&quot; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;&quot;&gt;A lot of info A lot of info A lot of info A lot of info A lot of info A lot of info&lt;/p&gt;
+&lt;p style=&quot; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;&quot;&gt;A lot of info A lot of info A lot of info A lot of info A lot of info A lot of info&lt;/p&gt;
+&lt;p style=&quot; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;&quot;&gt;A lot of info A lot of info A lot of info A lot of info A lot of info A lot of info&lt;/p&gt;
+&lt;p style=&quot; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;&quot;&gt;A lot of info A lot of info A lot of info A lot of info A lot of info A lot of info&lt;br /&gt;A lot of info&lt;/p&gt;&lt;/body&gt;&lt;/html&gt;</string>
+      </property>
+     </widget>
+    </item>
+    <item>
+     <widget class="QLabel" name="label_7">
+      <property name="font">
+       <font>
+        <pointsize>12</pointsize>
+        <weight>75</weight>
+        <italic>false</italic>
+        <bold>true</bold>
+        <underline>false</underline>
+       </font>
+      </property>
+      <property name="text">
+       <string>To accept press Finish. </string>
+      </property>
+     </widget>
+    </item>
+   </layout>
+  </widget>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>