รับ __name__ ของโมดูลการเรียกฟังก์ชันใน Python

สมมติว่าmyapp/foo.pyประกอบด้วย:

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

และmyapp/bar.pyประกอบด้วย:

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

ฉันต้องการcaller_nameตั้งค่าเป็น__name__แอตทริบิวต์ของโมดูลการเรียกฟังก์ชัน (ซึ่งก็คือ 'myapp.foo') ในกรณีนี้ จะทำได้อย่างไร?

ตรวจสอบโมดูลการตรวจสอบ:

inspect.stack() จะส่งคืนข้อมูลสแต็ก

ภายในฟังก์ชันinspect.stack()[1]จะส่งคืนสแต็กผู้โทรของคุณ จากนั้นคุณจะได้รับข้อมูลเพิ่มเติมเกี่ยวกับชื่อฟังก์ชันโมดูล ฯลฯ ของผู้โทร

ดูเอกสารเพื่อดูรายละเอียด:

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

นอกจากนี้ Doug Hellmann ยังมีการเขียนโมดูลการตรวจสอบที่ดีในซีรีส์ PyMOTW ของเขา:

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

แก้ไข: นี่คือรหัสบางส่วนที่ทำในสิ่งที่คุณต้องการฉันคิดว่า:

import inspect 

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

เมื่อเผชิญกับปัญหาที่คล้ายกันฉันพบว่าsys._current_frames ()จากโมดูล sys มีข้อมูลที่น่าสนใจที่สามารถช่วยคุณได้โดยไม่จำเป็นต้องนำเข้าการตรวจสอบอย่างน้อยก็ในกรณีการใช้งานเฉพาะ

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

จากนั้นคุณสามารถ "เลื่อนขึ้น" โดยใช้ f_back:

>>> 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.f_back.f_code.co_filename ได้ตามที่ Mark Roddy แนะนำไว้ข้างต้น ฉันไม่แน่ใจเกี่ยวกับขีด จำกัด และข้อควรระวังของวิธีนี้ (หลายเธรดน่าจะเป็นปัญหา) แต่ฉันตั้งใจจะใช้ในกรณีของฉัน

ฉันไม่แนะนำให้ทำเช่นนี้ แต่คุณสามารถบรรลุเป้าหมายได้ด้วยวิธีการต่อไปนี้:

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

จากนั้นอัปเดตวิธีการที่มีอยู่ของคุณดังนี้:

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