4 from dbuscron.bus import get_dbus_message_type, dbus_to_str
5 from dbuscron.logger import Logger
9 def __init__(self, cmd):
11 self.__auto_args = False
13 self.__file = os.environ.get('SHELL', '/bin/sh')
14 self.__args = [self.__file, '-c', self.__value]
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
23 def __call__(self, bus, message, environ):
24 args_list = map(dbus_to_str, message.get_args_list())
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)
42 log.error('environ exception', e)
46 if dbus_env['DBUS_TYPE'] in ('signal', 'method_call'):
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'])
55 dbus_env['DBUS_SENDER'],
56 dbus_env['DBUS_DEST']]
58 args_list = self.__args
60 result = os.spawnvpe(os.P_WAIT, self.__file, args_list, env)
62 log.warn('command returned non-zero status', self.__file, args_list, dbus_env, result)
66 def is_shell_cmd(self):
75 class Commands(object):
80 for m, c in self.__commands.iteritems():
83 def _get_environ(self):
86 def _set_environ(self, value):
87 self.__environ = dict()
88 self.__environ.update(os.environ)
89 self.__environ.update(value)
91 environ = property(_get_environ, _set_environ)
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)
100 def add(self, matcher, command):
101 self.__commands[matcher] = command