cubicweb/dataimport/massive_store.py
changeset 11308 df75fe529ba8
parent 11305 118d83e65ca8
child 11310 e0b7277e5394
equal deleted inserted replaced
11307:74f5814ecdf0 11308:df75fe529ba8
    21 from datetime import datetime
    21 from datetime import datetime
    22 from collections import defaultdict
    22 from collections import defaultdict
    23 from io import StringIO
    23 from io import StringIO
    24 
    24 
    25 from six.moves import range
    25 from six.moves import range
       
    26 
       
    27 import pytz
    26 
    28 
    27 from yams.constraints import SizeConstraint
    29 from yams.constraints import SizeConstraint
    28 
    30 
    29 from psycopg2 import ProgrammingError
    31 from psycopg2 import ProgrammingError
    30 
    32 
   117         self.default_values = get_default_values(cnx.vreg.schema)
   119         self.default_values = get_default_values(cnx.vreg.schema)
   118         pg_schema = cnx.repo.config.system_source_config.get('db-namespace') or 'public'
   120         pg_schema = cnx.repo.config.system_source_config.get('db-namespace') or 'public'
   119         self._dbh = PGHelper(self._cnx, pg_schema)
   121         self._dbh = PGHelper(self._cnx, pg_schema)
   120         self._data_entities = defaultdict(list)
   122         self._data_entities = defaultdict(list)
   121         self._data_relations = defaultdict(list)
   123         self._data_relations = defaultdict(list)
   122         self._now = datetime.utcnow()
   124         self._now = datetime.now(pytz.utc)
   123         self._default_cwuri = make_uid('_auto_generated')
   125         self._default_cwuri = make_uid('_auto_generated')
   124         self._count_cwuri = 0
   126         self._count_cwuri = 0
   125         self.on_commit_callback = on_commit_callback
   127         self.on_commit_callback = on_commit_callback
   126         self.on_rollback_callback = on_rollback_callback
   128         self.on_rollback_callback = on_rollback_callback
   127         # Do our meta tables already exist?
   129         # Do our meta tables already exist?