Discussion :

[shaiHulud]

Bonjour,

Comment faire pour surcharger une méthode dont on ne connait que le nom ?

Cette question m'est apparue sur l'exemple suivant:

Je cherche à décorer une classe de sorte qu'un point d'arret soit rencontré lorsqu'une certaine méthode est appelée. Mon problème vient de ce que je ne connais que le nom de la methode (method_name dans le code).

L'exception
TypeError: 'dictproxy' object does not support item assignment
est levée par la ligne

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

_watch.__dict__[method_name]= _watch._watched_method

c'est à dire au moment ou j'essaye d'assigner via son nom la méthode avec point d'arret à la nouvelle classe.

Est-ce le fait que la classe _watch soit intérieure à une fonction qui fait que sont __dict__ est un dictproxy ? Comment puis-je faire ?

Merci d'avance

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
def any_args(self,*a,**k): return True
 
def watch_call(method_name,arg_test= any_args):
    def make_watcher(obj):
        class _watch(obj):
            def __init__(self,*a,**k): obj.__init__(self,*a,**k)
            def _watched_method(self,*a,**k):
                if arg_test(self,*a,**k):
                    WATCH_DEBUG= None # POINT D'ARRET ICI
                return obj.__dict__["method_name"](self,*a ,**k)
        _watch.__dict__[method_name]= _watch._watched_method
        return _watch
    return make_watcher
 
if __name__== "__main__":
    l= watch_call("append")(list)()
    l.append(1) # cet appel doit passer par le point d'arret

(arg_test permet juste de ne passer sur le point d'arret que lorsque les arguments de la méthodes vérifient une condition donnée)

[mont29]

Salut,

En utilisant setattr, ça marche (au passage, ’y avait aussi une erreur ligne 11… Du coup, j’en ai profité pour là aussi utiliser getattr, c’est mieux comme ça, amha*! )*:

Code python :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
def any_args(self,*a,**k): return True
 
def watch_call(method_name,arg_test= any_args):
    def make_watcher(obj):
        class _watch(obj):
            def __init__(self,*a,**k): obj.__init__(self,*a,**k)
            def _watched_method(self,*a,**k):
                if arg_test(self,*a,**k):
                    print("point d’arrêt*!")
                    WATCH_DEBUG = None # POINT D'ARRET ICI
                return getattr(obj, method_name)(self,*a ,**k)
        setattr(_watch, method_name, _watch._watched_method)
        return _watch
    return make_watcher
 
if __name__== "__main__":
    l = watch_call("append")(list)()
    l.append(1) # cet appel doit passer par le point d'arret

[PauseKawa]

Bonsoir,

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
>>> class Foo(object):
...     pass
... 
>>> Foo.__dict__.items()
[('__dict__', <attribute '__dict__' of 'Foo' objects>), ('__module__', '__main__'), ('__weakref__', <attribute '__weakref__' of 'Foo' objects>), ('__doc__', None)]
>>> Foo.__dict__['foo'] = 1
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'dict_proxy' object does not support item assignment
>>> f = Foo()
>>> f.__dict__['foo'] = 1
>>> f.__dict__.items()
dict_items([('foo', 1)])

@+

[shaiHulud]

Merci à tous les 2.

Je précise tout de même 2 choses qui diffèrent chez moi (je suis sous py 2.6, tandis que vous avez l'air d'etre en 3.x)

1/ return obj.__dict__["method_name"](self,*a ,**k) marche sans avoir besoin de getattr

2/ Dans l'exemple de PauseKawa, Foo.__dict__['foo'] = 1 ne lève pas d'erreur, lorsque la classe n'est pas définie dans une fonction

Dernière petite question, je trouve l'instruction un peu lourde

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
 
watch_call("__setitem__",lambda s,*a,**k: a[0]== "a" )(dict)()

Pas possible de faire plus court ? avec un @ bien placé (et en une seule ligne) ?

[mont29]

1/ return obj.__dict__["method_name"](self,*a ,**k) marche sans avoir besoin de getattr

Ça, ça m’étonnerait vraiment, mais alors vraiment beaucoup… return obj.__dict__[method_name](self,*a ,**k), par contre, doit effectivement marcher (y compris en py3k)*!

Dernière petite question, je trouve l'instruction un peu lourde

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
 
watch_call("__setitem__",lambda s,*a,**k: a[0]== "a" )(dict)()

Pas possible de faire plus court ? avec un @ bien placé (et en une seule ligne) ?

Ben vi, c’est un peu lourd… Mais je ne vois pas trop comment un decorator pourrait t’aider, là. Par contre, tu peux nettement alléger ton code en supprimant un niveau intermédiaire, apparemment totalement inutile*?

Code python :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
def any_args(self,*a,**k): return True
 
def watch_call(obj, method_name, arg_test= any_args):
    class _watch(obj):
        def __init__(self,*a,**k): obj.__init__(self,*a,**k)
        def _watched_method(self,*a,**k):
            if arg_test(self,*a,**k):
                print("point d’arrêt (a[0] = %s)*!" % a[0])
                WATCH_DEBUG = None # POINT D'ARRET ICI
            return getattr(obj,method_name)(self,*a ,**k)
    setattr(_watch, method_name, _watch._watched_method)
    return _watch
 
if __name__== "__main__":
    l = watch_call(list, "append", lambda s,*a,**k: a[0]== "a")()
    l.append("a") # cet appel doit passer par le point d'arret
    l.append("b") # cet appel ne doit pas passer par le point d'arret

[dividee]

Bonsoir,

quelques simplifications supplémentaires (et j'ai renommé obj en cls; c'est plus clair).

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
def watch_call(cls, method_name, arg_test=lambda *a,**k:True):
    def _watched_method(self,*a,**k):
        if arg_test(self,*a,**k):
            print("point d’arrêt (a[0] = %s)!" % a[0])
            WATCH_DEBUG = None # POINT D'ARRET ICI
        return getattr(cls,method_name)(self, *a ,**k)
    return type(cls.__name__, (cls,), {method_name: _watched_method})
 
if __name__== "__main__":
    l = watch_call(list, "append", lambda self,x: x == "a")()
    l.append("a") # cet appel doit passer par le point d'arret
    l.append("b") # cet appel ne doit pas passer par le point d'arret

[mont29]

Bonne idée l’utilisation de type(), dividee, j’y aurais pas pensé…