Applying a decorator to every method in a class?

by

In Python 2.6, a class decorator is definitely the way to go. e.g., here’s a pretty general one for these kind of tasks:

import inspect

def decallmethods(decorator, prefix='test_'):
    def dectheclass(cls):
        for name, m in inspect.getmembers(cls, inspect.isfunction):
            if name.startswith(prefix):
                setattr(cls, name, decorator(m))
        return cls
    return dectheclass


@decallmethods(login_testuser)
class TestCase(object):
    def setUp(self):
        pass

    def test_1(self):
        print("test_1()")

    def test_2(self):
        print("test_2()")

will get you what you desire. In Python 2.5 or worse, the @decallmethods syntax doesn’t work for class decoration, but with otherwise exactly the same code you can replace it with the following statement right after the end of the class TestCase statement:

TestCase = decallmethods(login_testuser)(TestCase)

More Related Contents:

  1. How to make function decorators and chain them together?
  2. Creating a singleton in Python
  3. Decorators with parameters?
  4. How to get method parameter names?
  5. Preserving signatures of decorated functions
  6. Calling class staticmethod within the class body?
  7. How to decorate a class?
  8. How can I separate the functions of a class into multiple files?
  9. Attaching a decorator to all functions within a class
  10. Decorator execution order
  11. How to create a decorator that can be used either with or without parameters?
  12. How to inject variable into scope with a decorator?
  13. Introspection to get decorator names on a method?
  14. What are some common uses for Python decorators? [closed]
  15. How to decorate a method inside a class?
  16. Python Class Based Decorator with parameters that can decorate a method or a function
  17. What is the best way to do automatic attribute assignment in Python, and is it a good idea?
  18. How do I make function decorators and chain them together?
  19. Python decorator? – can someone please explain this? [duplicate]
  20. How to do a conditional decorator in python?
  21. Python decorator as a staticmethod
  22. How to create a custom decorator in Django?
  23. Currying decorator in python
  24. Generating dictionary keys on the fly [duplicate]
  25. Python functools.wraps equivalent for classes
  26. Python decorator best practice, using a class vs a function
  27. How to auto register a class when it’s defined
  28. Auto-register class methods using decorator
  29. Decorator error: NoneType object is not callable
  30. How to skip a pytest using an external fixture?

Leave a Comment