Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use defaultdict instead of dict to simplify code and improve performance

Browse Source 
svn: r14060
This commit is contained in:
Gerald Britton 2010-01-13 15:46:12 +00:00
parent c7293346bd
commit a4b22193c2
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/gen/db/txn.py
View File

@ -34,6 +34,7 @@ from __future__ import with_statement
import cPickle as pickle import cPickle as pickle
from bsddb import dbshelve, db from bsddb import dbshelve, db
import logging import logging
from collections import defaultdict
#------------------------------------------------------------------------- #-------------------------------------------------------------------------
# #
@ -51,7 +52,7 @@ _LOG = logging.getLogger(DBLOGNAME)
# Gramps transaction class # Gramps transaction class
# #
#------------------------------------------------------------------------- #-------------------------------------------------------------------------
class DbTxn(dict): class DbTxn(defaultdict):
""" """
Define a group of database commits that define a single logical operation. Define a group of database commits that define a single logical operation.
This class should not be used directly, but subclassed to reference a real This class should not be used directly, but subclassed to reference a real
@ -108,7 +109,7 @@ class DbTxn(dict):
data = pickled representation of the object data = pickled representation of the object
""" """
super(DbTxn, self).__init__({}) defaultdict.__init__(self, list, {})
self.msg = msg self.msg = msg
self.commitdb = commitdb self.commitdb = commitdb
@ -158,10 +159,8 @@ class DbTxn(dict):
self.last = len(self.commitdb) -1 self.last = len(self.commitdb) -1
if self.first is None: if self.first is None:
self.first = self.last self.first = self.last
if (obj_type, trans_type) in self: self[(obj_type, trans_type)] += [(handle, new_data)]
self[(obj_type, trans_type)] += [(handle, new_data)] return
else:
self[(obj_type, trans_type)] = [(handle, new_data)]
def get_recnos(self, reverse=False): def get_recnos(self, reverse=False):
""" """