Ruft name des aufrufenden Funktionsmoduls in Python ab

Angenommen, myapp/foo.pyenthält:

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

Und myapp/bar.pyenthält:

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

Ich möchte in diesem Fall caller_nameauf das __name__Attribut des aufrufenden Funktionsmoduls (myapp.foo) gesetzt werden. Wie kann das gemacht werden?

python stack-trace introspection

— Sridhar Ratnakumar
quelle

Angenommen, ein anderes Einstiegspunktskript ruft bar.py .. auf und caller_namekann es daher nicht sein__main__

— Sridhar Ratnakumar

Antworten:

122

Überprüfen Sie das Inspektionsmodul:

inspect.stack() gibt die Stapelinformationen zurück.

Gibt innerhalb einer Funktion inspect.stack()[1]den Stapel Ihres Anrufers zurück. Von dort erhalten Sie weitere Informationen über den Funktionsnamen, das Modul usw. des Anrufers.

Einzelheiten finden Sie in den Dokumenten:

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

Außerdem hat Doug Hellmann eine schöne Beschreibung des Inspektionsmoduls in seiner PyMOTW-Serie:

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

EDIT: Hier ist ein Code, der macht, was Sie wollen, denke ich:

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

— ars
quelle

Wie erhalten Sie das __name__Attribut dieses Moduls mithilfe des inspectModuls? Wie komme ich beispielsweise in meinem obigen Beispiel zurück myapp.foo(nicht myapp/foo.py)? Ich habe bereits vor dem Posten bei SO versucht, das Inspect-Modul zu verwenden.

— Sridhar Ratnakumar

Beachten Sie, dass dies seltsamerweise mit Import-Hooks interagiert, unter Ironpython nicht funktioniert und sich unter Jython möglicherweise überraschend verhält. Es ist am besten, wenn Sie solche Magie vermeiden können.

— Glyphe

Beachten Sie auch, dass das Beibehalten eines Verweises auf einen Stapelrahmen die ordnungsgemäße Funktion von Pythons GC beeinträchtigen kann. Siehe Warnung hier: docs.python.org/library/inspect.html#the-interpreter-stack

— Kamil Kisiel

Beachten Sie, dass Sie inspect.stack()[2]für den echten Anrufer zugreifen müssen, wenn die Anruferfunktion dekoriert ist (@ ...) .

— Amir Ali Akbari

Beachten Sie auch, dass diese Logik nicht ordnungsgemäß funktioniert, wenn Sie Ihren Python-Code mithilfe von Pyinstaller in eine Exe kompilieren.

— Panofish

Angesichts eines ähnlichen Problems habe ich festgestellt, dass sys._current_frames () aus dem sys-Modul interessante Informationen enthält, die Ihnen helfen können, Inspektion zumindest in bestimmten Anwendungsfällen zu importieren.

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

Sie können dann mit f_back "aufsteigen":

>>> f = sys._current_frames().values()[0]
>>> # for python3: f = list(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__'

Für den Dateinamen können Sie auch f.f_back.f_code.co_filename verwenden, wie oben von Mark Roddy vorgeschlagen. Ich bin mir der Grenzen und Einschränkungen dieser Methode nicht sicher (mehrere Threads werden höchstwahrscheinlich ein Problem sein), aber ich beabsichtige, sie in meinem Fall zu verwenden.

— Louis LC
quelle

Hinweis: Der inspect.stack-Code schlägt fehl, nachdem er mit pyinstaller kompiliert wurde, aber mit sys._current_frames. WORKS FINE ... das ist die bevorzugte Technik für mich.

— Panofish

Ich denke, es ist einfacher, den vorherigen Frame zu erhalten sys._getframe(1), als ihn aufzurufen sys._current_frames()(übrigens, der für jeden Thread eine Frame-Zuordnung zurückgibt).

— Hooblei

Vielen Dank, hooblei, ich habe es noch nicht getestet, aber es scheint sehr nützlich für Multithread-Situationen zu sein.

— Louis LC

Ich benutze lieber inspect.currentframe()statt sys._current_frames().values()[0].

— Aran-Fey

Ich empfehle dies nicht, aber Sie können Ihr Ziel mit der folgenden Methode erreichen:

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

Aktualisieren Sie dann Ihre vorhandene Methode wie folgt:

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

— Mark Roddy
quelle

Dateiname ist nicht dasselbe wie__name__

— Sridhar Ratnakumar

Ruft __name__ des aufrufenden Funktionsmoduls in Python ab

Ruft name des aufrufenden Funktionsmoduls in Python ab