Ответ 1
import __main__
print __main__.__file__
Предположим, что у меня есть два модуля:
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"?
import __main__
print __main__.__file__
Возможно, это будет трюк:
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, если это возможно в вашем приложении.
Ниже приведенный ниже код на Python обеспечивает дополнительную функциональность, включая то, что он легко работает с py2exe
исполняемыми файлами.
Я использую аналогичный код для этого, чтобы найти пути относительно запуска script, aka __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')
Надеюсь, приведенный выше примерный код может дать дополнительное представление о том, как определить путь к запущенному script...
Другим методом будет использование 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.