I don't think you can write a function with the same API as varname() using sorcery, but you can do it with executing. In any case:

Note: previously spells had a complicated implementation that placed limitations on how they could be called. Now spells are just a thin wrapper around executing which is much better. You may be better off using executing directly depending on your use case. This repo is now mostly just a fun collection of things to do with it.

Here's a rough implementation:

import ast

import executing
import sys


def varname(caller=0):
    ex = executing.Source.executing(sys._getframe(caller + 1))
    node = ex.node
    if not node:
        return "var1"  # etc

    while hasattr(node, 'parent'):
        node = node.parent

        if isinstance(node, ast.Assign):
            # Need to actually check that there's just one
            target = node.targets[0]

            # Need to check that it's a variable
            return target.id


def function():
    # I know that at which stack this will be called
    return varname(caller=3)


def function1():
    return function()


def function2():
    return function1()


func = function2()
print(func)

x = varname()
print(x)

executing #3

Description

Metadata

Metadata

Assignees

Labels

Projects

Milestone

Relationships

Development

Issue actions