X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=daklib%2Fdbconn.py;h=f5a19c1236442360ca57b7b867eed9f24f5f5003;hb=ce324ffde27b192a8e713b6e98368a3b00edaf27;hp=3aa7b8e2458965f152ede8dd57b361bc72bd338a;hpb=219b62c7ed60918097f38eeacfb734bc78588696;p=dak.git

diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 3aa7b8e2..f5a19c12 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -64,7 +64,7 @@ def session_wrapper(fn):
         session = kwargs.get('session')
 
         # No session specified as last argument or in kwargs, create one.
-        if session is None or len(args) == len(getargspec(fn)[0]) - 1:
+        if session is None and len(args) <= len(getargspec(fn)[0]) - 1:
             private_transaction = True
             kwargs['session'] = DBConn().session()
 
@@ -75,6 +75,9 @@ def session_wrapper(fn):
                 # We created a session; close it.
                 kwargs['session'].close()
 
+    wrapped.__doc__ = fn.__doc__
+    wrapped.func_name = fn.func_name
+
     return wrapped
 
 ################################################################################