Question

Before I ask this, do note: I want this for debugging purposes. I know that this is going to be some bad black magic, but I want to use it just during debugging so I could identify my objects more easily.

It’s like this. I have some object from class A that creates a few B instances as attributes:

class A(object):
    def __init__(self)
        self.vanilla_b = B()
        self.chocolate_b = B()

class B(object):
    def __init__(self):
        # ...

What I want is that in B.__init__, it will figure out the "vanilla_b" or whatever attribute name it was given, and then put that as the .name attribute to this specific B.

Then in debugging when I see some B object floating around, I could know which one it is.

Is there any way to do this?

Answer 1

You can use sys._getframe to get the line number where B() is called, then you can use inspect.getsourcelines to get the actual line of code. From there you can parse the line of code to get the thing to which B() is being assigned:

import sys
import inspect

class A(object):
    def __init__(self):
        self.vanilla_b = B()
        self.chocolate_b = B()

class B(object):
    def __init__(self):
        line_num = sys._getframe().f_back.f_lineno
        lines = inspect.getsourcelines( sys.modules[__name__] )[0]
        line = lines[line_num - 1]
        attr = line.split("=")[0].split(".")[1].strip()
        print "B() is being assigned to", attr

A()

If you put the above code into a python script and run it, then it will print

B() is being assigned to vanilla_b
B() is being assigned to chocolate_b

However, this won’t work at the Python command prompt, since __main__ is a builtin module, and thus the inspect module can’t retrieve its source lines. So you might want to wrap this in a try/catch block or something just in case your code ever gets called from any builtin module.

For the record, this is probably a bad idea, but you say that you’re aware that it’s not good practice and you’re only doing it for debugging, so hopefully you’ll be able to use this kind of trickery wisely.