From 39915a3eb3eb6ce2d1c2c5c2d8c4905c1a00ad97 Mon Sep 17 00:00:00 2001
From: Cole Robinson <crobinso@redhat.com>
Date: Mon, 17 Mar 2008 13:10:15 -0400
Subject: [PATCH] Add a get_type methon to domain object to facilitate getting
 a correct max_vcpus value.

---
 src/virtManager/connection.py | 4 ++--
 src/virtManager/details.py    | 2 +-
 src/virtManager/domain.py     | 3 +++
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/virtManager/connection.py b/src/virtManager/connection.py
index 2f3457da0..13ecd4ba5 100644
--- a/src/virtManager/connection.py
+++ b/src/virtManager/connection.py
@@ -501,8 +501,8 @@ class vmmConnection(gobject.GObject):
     def get_host_info(self):
         return self.hostinfo
 
-    def get_max_vcpus(self):
-        return virtinst.util.get_max_vcpus(self.vmm)
+    def get_max_vcpus(self, type=None):
+        return virtinst.util.get_max_vcpus(self.vmm, type)
 
     def connect(self, name, callback):
         handle_id = gobject.GObject.connect(self, name, callback)
diff --git a/src/virtManager/details.py b/src/virtManager/details.py
index 5470cbe30..b0b2b8979 100644
--- a/src/virtManager/details.py
+++ b/src/virtManager/details.py
@@ -374,7 +374,7 @@ class vmmDetails(gobject.GObject):
         self.window.get_widget("state-host-cpus").set_text("%d" % self.vm.get_connection().host_active_processor_count())
         status = self.vm.status()
         if status in [ libvirt.VIR_DOMAIN_SHUTOFF, libvirt.VIR_DOMAIN_CRASHED ]:
-            cpu_max = self.vm.get_connection().get_max_vcpus()
+            cpu_max = self.vm.get_connection().get_max_vcpus(self.vm.get_type())
             self.window.get_widget("config-vcpus").get_adjustment().upper = cpu_max
             self.window.get_widget("state-vm-maxvcpus").set_text(str(cpu_max))
         else:
diff --git a/src/virtManager/domain.py b/src/virtManager/domain.py
index 65cef20a7..215023cc6 100644
--- a/src/virtManager/domain.py
+++ b/src/virtManager/domain.py
@@ -115,6 +115,9 @@ class vmmDomain(gobject.GObject):
             return True
         return False
 
+    def get_type(self):
+        return self.get_xml_string("/domain/@type")
+
     def is_vcpu_hotplug_capable(self):
         # Read only connections aren't allowed to change it
         if self.connection.is_read_only():