From 2f60fb1b64741fd0c8f6507315f2e1df2a935f0a Mon Sep 17 00:00:00 2001
From: Tomeu Vizoso <tomeu@tomeuvizoso.net>
Date: Mon, 13 Oct 2008 15:31:37 +0000
Subject: Move volumesmanager to jarabe.model and add a frame icon for a removable storage device.

---
(limited to 'extensions/deviceicon/volume.py')

diff --git a/extensions/deviceicon/volume.py b/extensions/deviceicon/volume.py
new file mode 100644
index 0000000..a7afe99
--- /dev/null
+++ b/extensions/deviceicon/volume.py
@@ -0,0 +1,53 @@
+# 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
+
+from sugar.graphics.tray import TrayIcon
+
+from jarabe.model import volume
+
+_icons = {}
+
+class DeviceView(TrayIcon):
+    FRAME_POSITION_RELATIVE = 800
+    def __init__(self, volume):
+        TrayIcon.__init__(self, icon_name=volume.icon_name,
+                          xo_color=volume.icon_color)
+
+def setup(tray):
+    volumes_manager = volume.get_volumes_manager()
+
+    for vol in volumes_manager.get_volumes():
+        _add_device(vol, tray)
+
+    volumes_manager.connect('volume-added', _volume_added_cb, tray)
+    volumes_manager.connect('volume-removed', _volume_removed_cb, tray)
+
+def _volume_added_cb(volumes_manager, vol, tray):
+    _add_device(vol, tray)
+
+def _volume_removed_cb(volumes_manager, vol, tray):
+    _remove_device(vol, tray)
+
+def _add_device(volume, tray):
+    icon = DeviceView(volume)
+    _icons[volume] = icon
+    tray.add_device(icon)
+
+def _remove_device(volume, tray):
+    icon = _icons[volume]
+    tray.remove_device(icon)
+    del _icons[volume]
+
--
cgit v0.9.1