Как получить имя файла модуля __main__ в Python?

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

Вопрос: как мне получить путь к __main__ модуль (в данном случае "a.py") из библиотеки "b.py"?

Answer 1

import __main__
print __main__.__file__

Answer 2

Возможно, это поможет.

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

Обратите внимание, что в целях безопасности следует проверить, __main__ модуль имеет __file__ приписывать. Если он создается динамически или просто запускается в интерактивной консоли Python, он не будет иметь __file__:

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

Простая проверка hasattr() поможет избежать сценария 2, если это возможно в вашем приложении.

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')

Надеемся, что приведенный выше пример кода может дать дополнительное представление о том, как определить путь к запущенному сценарию...

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] будет пустой строкой, если Python начнет с -c или если проверено с консоли Python.

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.

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