X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=daklib%2Fdbconn.py;h=39285f0872673cd28c07fd59303bdabb97fa417d;hb=25a56c7669cfdd660472b1bbc6a674113f3346ef;hp=2f5fef30f9866d2ee158b010fe89ed9cae5d9bb2;hpb=32dc9d52c53207a6524ecf63ac7866435ae40374;p=dak.git

diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 2f5fef30..39285f08 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -73,9 +73,9 @@ def session_wrapper(fn):
         private_transaction = False
 
         # Find the session object
-        try:
-            session = kwargs['session']
-        except KeyError:
+        session = kwargs.get('session')
+
+        if session is None:
             if len(args) <= len(getargspec(fn)[0]) - 1:
                 # No session specified as last argument or in kwargs
                 private_transaction = True