root/hodgestar/PythonCode/SmallScripts/launch_pass.py

#!/usr/bin/env python

import dbus
import gobject
import time
import os
import logging
import subprocess
from dbus.mainloop.glib import DBusGMainLoop

class DriveVolumeListener(object):
    def __init__(self, uuid):
        self._uuid = uuid
        self._active_udis = set()

        dbus_loop = DBusGMainLoop()
        self.bus = dbus.SystemBus(mainloop=dbus_loop)

        hal_manager_obj = self.bus.get_object('org.freedesktop.Hal','/org/freedesktop/Hal/Manager')
        hal_manager = dbus.Interface(hal_manager_obj,'org.freedesktop.Hal.Manager')

        hal_manager.connect_to_signal('DeviceAdded', self.device_added_callback)
        hal_manager.connect_to_signal('DeviceRemoved', self.device_removed_callback)

        logging.info("Listening for volume with uuid %s." % (self._uuid,))

    def device_added_callback(self,udi):
        """Callback for listening to devices being added."""
        logging.debug("Saw adding of udi %s." % (udi,))
        dev_obj = self.bus.get_object('org.freedesktop.Hal', udi)
        dev_iface = dbus.Interface(dev_obj, 'org.freedesktop.Hal.Device')

        if dev_iface.QueryCapability('volume') and dev_iface.GetProperty('volume.uuid') == self._uuid:
            logging.info("Volume %s mounted." % (self._uuid,))
            self._active_udis.add(udi)
            self.volume_mounted(dev_obj)

    def device_removed_callback(self,udi):
        """Callback for listening to devices being removed."""
        logging.debug("Saw removal of udi %s." % (udi,))

        if udi in self._active_udis:
            logging.info("Volume %s unmounted." % (self._uuid,))
            self._active_udis.remove(udi)
            self.volume_unmounted(udi)

    def volume_mounted(self,dev_obj):
        """Called when the volume is mounted (sub-classes should override this)."""
        pass

    def volume_unmounted(self,udi):
        """Called when the volume is unmounted (sub-classes should override this)."""
        pass

class ActionLauncher(DriveVolumeListener):
    MOUNT_POINT_MARKER = object()

    def __init__(self, uuid, script, subfolder):
        super(ActionLauncher, self).__init__(uuid)
        self._script = script
        self._subfolder = subfolder
        self._mount_pos = script.index(self.MOUNT_POINT_MARKER)

    def volume_mounted(self,dev_obj):
        try:
            vol_iface = dbus.Interface(dev_obj, 'org.freedesktop.Hal.Device.Volume')
            dev_iface = dbus.Interface(dev_obj, 'org.freedesktop.Hal.Device')
            # vol_iface.Mount(self._mount_point, "", dbus.Array(signature="s"))

            tries_left = 10
            mount_point = dev_iface.GetProperty('volume.mount_point')
            while tries_left and not mount_point:
                time.sleep(1.0)
                tries_left -= 1
                mount_point = dev_iface.GetProperty('volume.mount_point')

            logging.info("Volume mounted on %s." % (mount_point,))

            enc_folder = os.path.join(mount_point, self._subfolder)
            while tries_left and not os.path.exists(enc_folder):
                time.sleep(1.0)
                tries_left -= 1

            logging.info("Found encryption folder at %s." % (enc_folder,))

            if tries_left:
                self._script[self._mount_pos] = enc_folder
                retcode = subprocess.call(self._script)
        except Exception, e:
            logging.error(str(e))

    def volume_unmounted(self,udi):
        # vol_iface = dbus.Interface(dev_obj, 'org.freedesktop.Hal.Device.Volume')
        # vol_iface.Unmount(dbus.Array(signature="s"))
        pass

UUID = "4298-61BB"
SCRIPT = ["/usr/bin/encfs", "--extpass=/usr/bin/ssh-askpass", "--no-default-flags",
          ActionLauncher.MOUNT_POINT_MARKER, "/home/simon/keys"]
SUBFOLDER = "keys"

logging.basicConfig(level=logging.DEBUG,
                    filemode='a')

oL = ActionLauncher(UUID, SCRIPT, SUBFOLDER)

mainloop = gobject.MainLoop()
mainloop.run()