Hi everyone,

I know that the subject of mutable objects as dictionary keys has been 
discussed a number of times in this forum (see for instance "freezing" of 
classes), but I would love to hear the thoughts of the experts on the approach 
below. 

The use case that I encounter frequently is the classification of objects 
according to certain rules or properties: Say, I have objects A, B, C, ... 
(e.g. instances of a class, dict, etc), I can write

    d = {}
    d.setdefault(A,[]).append(A)
    d.setdefault(B,[]).append(B)
    ...

so objects that map to the same hash key will end up in the same bucket. (A 
"real world" example I had to deal with recently was for instance the 
construction of a directed acyclic graph from many directed trees where 
identical subtrees needed to be identified.)

The easiest way is of course to define custom __hash__() and __eq__() methods, 
but this breaks if objects are mutated (accidentally) after having been 
inserted into the dictionary. The "best" working approach I came up with so far 
is to generate an "immutable view" V of a mutable object O according to my 
classification rule, delegate O.__hash__ and O.__eq__ to V, and make sure that 
the V is memoized and cannot (easily) be altered later, even when O is mutated:

def hashable_mutable_factory(mutable,rule):
  class _mutable(mutable):
    def __init__(self,*args,**kw):
      self._view_cache = {}
      super(_mutable,self).__init__(*args,**kw)
    def _view(self):
      id_ = id(self)
      if not self._view_cache.has_key(id_):
        self._view_cache[id_] = rule(self)
      return self._view_cache[id_]
    def __hash__(self):
      return hash(self._view())
    def __eq__(self,other):
      return self._view() == other._view()
  return _mutable

E.g.:

>>> hashable_dict = hashable_mutable_factory(dict,lambda obj: 
>>> frozenset(obj.iteritems()))
>>> h = hashable_dict(a=1,b=2)
>>> d = {}
>>> d[h] = 'foo'
>>> d
{{'a': 1, 'b': 2}: 'foo'}
>>> h['c'] = 'bar'
>>> d
{{'a': 1, 'c': 'bar', 'b': 2}: 'foo'}
>>> g = hashable_dict(a=1,b=2)
>>> h

{'a': 1, 'c': 'bar', 'b': 2}

>>> g

{'a': 1, 'b': 2}
>>> id(g) == id(h)
False
>>> g == h
True

I slightly favor the factory function idiom above over defining the rule in a 
super class (this would have to be done for each mutable type and rule function 
separately), especially since I read that future versions of python (2.6 ?, 3.0 
?) will contain class decorators and allow syntax like class A(*bases): pass

Is there a better approach? Any comments are appreciated. 

I have been seriously using Python for one year know, mostly in the context of 
graph algorithms etc., and it has always been a delightful coding experience!

Best regards,

Andreas




-- 
http://mail.python.org/mailman/listinfo/python-list

Reply via email to