equal
deleted
inserted
replaced
18 """Postgres specific store""" |
18 """Postgres specific store""" |
19 from __future__ import print_function |
19 from __future__ import print_function |
20 |
20 |
21 import threading |
21 import threading |
22 import warnings |
22 import warnings |
23 import cPickle |
|
24 import os.path as osp |
23 import os.path as osp |
25 from StringIO import StringIO |
24 from StringIO import StringIO |
26 from time import asctime |
25 from time import asctime |
27 from datetime import date, datetime, time |
26 from datetime import date, datetime, time |
28 from collections import defaultdict |
27 from collections import defaultdict |
29 from base64 import b64encode |
28 from base64 import b64encode |
|
29 |
|
30 from six.moves import cPickle as pickle |
30 |
31 |
31 from cubicweb.utils import make_uid |
32 from cubicweb.utils import make_uid |
32 from cubicweb.server.sqlutils import SQL_PREFIX |
33 from cubicweb.server.sqlutils import SQL_PREFIX |
33 from cubicweb.dataimport.stores import NoHookRQLObjectStore |
34 from cubicweb.dataimport.stores import NoHookRQLObjectStore |
34 |
35 |
108 'time': asctime(), 'columns': columns} |
109 'time': asctime(), 'columns': columns} |
109 filename = make_uid() |
110 filename = make_uid() |
110 try: |
111 try: |
111 with open(osp.join(dump_output_dir, |
112 with open(osp.join(dump_output_dir, |
112 '%s.pickle' % filename), 'wb') as fobj: |
113 '%s.pickle' % filename), 'wb') as fobj: |
113 cPickle.dump(pdata, fobj) |
114 pickle.dump(pdata, fobj) |
114 except IOError: |
115 except IOError: |
115 print('ERROR while pickling in', dump_output_dir, filename+'.pickle') |
116 print('ERROR while pickling in', dump_output_dir, filename+'.pickle') |
116 cnx.rollback() |
117 cnx.rollback() |
117 raise |
118 raise |
118 finally: |
119 finally: |