[python] Un décorateur pour inventorier les objets

[b0fh]

Hello,

Il y a quelques jours, j'ai vu passer sur IRC une question: En python, comment itérer sur tous les objets d'une classe ?

La manière recommandée est de gérer son propre pool d'objets, parce qu'on peut difficilement prédire comment le garbage collector
va se comporter, et à quel moment un objet disparaît. Néanmoins, j'ai trouvé intéressant d'écrire la petite lib suivante:

Code PYTHON :

from weakref import WeakSet

def clique(klass):
    klass.all = WeakSet()
   
    if klass.__new__ is object.__new__:
        old = klass.__init__
        def new(self,*a,**kw):
            klass.all.add(self)
            old(self,*a,**kw)
        klass.__init__ = new
       
    else:
        old = klass.__new__
        def new(*a, **kw):
            obj = old(*a, **kw)
            klass.all.add(obj)
            return obj    
        klass.__new__ = new
       
    return klass

Cette lib définit un décorateur qui ajoute un membre statique `all` à chaque classe décorée. Il marche avec les dérivés d'objets immutables (comme les strings), et avec l'héritage multiple. Il s'utilise comme ceci:

Code :

>>> from clique import *
>>>
>>> @clique
... class Foo:
...   pass
...  
>>> a = Foo()
>>> b = Foo()
>>>
>>> list(Foo.all)
[<__main__.Foo object at 0x7ffc98477da0>, <__main__.Foo object at 0x7ffc98477c50>]

Essayons d'effacer la référence à un des deux objets:

Code :

>>> a = None
>>> list(Foo.all)
[<__main__.Foo object at 0x7ffc98477da0>, <__main__.Foo object at 0x7ffc98477c50>]

Ca n'a pas marché parce que le garbage collector n'a pas tourné. On peut le forcer manuellement:

Code :

>>> import gc
>>> gc.collect()
0
>>> list(Foo.all)
[<__main__.Foo object at 0x7ffc98477da0>]

La ça marche !