Jak uzyskać nazwę pliku głównego modułu w Pythonie?

Załóżmy, że mam dwa moduły:

A.py:

import b
print __name__, __file__

B.py:

print __name__, __file__

Prowadzę "a.py" plik. To drukuje:

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

Pytanie: Jak uzyskać ścieżkę do modułu __main__ ("a.py" w tym przypadku) od wewnątrz "b.py" biblioteka?

Author: Matt Fenwick, 2009-03-03

5 answers

import __main__
print __main__.__file__
 61
Author: ironfroggy,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2009-03-03 16:04:22

Być może to załatwi sprawę:

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

Należy pamiętać, że dla bezpieczeństwa należy sprawdzić, czy moduł __main__ ma atrybut __file__. Jeśli jest tworzony dynamicznie lub jest uruchamiany w interaktywnej konsoli Pythona, nie będzie miał __file__:

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

Proste sprawdzenie hasattr() wystarczy, aby zabezpieczyć się przed scenariuszem 2, jeśli jest to możliwe w Twojej aplikacji.

 30
Author: Jarret Hardie,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2009-03-03 18:39:58

Poniższy kod Pythona zapewnia dodatkową funkcjonalność, w tym bezproblemową pracę z py2exe wykonywalne.

Używam podobnego kodu, aby znaleźć ścieżki względem uruchomionego skryptu, aka __main__. dodatkową korzyścią jest to, że działa na wielu platformach, w tym w systemie 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')

Mam nadzieję, że powyższy przykładowy kod może dostarczyć dodatkowego wglądu w sposób określania ścieżki do uruchomionego skryptu...

 14
Author: popcnt,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2009-03-03 20:29:30

Inną metodą byłoby użycie sys.argv[0].

import os
import sys

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

sys.argv[0] będzie pustym ciągiem znaków, jeśli Python zaczyna się od -c lub jeśli jest zaznaczony z konsoli Pythona.

 7
Author: lrsjng,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2014-04-05 18:16:14
import sys, os

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

Ta funkcja będzie działać dla programów skompilowanych w Pythonie i Cythonie.

 1
Author: James,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2013-01-17 21:17:50