Cách lấy tên tệp của mô-đun __main__ bằng Python?

Question

Giả sử tôi có hai mô-đun:

a.py:

import b 
print __name__, __file__ 

b.py:

print __name__, __file__ 

Tôi chạy "a.py" tập tin. in này:

b  C:\path\to\code\b.py 
__main__ C:\path\to\code\a.py 

Câu hỏi: làm thế nào để có được đường dẫn đến __main__ mô-đun ("a.py" trong trường hợp này) từ bên trong thư viện "b.py"?

Answer 1

import __main__ 
print __main__.__file__ 

Answer 2

lẽ điều này sẽ làm các trick:

import sys 
from os import path 
print path.abspath(sys.modules['__main__'].__file__) 

Lưu ý rằng, vì sự an toàn, bạn nên kiểm tra xem các mô-đun __main__ có một thuộc tính __file__. Nếu nó tự động tạo ra, hoặc là chỉ được chạy trong python console tương tác, nó sẽ không có một __file__:

python 
>>> import sys 
>>> print sys.modules['__main__'] 
<module '__main__' (built-in)> 
>>> print sys.modules['__main__'].__file__ 
AttributeError: 'module' object has no attribute '__file__' 

Một hasattr đơn giản() kiểm tra sẽ làm các trick để bảo vệ chống lại kịch bản 2 nếu đó là một khả năng trong ứng dụng của bạn.

Answer 3

Mã python bên dưới cung cấp chức năng bổ sung, bao gồm chức năng hoạt động liền mạch với py2exe thực thi.

Tôi sử dụng mã tương tự như thế này để tìm đường dẫn liên quan đến tập lệnh đang chạy, còn gọi là __main__. như một lợi ích bổ sung, nó hoạt động đa nền tảng bao gồm cả Windows.

import imp 
import os 
import sys 

def main_is_frozen(): 
    return (hasattr(sys, "frozen") or # new py2exe 
      hasattr(sys, "importers") # old py2exe 
      or imp.is_frozen("__main__")) # tools/freeze 

def get_main_dir(): 
    if main_is_frozen(): 
     # print 'Running from path', os.path.dirname(sys.executable) 
     return os.path.dirname(sys.executable) 
    return os.path.dirname(sys.argv[0]) 

# find path to where we are running 
path_to_script=get_main_dir() 

# OPTIONAL: 
# add the sibling 'lib' dir to our module search path 
lib_path = os.path.join(get_main_dir(), os.path.pardir, 'lib') 
sys.path.insert(0, lib_path) 

# OPTIONAL: 
# use info to find relative data files in 'data' subdir 
datafile1 = os.path.join(get_main_dir(), 'data', 'file1') 

Hy vọng rằng các mã ví dụ trên có thể cung cấp cái nhìn sâu sắc hơn về cách để xác định đường dẫn đến kịch bản chạy ...

Answer 4

import sys, os 

def getExecPath(): 
    try: 
     sFile = os.path.abspath(sys.modules['__main__'].__file__) 
    except: 
     sFile = sys.executable 
    return os.path.dirname(sFile) 

Chức năng này sẽ làm việc cho chương trình Python và Cython biên soạn.

Answer 5

Phương pháp khác sẽ là sử dụng sys.argv[0].

import os 
import sys 

main_file = os.path.realpath(sys.argv[0]) if sys.argv[0] else None 

sys.argv[0] sẽ là một chuỗi rỗng nếu Python được bắt đầu với -c hoặc nếu kiểm tra từ Python console.