Utiliser un décorateur qui renvoie une classe sur une méthode?

Question

J'ai actuellement un décorateur qui enveloppe une fonction dans une classe.

(Nous utilisons actuellement ce bizarre, cadre async sur mesure où chaque appel asynchrone est défini comme une classe avec une tonne de code de passe-partout. Mon idée était de décorer les fonctions puis retourner la classe appropriée.)

Ce décorateur fonctionne bien sur les fonctions en dehors des classes. Toutefois, lors de l'utilisation avec des méthodes, l'argument self n'est plus passé implicitement, et je ne sais pas pourquoi.

Voici le meilleur exemple que je pourrais mettre ensemble

from __future__ import print_function 

import functools 


def test_wrap(func): 
    @functools.wraps(func) 
    def wrapper(*args, **kwargs): 
     print("Args:", args) 
     print("Kwargs:", kwargs) 
     func(*args, **kwargs) 
    return wrapper 


def test_class_wrap(func): 
    """Return a Command object for use with the custom framework we are using.""" 
    @functools.wraps(func, assigned=('__name__', '__module__'), updated=()) 
    class Command(object): 
     def __init__(self, *args, **kwargs): 
      print("Args:", args) 
      print("Kwargs:", kwargs) 
      func(*args, **kwargs) 

    return Command 


class MyObject(object): 

    def __init__(self): 
     self.value = 100 

    @test_wrap 
    def foo(self): 
     print(self.value) 

    @test_class_wrap 
    def bar(self): 
     print(self.value) 


if __name__ == '__main__': 
    obj = MyObject() 
    obj.foo() 

    print() 
    obj.bar(obj) # works 
    # obj.bar() # TypeError: bar() takes exactly 1 argument (0 given) 
    # Why is self implicitly passed as an argument like with outher methods? 

# Output 
# Args: (<__main__.MyObject object at 0x7fe2bf9bb590>,) 
# Kwargs: {} 
# 100 

# Args: (<__main__.MyObject object at 0x7fe2bf9bb590>,) 
# Kwargs: {} 
# 100 

Answer 1

test_class_wrap ne fait rien, juste retourner une classe si __init__ est pas appelé. Essayez d'envelopper la classe avec une fonction en passant args et kwargs:

def test_class_wrap(func): 
    """Return a Command object for use with the custom framework we are using.""" 
    @functools.wraps(func, assigned=('__name__', '__module__'), updated=()) 
    def wrapper(*args, **kwargs): 
     class Command(object): 
      def __init__(self, *args, **kwargs): 
       print("Args:", args) 
       print("Kwargs:", kwargs) 
       func(*args, **kwargs) 
     return Command(*args, **kwargs) 
    return wrapper 

... 

if __name__ == '__main__': 
    obj = MyObject() 
    obj.foo() 

    print() 
    obj.bar()