def __init__(self, application):
self._application = application
self._mounts = {}
# create user interface
self.window = MountsManagerWindow(self)
# create volume monitor
self._volume_monitor = VolumeMonitor()
self._volume_monitor.connect('mount-added', self._add_mount)
self._volume_monitor.connect('mount-removed', self._remove_mount)
self._volume_monitor.connect('volume-added', self._add_volume)
self._volume_monitor.connect('volume-removed', self._remove_volume)
def __init__(self, application):
self._application = application
self._mounts = {}
# create user interface
self.window = MountsManagerWindow(self)
# create volume monitor
self._volume_monitor = VolumeMonitor()
self._volume_monitor.connect('mount-added', self._add_mount)
self._volume_monitor.connect('mount-removed', self._remove_mount)
self._volume_monitor.connect('volume-added', self._add_volume)
self._volume_monitor.connect('volume-removed', self._remove_volume)
class MountsManager:
"""Class used for monitoring and managing mounts menu"""
def __init__(self, application):
self._application = application
self._mounts = {}
# create user interface
self.window = MountsManagerWindow(self)
# create volume monitor
self._volume_monitor = VolumeMonitor()
self._volume_monitor.connect('mount-added', self._add_mount)
self._volume_monitor.connect('mount-removed', self._remove_mount)
self._volume_monitor.connect('volume-added', self._add_volume)
self._volume_monitor.connect('volume-removed', self._remove_volume)
def _populate_list(self):
"""Populate mount/volume list"""
# get list of volumes
for volume in self._volume_monitor.get_volumes():
self.window._add_volume(volume, startup=True)
# get list of mounted volumes
for mount in self._volume_monitor.get_mounts():
if mount.get_volume() is None:
self._add_mount(self._volume_monitor, mount)
# update menus
self.window._menu_updated()
def _add_mount(self, monitor, mount):
"""Catch volume-mounted signal and update mounts menu"""
icon_names = mount.get_icon().to_string()
mount_icon = self._application.icon_manager.get_mount_icon_name(icon_names)
mount_uri = mount.get_root().get_uri()
volume = mount.get_volume()
# if mount has volume, set mounted flag
if volume is not None:
self.window._volume_mounted(volume)
# add mount to the list
self.window._notify_mount_add(mount_icon, mount.get_name(), mount_uri)
# add bookmark menu item
self.window._add_item(
mount.get_name(),
mount_uri,
mount_icon
)
# add unmount menu item
self.window._add_unmount_item(
mount.get_name(),
mount_uri,
mount_icon
)
# add mount object to local cache list
self._mounts[mount_uri] = mount
def _remove_mount(self, monitor, mount):
"""Remove volume menu item from the mounts menu"""
volume = mount.get_volume()
mount_uri = mount.get_root().get_uri()
# update volume list if possible
if volume is not None:
self.window._volume_unmounted(volume)
# remove mount from list
self.window._notify_mount_remove(mount_uri)
# remove item from menus
self.window._remove_item(mount_uri)
# remove mount object from cache list
if mount_uri in self._mounts:
self._mounts.pop(mount_uri)
def _add_volume(self, monitor, volume):
"""Event called when new volume is connected"""
self.window._add_volume(volume)
def _remove_volume(self, monitor, volume):
"""Event called when volume is removed/unmounted"""
self.window._remove_volume(volume)
def _unmount_item_menu_callback(self, widget, data=None):
"""Event called by the unmount menu item or unmount button from manager"""
uri = widget.get_data('uri')
if uri is not None:
self._unmount(self._mounts[uri])
def _unmount_by_uri(self, uri):
"""Perform unmount by URI"""
if uri in self._mounts:
self._unmount(self._mounts[uri])
def _unmount(self, mount):
"""Perform unmounting"""
if mount.can_unmount():
# notify volume manager extension if mount is part of volume
volume = mount.get_volume()
if volume is not None:
self.window._volume_unmounted(volume)
# we can safely unmount
mount.unmount(self._unmount_finish)
else:
# print error
dialog = gtk.MessageDialog(
self,
gtk.DIALOG_DESTROY_WITH_PARENT,
gtk.MESSAGE_WARNING,
gtk.BUTTONS_OK,
_('Specified item can not be unmounted.')
)
dialog.run()
dialog.destroy()
def _unmount_finish(self, mount, result):
"""Callback for unmount events"""
mount.unmount_finish(result)
def _attach_menus(self):
"""Attach mounts to menus"""
self.window._attach_menus()
def show(self, widget=None, data=None):
"""Show mounts manager window"""
self.window.show_all()
return True
def create_extensions(self):
"""Create mounts manager extensions"""
self.window.create_extensions()
def is_mounted(self, path):
"""Check if specified path is mounted"""
pass
def mount_path(self, path):
"""Mount specified path if extensions know how"""
pass
class MountsManager:
"""Class used for monitoring and managing mounts menu"""
def __init__(self, application):
self._application = application
self._mounts = {}
# create user interface
self.window = MountsManagerWindow(self)
# create volume monitor
self._volume_monitor = VolumeMonitor()
self._volume_monitor.connect('mount-added', self._add_mount)
self._volume_monitor.connect('mount-removed', self._remove_mount)
self._volume_monitor.connect('volume-added', self._add_volume)
self._volume_monitor.connect('volume-removed', self._remove_volume)
def _populate_list(self):
"""Populate mount/volume list"""
# get list of volumes
for volume in self._volume_monitor.get_volumes():
self.window._add_volume(volume, startup=True)
# get list of mounted volumes
for mount in self._volume_monitor.get_mounts():
self._add_mount(self._volume_monitor, mount)
# update menus
self.window._menu_updated()
def _add_mount(self, monitor, mount):
"""Catch volume-mounted signal and update mounts menu"""
icon_names = mount.get_icon().to_string()
mount_icon = self._application.icon_manager.get_mount_icon_name(
icon_names)
mount_uri = mount.get_root().get_uri()
volume = mount.get_volume()
# if mount has volume, set mounted flag
if volume is not None:
self.window._volume_mounted(volume)
# add mount to the list in Mount Manager window
self.window._notify_mount_add(mount_icon, mount.get_name(), mount_uri)
# add bookmark menu item
self.window._add_item(mount.get_name(), mount_uri, mount_icon)
# add unmount menu item
self.window._add_unmount_item(mount.get_name(), mount_uri, mount_icon)
# add mount object to local cache list
self._mounts[mount_uri] = mount
def _remove_mount(self, monitor, mount):
"""Remove volume menu item from the mounts menu"""
volume = mount.get_volume()
mount_uri = mount.get_root().get_uri()
# update volume list if possible
if volume is not None:
self.window._volume_unmounted(volume)
# remove mount from list
self.window._notify_mount_remove(mount_uri)
# remove item from menus
self.window._remove_item(mount_uri)
# remove mount object from cache list
if mount_uri in self._mounts:
self._mounts.pop(mount_uri)
def _add_volume(self, monitor, volume):
"""Event called when new volume is connected"""
self.window._add_volume(volume)
def _remove_volume(self, monitor, volume):
"""Event called when volume is removed/unmounted"""
self.window._remove_volume(volume)
def _unmount_item_menu_callback(self, widget, data=None):
"""Event called by the unmount menu item or unmount button from manager"""
uri = widget.get_data('uri')
if uri is not None:
self._unmount(self._mounts[uri])
def _unmount_by_uri(self, uri):
"""Perform unmount by URI"""
if uri in self._mounts:
self._unmount(self._mounts[uri])
def _unmount(self, mount):
"""Perform unmounting"""
if mount.can_unmount():
# notify volume manager extension if mount is part of volume
volume = mount.get_volume()
if volume is not None:
self.window._volume_unmounted(volume)
# we can safely unmount
mount.unmount(self._unmount_finish)
else:
# print error
dialog = gtk.MessageDialog(
self, gtk.DIALOG_DESTROY_WITH_PARENT, gtk.MESSAGE_WARNING,
gtk.BUTTONS_OK, _('Specified item can not be unmounted.'))
dialog.run()
dialog.destroy()
def _unmount_finish(self, mount, result):
"""Callback for unmount events"""
mount.unmount_finish(result)
def _attach_menus(self):
"""Attach mounts to menus"""
self.window._attach_menus()
def show(self, widget=None, data=None):
"""Show mounts manager window"""
self.window.show_all()
return True
def create_extensions(self):
"""Create mounts manager extensions"""
self.window.create_extensions()
def is_mounted(self, path):
"""Check if specified path is mounted"""
pass
def mount_path(self, path):
"""Mount specified path if extensions know how"""
pass