dataimport.py
branchstable
changeset 8403 a6ee3cd783e1
parent 7815 2a164a9cf81c
child 8406 f3bc8ca0b715
--- a/dataimport.py	Tue May 15 10:34:34 2012 +0200
+++ b/dataimport.py	Tue May 15 10:36:02 2012 +0200
@@ -446,9 +446,12 @@
         if session is None:
             sys.exit('please provide a session of run this script with cubicweb-ctl shell and pass cnx as session')
         if not hasattr(session, 'set_cnxset'):
-            # connection
-            cnx = session
-            session = session.request()
+            if hasattr(session, 'request'):
+                # connection object
+                cnx = session
+                session = session.request()
+            else: # object is already a request
+                cnx = session.cnx
             session.set_cnxset = lambda : None
             commit = commit or cnx.commit
         else: