[dataimport] Fix method signature
authorSamuel Trégouët <samuel.tregouet@logilab.fr>
Tue, 01 Dec 2015 13:39:17 +0100
changeset 10942 b089858c49a2
parent 10941 e3ec46b5c710
child 10943 1079d68130e1
[dataimport] Fix method signature Since d4bd28d5fca8, _storage_handler expect 4 args.
dataimport/pgstore.py
--- a/dataimport/pgstore.py	Tue Dec 01 13:38:21 2015 +0100
+++ b/dataimport/pgstore.py	Tue Dec 01 13:39:17 2015 +0100
@@ -408,7 +408,7 @@
             _sql[statement] = [data]
 
     def add_entity(self, cnx, entity):
-        with self._storage_handler(entity, 'added'):
+        with self._storage_handler(cnx, entity, 'added'):
             attrs = self.preprocess_entity(entity)
             rtypes = self._inlined_rtypes_cache.get(entity.cw_etype, ())
             if isinstance(rtypes, str):