2005-11-04 Robert McQueen <robot101@debian.org>

        * python/dbus_bindings.pyx, test/python/test-client.py: Fix
        marshalling of boolean values. Add some booleans to the values in
        the test client.

        * python/decorators.py, python/service.py: Add an 'async_callbacks'
        argument to the dbus.service.method decorator, which allows you to
        name arguments to take two callback functions for replying with
        return values or an exception.

        * test/python/test-client.py, test/python/test-service.py: Add test
        case using asynchronous method reply functions, both return values and
        errors, and from within both the function itself and from a mainloop
        callback.

        * python/decorators.py, python/service.py: Perform checking that the
        number of method/signal arguments matches the number of types in the
        signature at class loading time, not when you first introspect the
        class.

        * python/service.py: Remove debug print left by the last commit.

3 files changed, 60 insertions, 24 deletions

diff --git a/python/dbus_bindings.pyx b/python/dbus_bindings.pyx
index c693a3d0..a96a5adc 100644
--- a/python/dbus_bindings.pyx
+++ b/python/dbus_bindings.pyx
@@ -893,7 +893,7 @@ cdef class MessageIter:
         ptype = type(value)
         ret = ""
         if ptype == bool:
-            ret = TYPE_BOOL
+            ret = TYPE_BOOLEAN
             ret = str(chr(ret))
         elif ptype == int:
             ret = TYPE_INT32
@@ -937,7 +937,7 @@ cdef class MessageIter:
             ret = TYPE_BYTE
             ret = str(chr(ret))
         elif isinstance(value, Boolean) or value == Boolean:
-            ret = TYPE_BOOL
+            ret = TYPE_BOOLEAN
             ret = str(chr(ret))
         elif isinstance(value, Int16) or value == Int16:
             ret = TYPE_INT16
diff --git a/python/decorators.py b/python/decorators.py
index 8b553736..3973f3e4 100644
--- a/python/decorators.py
+++ b/python/decorators.py
@@ -2,16 +2,35 @@ import _util
 import inspect
 import dbus_bindings
 
-def method(dbus_interface, in_signature=None, out_signature=None):
+def method(dbus_interface, in_signature=None, out_signature=None, async_callbacks=None):
     _util._validate_interface_or_name(dbus_interface)
 
     def decorator(func):
+        args = inspect.getargspec(func)[0]
+        args.pop(0)
+
+        if async_callbacks:
+            if type(async_callbacks) != tuple:
+                raise TypeError('async_callbacks must be a tuple of (keyword for return callback, keyword for error callback)')
+            if len(async_callbacks) != 2:
+                raise ValueError('async_callbacks must be a tuple of (keyword for return callback, keyword for error callback)')
+            args.remove(async_callbacks[0])
+            args.remove(async_callbacks[1])
+
+        if in_signature:
+            in_sig = tuple(dbus_bindings.Signature(in_signature))
+
+            if len(in_sig) > len(args):
+                raise ValueError, 'input signature is longer than the number of arguments taken'
+            elif len(in_sig) < len(args):
+                raise ValueError, 'input signature is shorter than the number of arguments taken'
+
         func._dbus_is_method = True
+        func._dbus_async_callbacks = async_callbacks
         func._dbus_interface = dbus_interface
         func._dbus_in_signature = in_signature
         func._dbus_out_signature = out_signature
-        func._dbus_args = inspect.getargspec(func)[0]
-        func._dbus_args.pop(0)
+        func._dbus_args = args
         return func
 
     return decorator
@@ -29,13 +48,23 @@ def signal(dbus_interface, signature=None):
       
             self._connection.send(message)
 
+        args = inspect.getargspec(func)[0]
+        args.pop(0)
+
+        if signature:
+            sig = tuple(dbus_bindings.Signature(func._dbus_signature))
+
+            if len(sig) > len(args):
+                raise ValueError, 'signal signature is longer than the number of arguments provided'
+            elif len(sig) < len(args):
+                raise ValueError, 'signal signature is shorter than the number of arguments provided'
+
         emit_signal.__name__ = func.__name__
         emit_signal.__doc__ = func.__doc__
         emit_signal._dbus_is_signal = True
         emit_signal._dbus_interface = dbus_interface
         emit_signal._dbus_signature = signature
-        emit_signal._dbus_args = inspect.getargspec(func)[0]
-        emit_signal._dbus_args.pop(0)
+        emit_signal._dbus_args = args
         return emit_signal
 
     return decorator
diff --git a/python/service.py b/python/service.py
index 34b0384d..14a2d6d3 100644
--- a/python/service.py
+++ b/python/service.py
@@ -169,13 +169,9 @@ class InterfaceType(type):
 
         if func._dbus_in_signature:
             # convert signature into a tuple so length refers to number of
-            # types, not number of characters
+            # types, not number of characters. the length is checked by
+            # the decorator to make sure it matches the length of args.
             in_sig = tuple(dbus_bindings.Signature(func._dbus_in_signature))
-
-            if len(in_sig) > len(args):
-                raise ValueError, 'input signature is longer than the number of arguments taken'
-            elif len(in_sig) < len(args):
-                raise ValueError, 'input signature is shorter than the number of arguments taken'
         else:
             # magic iterator which returns as many v's as we need
             in_sig = dbus_bindings.VariantSignature()
@@ -204,11 +200,6 @@ class InterfaceType(type):
             # convert signature into a tuple so length refers to number of
             # types, not number of characters
             sig = tuple(dbus_bindings.Signature(func._dbus_signature))
-
-            if len(sig) > len(args):
-                raise ValueError, 'signal signature is longer than the number of arguments provided'
-            elif len(sig) < len(args):
-                raise ValueError, 'signal signature is shorter than the number of arguments provided'
         else:
             # magic iterator which returns as many v's as we need
             sig = dbus_bindings.VariantSignature()
@@ -248,16 +239,33 @@ class Object(Interface):
             interface_name = message.get_interface()
             (candidate_method, parent_method) = _method_lookup(self, method_name, interface_name)
 
-            # call method
+            # set up method call parameters
             args = message.get_args_list()
-            retval = candidate_method(self, *args)
+            keywords = {}
 
-            # send return reply if it's not an asynchronous function
-            # if we have a signature, use it to turn the return value into a tuple as appropriate
+            # iterate signature into list of complete types
             if parent_method._dbus_out_signature:
-                # iterate signature into list of complete types
                 signature = tuple(dbus_bindings.Signature(parent_method._dbus_out_signature))
+            else:
+                signature = None
 
+            # set up async callback functions
+            if parent_method._dbus_async_callbacks:
+                (return_callback, error_callback) = parent_method._dbus_async_callbacks
+                keywords[return_callback] = lambda *retval: _method_reply_return(connection, message, method_name, signature, *retval)
+                keywords[error_callback] = lambda exception: _method_reply_error(connection, message, exception)
+
+            # call method
+            retval = candidate_method(self, *args, **keywords)
+
+            # we're done - the method has got callback functions to reply with
+            if parent_method._dbus_async_callbacks:
+                return
+
+            # otherwise we send the return values in a reply. if we have a
+            # signature, use it to turn the return value into a tuple as
+            # appropriate
+            if parent_method._dbus_out_signature:
                 # if we have zero or one return values we want make a tuple
                 # for the _method_reply_return function, otherwise we need
                 # to check we're passing it a sequence
@@ -285,7 +293,6 @@ class Object(Interface):
                 else:
                     retval = (retval,)
 
-            print retval, signature
             _method_reply_return(connection, message, method_name, signature, *retval)
         except Exception, exception:
             # send error reply