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