Function Names on a function call

[indiarocks]

Let's say for eg. I have one common function say 'common_func' and it can be called by any other functions. If func1, makes a call to common_func, I need to determine in common_func that this call was made by func1. If func2 makes a call to common_func, that this call was made by func2.

I need a way to figure out which function made a function call to common_func.

I don't know if I am making sense.

Thanks,
-V

[ilikepython]

Let's say for eg. I have one common function say 'common_func' and it can be called by any other functions. If func1, makes a call to common_func, I need to determine in common_func that this call was made by func1. If func2 makes a call to common_func, that this call was made by func2.

I need a way to figure out which function made a function call to common_func.

I don't know if I am making sense.

Thanks,
-V

You could have an argument in common_func that gets passed different values in func1 and func2:

def common_func(call_func):
    if call_func == 1:
        # called from func1
    if call_func == 2:
        # called from func2
 
def func1():
    common_func(1)
 
def func2():
    common_func(2)
 

Why do you need that?

[bartonc]

Let's say for eg. I have one common function say 'common_func' and it can be called by any other functions. If func1, makes a call to common_func, I need to determine in common_func that this call was made by func1. If func2 makes a call to common_func, that this call was made by func2.

I need a way to figure out which function made a function call to common_func.

I don't know if I am making sense.

Thanks,
-V

If you want to do that without adding parameters to the function parameter list, you'll want to use the introspection provided my the inspect module:

import inspect
 
class FunctionOwner:
    def __init__(self):
        self.caller1 = self.func1.im_func.func_code
        self.caller2 = self.func2.im_func.func_code
 
    def __del__(self):
        del self.caller1
        del self.caller2
 
    def common_func(self):
        caller = inspect.currentframe().f_back.f_code
        if caller == self.caller1:
            print "called by caller 1"
        elif caller == self.caller2:
            print "called by caller 2"
 
    def func1(self):
        self.common_func()
 
    def func2(self):
        self.common_func()

This also applies to module functions, but I'm a class object fan and rarely define things like this at the module level.

[bartonc]

If you want to do that without adding parameters to the function parameter list, you'll want to use the introspection provided my the inspect module:

import inspect
 
class FunctionOwner:
    def __init__(self):
        self.caller1 = self.func1.im_func.func_code
        self.caller2 = self.func2.im_func.func_code
 
    def __del__(self):
        del self.caller1
        del self.caller2
 
    def common_func(self):
        caller = inspect.currentframe().f_back.f_code
        if caller == self.caller1:
            print "called by caller 1"
        elif caller == self.caller2:
            print "called by caller 2"
 
    def func1(self):
        self.common_func()
 
    def func2(self):
        self.common_func()

This also applies to module functions, but I'm a class object fan and rarely define things like this at the module level.

Well, I think it's kind of cool, even if no one else does.

[indiarocks]

Thanks for the tip. But what if I don't know the name of function that can call common_func. It could be any arbitrary function name defined by the user who calls common_func.

Thanks

If you want to do that without adding parameters to the function parameter list, you'll want to use the introspection provided my the inspect module:

import inspect
 
class FunctionOwner:
    def __init__(self):
        self.caller1 = self.func1.im_func.func_code
        self.caller2 = self.func2.im_func.func_code
 
    def __del__(self):
        del self.caller1
        del self.caller2
 
    def common_func(self):
        caller = inspect.currentframe().f_back.f_code
        if caller == self.caller1:
            print "called by caller 1"
        elif caller == self.caller2:
            print "called by caller 2"
 
    def func1(self):
        self.common_func()
 
    def func2(self):
        self.common_func()

This also applies to module functions, but I'm a class object fan and rarely define things like this at the module level.

[indiarocks]

Was playing around with bartonc's tip and found out the answer.

caller = inspect.currentframe().f_back.f_code
print "I was called by " , caller.co_name
 

Thanks,
-V

[bartonc]

Was playing around with bartonc's tip and found out the answer.

caller = inspect.currentframe().f_back.f_code
print "I was called by " , caller.co_name
 

Thanks,
-V

Thanks. Sorry that I missed that part. Glad you found your way through the docs!

[MisterPete]

Is there a way to determine the calling module name as well? Ie, I want to print "called by MyModule.SubModule.Foo()" not just "called by Foo()"
I think I saw the file path for the module in there but I don't want to have to try to determine the submodules from the path if possible.

Cheers,
Pete

[bartonc]

Is there a way to determine the calling module name as well? Ie, I want to print "called by MyModule.SubModule.Foo()" not just "called by Foo()"
I think I saw the file path for the module in there but I don't want to have to try to determine the submodules from the path if possible.

Cheers,
Pete

Hi Pete. Building on the example above:

#.....
        print inspect.getfile(caller)

There are others, as well. The docs are here.

Have fun!