如何在Python中获取__main__模块的文件名？

Question

假设我有两个模块：

a.py:

import b
print __name__, __file__

b.py：

print __name__, __file__

我运行了 "a.py" 文件。 它打印出：

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

问题：如何在“b.py”库中获取到__main__模块的路径（在本例中为“a.py”）？

- Roman Starkov

---

注意：这里的问题和许多答案使用Python 2的打印语法。对于Python 3，在print的参数周围添加括号，因为它现在是一个函数。 - SuperStormer

Answer 1

import __main__
print(__main__.__file__)

- ironfroggy

---

不错！重要的是要意识到需要导入__main__ - 我猜了一下并尝试了第二行，但它失败了 - 现在我知道原因了。 - Roman Starkov

__main__ 不一定拥有 __file__ 属性。 - jfs

由于romkyn在下面的帖子中的请求，我给你点了个赞。 - Jarret Hardie

@J.F.Sebastian，我们可不可以可靠地使用 inspect.getfile(__main__) 来获取文件路径？ - jpmc26

我不知道。要获取脚本所在的目录（而不是名称），您可以使用get_script_dir()函数。 - jfs

@jfs：如果__main__没有__file__属性，那几乎总是因为它不来自文件——例如，如果您正在交互式运行。在这种情况下，尝试获取文件名是徒劳的。 - user2357112

Answer 2

也许这个可以解决问题：

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

请注意，为了安全起见，您应该检查__main__模块是否有__file__属性。如果它是动态创建的，或者仅在交互式python控制台中运行，则不会有__file__：

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

如果你的应用程序可能会出现第二种情况，那么一个简单的 hasattr() 检查就可以防范问题。

- Jarret Hardie

---

太好了！我很高兴能够帮忙。如果这回答了你的问题，你介意在问题上点击“接受”按钮吗？ :-) - Jarret Hardie

我撤回了更改……公平地说，ironfroggy给出了对你最有效的答案，我真的认为他应该得到唯一的功劳。 - Jarret Hardie

假设主脚本没有__file__属性？那么你怎么获取路径呢？你说要检查主函数是否有__file__属性，但是你的代码没有展示如果主函数没有该属性应该怎么做。 - Toothpick Anemone

@ToothpickAnemone：如果__main__没有__file__属性，那几乎总是因为它不来自文件——例如，如果您正在交互式运行。在这种情况下，尝试获取文件名是徒劳的。 - user2357112

Answer 3

下面的Python代码提供了额外的功能，其中包括它与py2exe可执行文件无缝配合运行。

我使用类似这样的代码来查找相对于正在运行的脚本（即__main__）的路径，作为额外的好处，它跨平台运行，包括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')

希望以上示例代码能够为确定正在运行的脚本路径提供更多的见解...

- popcnt

---

如果在运行脚本和调用os.dirname(sys.argv[0])之间更改目录（使用os.chdir），则结果没有意义。 - RobM

我曾经制作过一个名为 GLOBALS.py 的模块，至少被主脚本导入，执行类似上述代码并将结果存储在变量 MAINDIR 中，因此我可以从任何其他模块访问 GLOBALS.MAINDIR，而不必担心使用 os.chdir。 - Remi

如果您想知道脚本的起始目录，并且还想使用 os.chdir()，那么您需要保存初始的 os.getcwd()。由于用户正在使用Windows，我想演示一些仍然可以在 py2exe 和 freeze 的上下文中工作的内容。 - popcnt

Answer 4

另一种方法是使用 sys.argv[0]。

import os
import sys

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

sys.argv[0]如果从-c启动Python或从Python控制台中检查，则将为空字符串。

- lrsjng

---

如果Python使用-c选项启动，或者从Python控制台检查sys.argv[0]，如何获取到__main__模块的路径？ - Toothpick Anemone

Answer 5

import sys, os

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

这个函数适用于Python和Cython编译程序。

- James

Answer 6

我使用这个：

import __main__
name = __main__.__file__ # this gets file path
lastBar =int(name[::-1].find('/'))  # change '/' for linux or '\' for windows
extension = 3 #3 for python file .py
main_file_name=name[::-1][extension:lastBar][::-1] #result

- Edu Fdez