Introspection to get decorator names on a method?

by

I’m surprised that this question is so old and no one has taken the time to add the actual introspective way to do this, so here it is:

The code you want to inspect…

def template(func):
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper

baz = template
che = template

class Foo(object):
    @baz
    @che
    def bar(self):
        pass

Now you can inspect the above Foo class with something like this…

import ast
import inspect

def get_decorators(cls):
    target = cls
    decorators = {}

    def visit_FunctionDef(node):
        decorators[node.name] = []
        for n in node.decorator_list:
            name=""
            if isinstance(n, ast.Call):
                name = n.func.attr if isinstance(n.func, ast.Attribute) else n.func.id
            else:
                name = n.attr if isinstance(n, ast.Attribute) else n.id

            decorators[node.name].append(name)

    node_iter = ast.NodeVisitor()
    node_iter.visit_FunctionDef = visit_FunctionDef
    node_iter.visit(ast.parse(inspect.getsource(target)))
    return decorators

print get_decorators(Foo)

That should print something like this…

{'bar': ['baz', 'che']}

or at least it did when I tested this with Python 2.7.9 real quick 🙂

More Related Contents:

  1. How to get method parameter names?
  2. How to make function decorators and chain them together?
  3. Creating a singleton in Python
  4. Getting the class name of an instance?
  5. How to get the caller’s method name in the called method?
  6. Determine function name from within that function (without using traceback)
  7. Preserving signatures of decorated functions
  8. How can you print a variable name in python? [duplicate]
  9. Calling class staticmethod within the class body?
  10. Finding what methods a Python object has
  11. How to decorate a class?
  12. List all the modules that are part of a python package?
  13. How can I separate the functions of a class into multiple files?
  14. Attaching a decorator to all functions within a class
  15. Get __name__ of calling function’s module in Python
  16. Decorator execution order
  17. How to create a decorator that can be used either with or without parameters?
  18. How to inject variable into scope with a decorator?
  19. How to decorate a method inside a class?
  20. Python Class Based Decorator with parameters that can decorate a method or a function
  21. What is the best way to do automatic attribute assignment in Python, and is it a good idea?
  22. How do I make function decorators and chain them together?
  23. Python decorator? – can someone please explain this? [duplicate]
  24. Python decorator as a staticmethod
  25. How to create a custom decorator in Django?
  26. Currying decorator in python
  27. Python decorator best practice, using a class vs a function
  28. Find functions explicitly defined in a module (python)
  29. Applying a decorator to every method in a class?
  30. How to watch for a variable change in python without dunder setattr or pdb

Leave a Comment