Obtener __name__ de llamar módulo de función en Python

Question

Supongamos myapp/foo.py contiene:

def info(msg): 
    caller_name = ???? 
    print '[%s] %s' % (caller_name, msg) 

Y myapp/bar.py contiene:

import foo 
foo.info('Hello') # => [myapp.bar] Hello 

Quiero caller_name se establezca en el atributo __name__ del módulo de las funciones de llamada (que es 'myapp.foo') en este caso. ¿Cómo puede hacerse esto?

Answer 1

Salida inspeccionar el módulo:

inspect.stack() devolverá la información de la pila.

Dentro de una función, inspect.stack()[1] devolverá la pila de su llamante. A partir de ahí, se puede obtener más información acerca de nombre de la función de la persona que llama, módulo, etc.

Consulte la documentación para más detalles:

http://docs.python.org/library/inspect.html

Además, Doug Hellmann tiene una buena valoración crítica del módulo de inspección en su serie PyMOTW:

http://pymotw.com/2/inspect/index.html#module-inspect

EDIT: Aquí hay algo de código que hace lo que quiere, pienso:

def info(msg): 
    frm = inspect.stack()[1] 
    mod = inspect.getmodule(frm[0]) 
    print '[%s] %s' % (mod.__name__, msg) 

Answer 2

No recomiendo hacer esto, pero se puede lograr su objetivo con el siguiente método:

def caller_name(): 
    frame=inspect.currentframe() 
    frame=frame.f_back.f_back 
    code=frame.f_code 
    return code.co_filename 

luego actualizar su método existente de la siguiente manera:

def info(msg): 
    caller = caller_name() 
    print '[%s] %s' % (caller, msg) 

Answer 3

Confrontado con un similares problema, he encontrado que sys._current_frames() del módulo sys contiene información interesante que puede ayudarlo, sin la necesidad de importar inspeccionar, al menos en casos de uso específicos.

>>> sys._current_frames() 
{4052: <frame object at 0x03200C98>} 

A continuación, puede "moverse hacia arriba" usando f_back:

>>> f = sys._current_frames().values()[0] 

>>> print f.f_back.f_globals['__file__'] 
'/base/data/home/apps/apricot/1.6456165165151/caller.py' 

>>> print f.f_back.f_globals['__name__'] 
'__main__' 

Para el nombre del archivo también se puede utilizar f.f_back.f_code.co_filename, según lo sugerido por Mark Roddy anteriormente. No estoy seguro de los límites y advertencias de este método (es probable que haya varios problemas) pero tengo la intención de usarlo en mi caso.