[PATCH v3 6/9] pool-wizard: Add virtual method

[Jiri Prochazka]

This patch adds method for adding virtual machines via pool wizard:
- Method queries user for pool_dir and libvirt_domain, if valid ones are entered and
  successfuly checked, it creates XML file containing hostname and libvirt_domain
  needed for valid slave virtual machine config

Signed-off-by: Jiri Prochazka <jprochaz(a)redhat.com&gt;
---
 lnst/Controller/Wizard.py | 46 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/lnst/Controller/Wizard.py b/lnst/Controller/Wizard.py
index 5e21efa..5ec48da 100644
--- a/lnst/Controller/Wizard.py
+++ b/lnst/Controller/Wizard.py
@@ -19,7 +19,7 @@ import socket
 import sys
 import time
 import os
-from lnst.Common.Utils import mkdir_p
+from lnst.Common.Utils import mkdir_p, check_process_running
 from lnst.Common.Config import DefaultRPCPort
 from lnst.Common.ConnectionHandler import send_data, recv_data
 from xml.dom.minidom import getDOMImplementation
@@ -143,6 +143,50 @@ class Wizard:
                                  filename=filename, port=port,
                                  mode="noninteractive")
 
+    def virtual(self, pool_dir=None):
+        """ Starts Wizard in a virtual mode
+        @param pool_dir Path to pool directory (optional)
+        """
+
+        print("WARNING: For LNST Pool Wizard to work with virtual guests, "
+              "several conditions have to be met: \n"
+              "\t1) Guests must be running under libvirt and QEMU\n"
+              "\t2) Guests must be in \"default\" network and have an IP
"
+              "address from DHCP in that network")
+
+        if not LIBVIRT_SUPPORT:
+            sys.stderr.write("Missing package libvirt-python, "
+                             "aborting wizard\n")
+            return
+
+        if not check_process_running("libvirtd"):
+            sys.stderr.write("libvirtd is not running, aborting wizard\n")
+            return
+
+        pool_dir = self._check_and_query_pool_dir(pool_dir)
+
+        conn = libvirt.openReadOnly("qemu:///system")
+
+        if conn is None:
+            sys.stderr.write("Failed to open connection to hypervisor, "
+                             "aborting wizard\n")
+            return
+
+        while True:
+            libvirt_domain, hostname = self._query_libvirt_domain(conn)
+            if hostname is None or libvirt_domain is None:
+                return
+            filename = self._query_filename(libvirt_domain)
+
+            self._create_xml(hostname=hostname, pool_dir=pool_dir,
+                             filename=filename, mode="virtual",
+                             libvirt_domain=libvirt_domain)
+
+            if self._query_continuation():
+                continue
+            else:
+                break
+
     def _check_hostname(self, hostname):
         """ Checks hostnames translatibility
         @param hostname Hostname which is checked whether it's valid
-- 
2.4.3