dataimport/pgstore.py
changeset 10602 4845012cfc8e
parent 10598 b3f9f929385f
child 10609 e2d8e81bfe68
--- a/dataimport/pgstore.py	Wed Sep 16 14:45:15 2015 +0200
+++ b/dataimport/pgstore.py	Mon Sep 14 12:19:48 2015 +0200
@@ -20,7 +20,6 @@
 
 import threading
 import warnings
-import cPickle
 import os.path as osp
 from StringIO import StringIO
 from time import asctime
@@ -28,6 +27,8 @@
 from collections import defaultdict
 from base64 import b64encode
 
+from six.moves import cPickle as pickle
+
 from cubicweb.utils import make_uid
 from cubicweb.server.sqlutils import SQL_PREFIX
 from cubicweb.dataimport.stores import NoHookRQLObjectStore
@@ -110,7 +111,7 @@
                     try:
                         with open(osp.join(dump_output_dir,
                                            '%s.pickle' % filename), 'wb') as fobj:
-                            cPickle.dump(pdata, fobj)
+                            pickle.dump(pdata, fobj)
                     except IOError:
                         print('ERROR while pickling in', dump_output_dir, filename+'.pickle')
                 cnx.rollback()