Как получить имя основного модуля в Python?
Предположим, у меня есть два модуля:
а.пы:
import b
print __name__, __file__
Б.пы:
print __name__, __file__
Я управляю "а".py " файл. Это отпечатки:
b C:pathtocodeb.py
__main__ C:pathtocodea.py
вопрос: как получить путь к __main__
модуль ("a.py" в этом случае) изнутри "b.py " библиотека?
5 ответов
возможно, это сделает трюк:
import sys
from os import path
print path.abspath(sys.modules['__main__'].__file__)
обратите внимание, что для безопасности, вы должны проверить, является ли __main__
модуль имеет . Если он динамически создается или просто запускается в интерактивной консоли 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, если это возможно в вашем приложении.
код 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')
надеюсь, приведенный выше пример кода может дать дополнительное представление о том, как определить путь к запущенному скрипту...
другим методом было бы использовать 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.
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.