From 8c4e1e0b09516071b17ab5b79cdf61ee6a643068 Mon Sep 17 00:00:00 2001
From: Marek Marczykowski <marmarek@invisiblethingslab.com>
Date: Wed, 22 Feb 2012 15:14:00 +0100
Subject: [PATCH] dom0/qvm-core: fix icon symlink after label change

---
 dom0/qvm-core/qubes.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/dom0/qvm-core/qubes.py b/dom0/qvm-core/qubes.py
index 333212e8..72a58106 100755
--- a/dom0/qvm-core/qubes.py
+++ b/dom0/qvm-core/qubes.py
@@ -239,7 +239,7 @@ class QubesVm(object):
         self.firewall_conf = self.absolute_path(firewall_conf, default_firewall_conf_file)
 
         self.updateable = updateable
-        self.label = label if label is not None else QubesVmLabels["red"]
+        self._label = label if label is not None else QubesVmLabels["red"]
         if self.dir_path is not None:
             self.icon_path = self.dir_path + "/icon.png"
         else:
@@ -335,6 +335,20 @@ class QubesVm(object):
     def qid(self):
         return self.__qid
 
+    @property
+    def label(self):
+        return self._label
+
+    @label.setter
+    def label(self, new_label):
+        self._label = new_label
+        if self.icon_path:
+            try:
+                os.remove(self.icon_path)
+            except:
+                pass
+            os.symlink (new_label.icon_path, self.icon_path)
+
     @property
     def ip(self):
         if self.netvm_vm is not None: