How to get current function into a variable?

by

The stack frame tells us what code object we’re in. If we can find a function object that refers to that code object in its __code__ attribute, we have found the function.

Fortunately, we can ask the garbage collector which objects hold a reference to our code object, and sift through those, rather than having to traverse every active object in the Python world. There are typically only a handful of references to a code object.

Now, functions can share code objects, and do in the case where you return a function from a function, i.e. a closure. When there’s more than one function using a given code object, we can’t tell which function it is, so we return None.

import inspect, gc

def giveupthefunc():
    frame = inspect.currentframe(1)
    code  = frame.f_code
    globs = frame.f_globals
    functype = type(lambda: 0)
    funcs = []
    for func in gc.get_referrers(code):
        if type(func) is functype:
            if getattr(func, "__code__", None) is code:
                if getattr(func, "__globals__", None) is globs:
                    funcs.append(func)
                    if len(funcs) > 1:
                        return None
    return funcs[0] if funcs else None

Some test cases:

def foo():
    return giveupthefunc()

zed = lambda: giveupthefunc()

bar, foo = foo, None

print bar()
print zed()

I’m not sure about the performance characteristics of this, but i think it should be fine for your use case.

More Related Contents:

  1. Determine function name from within that function (without using traceback)
  2. Python – Class.function().function()
  3. Function to return results in a dictionary in Python
  4. Why does this recursion return 0?
  5. Why is the output of my function printing out “None”?
  6. How to get the original variable name of variable passed to a function
  7. UnboundLocalError with nested function scopes
  8. Passing functions with arguments to another function in Python?
  9. What does it mean to “call” a function in Python? [closed]
  10. Python: Passing variables between functions
  11. Return value from button click
  12. Using a dictionary to select function to execute
  13. Calling a function upon button press
  14. Python: dynamically create function at runtime
  15. How to access a function inside a function?
  16. Get all object attributes in Python? [duplicate]
  17. Setting variables with exec inside a function
  18. List all base classes in a hierarchy of given class?
  19. How can I make a deepcopy of a function in Python?
  20. Function chaining in Python
  21. Efficient evaluation of a function at every cell of a NumPy array
  22. Why should functions always return the same type?
  23. In Python, what does ” mean?
  24. break and continue in function
  25. Python functions with multiple parameter brackets
  26. How to make function decorators and chain them together in Python?
  27. I have a string whose content is a function name, how to refer to the corresponding function in Python?
  28. Check if all values in list are greater than a certain number
  29. Possible to change a function’s repr in python?
  30. Is it ok to skip “return None”?

Leave a Comment