1 files changed, 66 insertions, 0 deletions
diff --git a/src/jarabe/view/frame/devicestray.py b/src/jarabe/view/frame/devicestray.py
new file mode 100644
index 0000000..eb6707e
--- /dev/null
+++ b/src/jarabe/view/frame/devicestray.py
@@ -0,0 +1,66 @@
+# Copyright (C) 2008 One Laptop Per Child
+#
+# 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 St, Fifth Floor, Boston, MA  02110-1301  USA
+
+import logging
+
+from sugar.graphics import tray
+
+from jarabe.view.devices import deviceview
+from jarabe.model import shellmodel
+
+_logger = logging.getLogger('DevicesTray')
+
+class DevicesTray(tray.HTray):
+    def __init__(self):
+        tray.HTray.__init__(self, align=tray.ALIGN_TO_END)
+        self._device_icons = {}
+
+        devices_model = shellmodel.get_instance().get_devices()
+
+        for device in devices_model:
+            self._add_device(device)
+
+        devices_model.connect('device-appeared',
+                              self.__device_appeared_cb)
+        devices_model.connect('device-disappeared',
+                              self.__device_disappeared_cb)
+
+    def _add_device(self, device):
+        try:
+            view = deviceview.create(device)
+            index = 0
+            for item in self.get_children():
+                index = self.get_item_index(item)
+                view_pos = getattr(view, "FRAME_POSITION_RELATIVE", -1)
+                item_pos = getattr(item, "FRAME_POSITION_RELATIVE", 0)
+                if view_pos < item_pos:
+                    break
+            self.add_item(view, index=index)
+            view.show()
+            self._device_icons[device.get_id()] = view
+        except Exception, message:
+            _logger.warn("Not able to add icon for device [%r], because of "
+                         "an error (%s). Continuing." % (device, message))
+
+    def _remove_device(self, device):
+        self.remove_item(self._device_icons[device.get_id()])
+        del self._device_icons[device.get_id()]
+
+    def __device_appeared_cb(self, model, device):
+        self._add_device(device)
+
+    def __device_disappeared_cb(self, model, device):
+        self._remove_device(device)