Discussion :

[sopsag]

Bonjour,

j'aimerais savoir s'il est possible d'utiliser une méthode pour décorer une autre méthode.
Voici un bout de code (d'exemple, mon vrai code est + compliqué) pour éclaircir mon truc :

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
19
20
21
22
class Spam :
 
    def __init__(self,verbose):
        self.verbose = verbose
        self.name = 'Spam'    
 
    def deco ( self,func ):
        if self.verbose: # NB j'accède au self de l'instance
            def sub ( *k,**a ):
                print 'entering',self.name
                return func( *k,**a )  # execute la methode
        else:
            def sub ( *k,**a ):
                return func( *k,**a )  # execute la methode
        return sub
 
    @deco
    def eggs (self,x):
        return x+1
 
s = Spam(True)
x = s.eggs(5)

Voici le code que j'imagine pour mon cas, mais ça ne compile pas : "deco() takes exactly 2 arguments (1 given)"

J'ai essayé toutes sortes de solutions de contournement, mais rien ne passe...
(déclarer deco en @staticmethod, décorer eggs avec @self.deco, etc.)

Bien sûr, je pourrais déclarer deco comme une fonction en dehors de Spam, mais ce serait moche car le code de deco est très lié à la classe Spam (ici, j'ai mis un code rikiki pour l'exemple)...

Attention : ce bout de code est là pour l'exemple... le but n'est pas de trouver comment faire marcher cet exemple, mais de trouver comment accéder au self de l'instance dans le décorateur.

Si quelqu'un a une idée...
merci d'avance !

[airod]

google dit:
http://www.gawel.org/howtos/python-decorators

[dahtah]

Salut,
Je passerai par une nested class. Par contre tu n'auras accès qu'aux méthodes statiques de ta classe externe, ce qui semble être limitant pour toi :

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
19
20
21
22
23
24
25
26
 
class test(object):
 
  a_string = "a_string"
 
  class decorator(object):
 
    def __init__(self, func):
      self.func = func
      print test.a_string
 
    def __call__(self, *args, **kwargs):
      print "Entering function %s" % self.func.__name__
      self.func(test.__class__, *args, **kwargs)
      print "Exiting function %s" % self.func.__name__
 
  def __init__(self):
    print "new object"
 
  @decorator
  def do_some(self, *args, **kwargs):
    print "doing something %s %s" % (args, kwargs)
 
if __name__ == "__main__":
  t = test()
  t.do_some("fun", "run")

Je sais pas si ça va te convenir.

[sopsag]

Merci airod, mais dans ces exemples, il n'y a des méthodes décorées, mais pas de méthodes "décoratrices"...

Merci dahtah, mais comme tu l'as deviné, ça ne me suffit pas, car je veux accéder au self (= membres d'instance et pas seulement de classe).

D'ailleurs, une classe imbriquée peut elle voir le self de sa classe imbriquante ?

[dahtah]

D'ailleurs, une classe imbriquée peut elle voir le self de sa classe imbriquante ?

Non, à moins que l'on passe l'instance de classe imbriquante au constructeur de la classe imbriquée. Et encore cela pose des problèmes dans les appels :

Code :

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

self.classe_imbriquante.methode()

car en fait on passe l'instance de la classe imbriquée à la méthode appelée (methode())et non l'instance de la classe imbriquante.

Je crois que là le plus simple sera de mettre ta méthode décoratrice à l'extérieur de ta classe.
Bon courage

[dividee]

Le premier problème est que le décorateur est exécuté lors de la création de la classe, et pas de l'instance, donc il n'est pas possible d'accéder à self dans le décorateur.
Ma première tentative a donc été:

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
19
class Spam :
    def __init__(self,verbose):
        self.verbose = verbose
        self.name = 'Spam'    
 
    @staticmethod
    def deco(func):
        def sub (self, *k, **a):
            if self.verbose: # NB j'accède au self de l'instance
                print 'entering', self.name
            return func(self, *k, **a)  # execute la methode
        return sub
 
    @Spam.deco
    def eggs(self,x):
        return x+1
 
s = Spam(True)
x = s.eggs(5)

Mais cela ne fonctionne pas:

Code :

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

TypeError: unbound method deco() must be called with Spam instance as first argument (got function instance instead)

Bizarre que Spam.deco soit vue comme une unbound method alors que c'est censé être une méthode statique... C'est sans doute lié à la façon donc la classe est construite.

Par contre ceci, qui semble un peu étrange, fonctionne:

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
class Spam :
    def __init__(self,verbose):
        self.verbose = verbose
        self.name = 'Spam'    
 
    def deco(func):
        def sub (self, *k, **a):
            if self.verbose: # NB j'accède au self de l'instance
                print 'entering', self.name
            return func(self, *k, **a)  # execute la methode
        return sub
 
    @deco
    def eggs(self,x):
        return x+1
 
s = Spam(True)
x = s.eggs(5)

Dans ce cas, je pense que '@deco' appelle en réalité la fonction deco, qui sera promue en méthode d'instance au sein de la classe.

Mais en réalité, ce n'a plus aucun intérêt de faire du décorateur une méthode, autant en faire une gentille fonction en dehors de la classe.

[EDIT]
Ceci fonctionne aussi et est peut-être un peu plus propre, mais toujours aussi inutile:

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
19
20
class Spam :
    def __init__(self,verbose):
        self.verbose = verbose
        self.name = 'Spam'    
 
    def deco(func):
        def sub (self, *k, **a):
            if self.verbose: # NB j'accède au self de l'instance
                print 'entering', self.name
            return func(self, *k, **a)  # execute la methode
        return sub
 
    @deco
    def eggs(self,x):
        return x+1
 
    deco = staticmethod(deco)
 
s = Spam(True)
x = s.eggs(5)