Nhận __name__ của việc gọi mô-đun của hàm bằng Python

Giả sử myapp/foo.pychứa:

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

Và myapp/bar.pychứa:

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

Tôi muốn caller_nameđược đặt thành __name__thuộc tính của mô-đun gọi hàm '(là' myapp.foo ') trong trường hợp này. Điều này có thể giải quyết như thế nào?

Kiểm tra mô-đun kiểm tra:

inspect.stack() sẽ trả về thông tin ngăn xếp.

Bên trong một hàm, inspect.stack()[1]sẽ trả về ngăn xếp của người gọi của bạn. Từ đó, bạn có thể biết thêm thông tin về tên chức năng, mô-đun, v.v. của người gọi.

Xem tài liệu để biết chi tiết:

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

Ngoài ra, Doug Hellmann có một bản viết tốt về mô-đun kiểm tra trong loạt bài PyMOTW của mình:

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

CHỈNH SỬA: Đây là một số mã làm những gì bạn muốn, tôi nghĩ:

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

Đối mặt với một vấn đề tương tự, tôi nhận thấy rằng sys._current_frames () từ mô-đun sys chứa thông tin thú vị có thể giúp bạn mà không cần phải nhập kiểm tra, ít nhất là trong các trường hợp sử dụng cụ thể.

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

Sau đó, bạn có thể "di chuyển lên" bằng cách sử dụng 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__'

Đối với tên tệp, bạn cũng có thể sử dụng f.f_back.f_code.co_filename, theo gợi ý của Mark Roddy ở trên. Tôi không chắc chắn về các giới hạn và lưu ý của phương pháp này (nhiều luồng rất có thể sẽ là một vấn đề) nhưng tôi định sử dụng nó trong trường hợp của mình.

Tôi không khuyên bạn nên làm điều này, nhưng bạn có thể hoàn thành mục tiêu của mình bằng phương pháp sau:

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

Sau đó cập nhật phương pháp hiện có của bạn như sau:

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