web-dev-qa-db-ja.com

Pythonで__main__モジュールのファイル名を取得する方法は？

2つのモジュールがあるとします。

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」）へのパスを取得するにはどうすればよいですか？

pythonpython-module
59
Roman Starkov
import __main__
print __main__.__file__
68
ironfroggy

おそらくこれでうまくいくでしょう：

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から保護するためのトリックを実行します。

34
Jarret Hardie

以下の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')

上記のサンプルコードが、実行中のスクリプトへのパスを決定する方法に関する追加の洞察を提供できることを願っています...

15
popcnt

別の方法は、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コンソールからチェックされる場合は空の文字列になります。

8
lrsjng
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コンパイル済みプログラムに対して機能します。

2
James