vcs.maemo.org Git - dbuscron/blob - dbuscron/command.py

   1
   2 import os
   3 from dbuscron.bus import get_dbus_message_type
   4
   5 class Command(object):
   6     def __init__(self, cmd):
   7         self.__value = cmd
   8         if self.is_shell_cmd:
   9             self.__file = os.environ.get('SHELL', '/bin/sh')
  10             self.__args = [self.__file, '-c', self.__value]
  11         else:
  12             self.__args = cmd.split(' ')
  13             self.__file = self.__args[0]
  14
  15     def __call__(self, bus, message, environ):
  16         args_list = message.get_args_list()
  17         env = dict()
  18         env.update(environ)
  19         env.update(dict(
  20                 (('DBUS_ARG%d' % i, str(args_list[i])) for i in range(0, len(args_list))),
  21                 DBUS_ARGN   = str(len(args_list)),
  22                 DBUS_SENDER = str(message.get_sender()),
  23                 DBUS_DEST   = str(message.get_destination()),
  24                 DBUS_IFACE  = str(message.get_interface()),
  25                 DBUS_PATH   = str(message.get_path()),
  26                 DBUS_MEMBER = str(message.get_member()),
  27                 DBUS_BUS    = bus.__class__.__name__.lower()[0:-3],
  28                 DBUS_TYPE   = get_dbus_message_type(message)
  29                 ))
  30         result = os.spawnvpe(os.P_WAIT, self.__file, self.__args, env)
  31         return result
  32
  33     @property
  34     def is_shell_cmd(self):
  35         for c in '|><$&;{}':
  36             if c in self.__value:
  37                 return True
  38         return False
  39
  40     def __str__(self):
  41         return self.__value
  42
  43 class Commands(object):
  44     __commands = {}
  45     __environ = {}
  46
  47     def _get_environ(self):
  48         return self.__environ
  49
  50     def _set_environ(self, value):
  51         self.__environ = dict()
  52         self.__environ.update(os.environ)
  53         self.__environ.update(value)
  54
  55     environ = property(_get_environ, _set_environ)
  56
  57     def handler(self, bus, message):
  58         for rule, command in self.__commands.iteritems():
  59             if rule.match(bus, message):
  60                 command(bus, message, self.__environ)
  61                 return
  62
  63     def add(self, matcher, command):
  64         self.__commands[matcher] = command
  65