実行ファイルのモジュールを取得する方法

Question

外部ファイルのモジュールはimport my_moduleで取得できますが、実行ファイルのモジュールを取得する方法はありますか？

import inspect
import my_module

print(inspect.getmembers(my_module, inspect.isfunction))

のようにな感じで、inspect.getmembersの引数に実行ファイルのモジュールを指定したいというモチベーションです。

Answer 1

やりたいことが、
print(inspect.getmembers(my_module, inspect.isfunction))
であるというのを読んでいませんでした。

それでしたら、以下です。

import sys
import inspect

print(inspect.getmembers(sys.modules[__name__], inspect.isfunction))

ただ、もう一つの質問の回答で書いたように、unknown_userさんが欲しいものが出てくる保証はありません。

>>> import sys
>>> import inspect

>>> def f():
...     pass
...
>>> from glob import glob
>>> print(inspect.getmembers(sys.modules[__name__], inspect.isfunction))
[('f', <function f at 0x000001EFFD659EE0>), ('glob', <function glob at 0x000001EFFA7D1280>)]

これが起きるのは、globのような標準モジュールの中にもユーザ定義関数globが入っていることと、from モジュール import 名前
を実行したときに、その名前は現在のモジュールのグローバル変数として登録されるためです。

－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－
以下は関係なかったようですが参考のために残しておきます。

誤りがあったので修正しました。

モジュール名というのが、モジュールの名前属性であるのか、ファイル名であるのかで答えが違います。

モジュールの名前属性であれば、

import sys
print(sys.modules[__name__].__name__)

です。

$ python - ファイル名.py
の場合は、ファイル名が表示されます。

$ python ファイル名.py
の場合は、常に'main'となっています。
Linuxでファイルに実行属性をつけてて以下のように実行した場合も
$ ./ファイル名.py
常に'main'となっています。

モジュールのファイル名であれば、
$ python ファイル名.py
の場合や
$ ./ファイル名.py
の場合は

import sys
print(sys.modules[__name__].__file__)

です。

$ python -m ファイル名.py
の場合は

import sys
print(sys.modules[__name__].__name__)

です。

Comments

[unknown_user]

ありがとうございます。`sys.modules[__name__]`で取得できるのですね。
`inspect.isfunction`だと不十分そうですね...
前の質問で教えていただいた`v.__module__ == my_module.__name__`でフィルタリングしてみようと思います。
また、実行方法の違いで`sys.modules[__name__].__name__`が変わることも教えていただき感謝します。
このあたりは調べてもあまり情報が見つからなかったので理解が深まりました。

Answer 2

if __name__ == "__main__":
    main()

というイディオムを知っていれば分かっているはずですが、実行ファイルのモジュールは常に名前が__main__です。

https://docs.python.org/ja/3/library/main.html#module-main

---

追記

したがってそれ以外のモジュールからは常に

import __main__

としてインポートすることができるということです。

循環参照にならないように気をつけなければいけない循環参照を意識する必要があることと、対話環境などでは__main__は見つからずにエラーになることなどが注意点です。

---

意図的に循環参照したコードの例

a.py

import b

def x():
    return 0

b.py

import __main__

# この時点でxは未定義なのでエラーになる
# __main__に含まれる関数を探してもそこにxはない
print(__main__.x())

Comments

[quickquip]

実行ファイルのモジュール が曖昧です。想像できる候補のもう一つの方ということで回答しておきます。

[unknown_user]

ありがとうございます。期待していた回答は後者の`import __main__`の方です。
`__main__`はimportしたあとでも関数定義をするとモジュールに関数が追加されるのが面白いですね。
`inspect.getmembers(__main__, inspect.isfunction)`はこのメソッドを実行したタイミングまでに定義した関数を取得できました。

[unknown_user]

恐縮ですが`import __main__`で発生しうる循環参照はどういったパターンなのか教えていただけますでしょうか？

[quickquip]

循環参照になるので気を付けてください が正確でしたね。