How to decorate all functions of a class without typing it over and over for each method? [duplicate]

by

Decorate the class with a function that walks through the class’s attributes and decorates callables. This may be the wrong thing to do if you have class variables that may happen to be callable, and will also decorate nested classes (credits to Sven Marnach for pointing this out) but generally it’s a rather clean and simple solution. Example implementation (note that this will not exclude special methods (__init__ etc.), which may or may not be desired):

def for_all_methods(decorator):
    def decorate(cls):
        for attr in cls.__dict__: # there's propably a better way to do this
            if callable(getattr(cls, attr)):
                setattr(cls, attr, decorator(getattr(cls, attr)))
        return cls
    return decorate

Use like this:

@for_all_methods(mydecorator)
class C(object):
    def m1(self): pass
    def m2(self, x): pass
    ...

In Python 3.0 and 3.1, callable does not exist. It existed since forever in Python 2.x and is back in Python 3.2 as wrapper for isinstance(x, collections.Callable), so you can use that (or define your own callable replacement using this) in those versions.

More Related Contents:

  1. How to make function decorators and chain them together?
  2. Creating a singleton in Python
  3. Decorators with parameters?
  4. What does functools.wraps do?
  5. Preserving signatures of decorated functions
  6. Python memoising/deferred lookup property decorator
  7. How to decorate a class?
  8. How can I separate the functions of a class into multiple files?
  9. How to strip decorators from a function in Python
  10. Can I patch a Python decorator before it wraps a function?
  11. Why does @foo.setter in Python not work for me?
  12. How to wrap every method of a class? [duplicate]
  13. Can a decorator of an instance method access the class?
  14. How to create a decorator that can be used either with or without parameters?
  15. Python decorator handling docstrings
  16. How to inject variable into scope with a decorator?
  17. How to decorate a method inside a class?
  18. Python Class Based Decorator with parameters that can decorate a method or a function
  19. Assigning (instead of defining) a __getitem__ magic method breaks indexing [duplicate]
  20. How do I make function decorators and chain them together?
  21. How to make function decorators and chain them together in Python?
  22. How to create abstract properties in python abstract classes
  23. Can I combine two decorators into a single one in Python?
  24. How to get all methods of a Python class with given decorator?
  25. How to do a conditional decorator in python?
  26. How can I pass a variable in a decorator to function’s argument in a decorated function?
  27. Using python decorator with or without parentheses
  28. How to create a custom decorator in Django?
  29. Currying decorator in python
  30. Python decorator best practice, using a class vs a function

Leave a Comment