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

   1 # encoding: utf-8
   2
   3 import os
   4 from dbuscron.bus import get_dbus_message_type, dbus_to_str
   5 from dbuscron.logger import Logger
   6 log = Logger(__name__)
   7
   8 class Command(object):
   9     def __init__(self, cmd):
  10         self.__value = cmd
  11         self.__auto_args = False
  12         if self.is_shell_cmd:
  13             self.__file = os.environ.get('SHELL', '/bin/sh')
  14             self.__args = [self.__file, '-c', self.__value]
  15         else:
  16             self.__args = cmd.split(' ')
  17             self.__file = self.__args[0]
  18             if len(self.__args) == 1 \
  19                 and self.__file.startswith('!'):
  20                 self.__file = self.__file.lstrip('!')
  21                 self.__auto_args = True
  22
  23     def __call__(self, bus, message, environ):
  24         args_list = map(dbus_to_str, message.get_args_list())
  25         env = dict()
  26         env.update(environ)
  27         try:
  28             dbus_env = dict(
  29                     (('DBUS_ARG%d' % i, a) for i, a in enumerate(args_list)),
  30                     DBUS_ARGN   = str(len(args_list)),
  31                     DBUS_SENDER = str(message.get_sender() or ''),
  32                     DBUS_DEST   = str(message.get_destination() or ''),
  33                     DBUS_IFACE  = str(message.get_interface() or ''),
  34                     DBUS_PATH   = str(message.get_path() or ''),
  35                     DBUS_MEMBER = str(message.get_member() or ''),
  36                     DBUS_BUS    = bus.__class__.__name__.lower()[0:-3],
  37                     DBUS_ERROR  = str(message.get_error_name() or ''),
  38                     DBUS_TYPE   = get_dbus_message_type(message)
  39                     )
  40             env.update(dbus_env)
  41         except Exception, e:
  42             log.error('environ exception', e)
  43             raise e
  44
  45         if self.__auto_args:
  46             if dbus_env['DBUS_TYPE'] in ('signal', 'method_call'):
  47                 args_list[0:0] = [
  48                     dbus_env['DBUS_IFACE'],
  49                     dbus_env['DBUS_MEMBER']]
  50             elif dbus_env['DBUS_TYPE'] == 'error':
  51                args_list.insert(0, dbus_env['DBUS_ERROR'])
  52
  53             args_list[0:0] = [
  54                     self.__file,
  55                     dbus_env['DBUS_SENDER'],
  56                     dbus_env['DBUS_DEST']]
  57         else:
  58             args_list = self.__args
  59
  60         result = os.spawnvpe(os.P_WAIT, self.__file, args_list, env)
  61         if result != 0:
  62             log.warn('command returned non-zero status', self.__file, args_list, dbus_env, result)
  63         return result
  64
  65     @property
  66     def is_shell_cmd(self):
  67         for c in '|><$&;{}':
  68             if c in self.__value:
  69                 return True
  70         return False
  71
  72     def __str__(self):
  73         return self.__value
  74
  75 class Commands(object):
  76     __commands = {}
  77     __environ = {}
  78
  79     def __iter__(self):
  80         for m, c in self.__commands.iteritems():
  81             yield m, c
  82
  83     def _get_environ(self):
  84         return self.__environ
  85
  86     def _set_environ(self, value):
  87         self.__environ = dict()
  88         self.__environ.update(os.environ)
  89         self.__environ.update(value)
  90
  91     environ = property(_get_environ, _set_environ)
  92
  93     def handler(self, bus, message):
  94         for rule, command in self.__commands.iteritems():
  95             if rule.match(bus, message):
  96                 log('rule matched', rule, command)
  97                 command(bus, message, self.__environ)
  98                 return
  99
 100     def add(self, matcher, command):
 101         self.__commands[matcher] = command
 102
 103     def clear(self):
 104         self.__commands = {}
 105