web-dev-qa-db-ja.com

Pythonの現在のモジュール内のすべてのクラスのリストを取得するにはどうすればよいですか?

モジュールからすべてのクラスを抽出する人々の例を見てきました。通常は次のようなものです。

# foo.py
class Foo:
    pass

# test.py
import inspect
import foo

for name, obj in inspect.getmembers(foo):
    if inspect.isclass(obj):
        print obj

驚くばかり。

しかし、currentモジュールからすべてのクラスを取得する方法がわかりません。

# foo.py
import inspect

class Foo:
    pass

def print_classes():
    for name, obj in inspect.getmembers(???): # what do I do here?
        if inspect.isclass(obj):
            print obj

# test.py
import foo

foo.print_classes()

これはおそらく本当に明らかなことですが、私は何も見つけることができませんでした。誰も私を助けることができますか?

263
mcccclean

これを試して:

import sys
current_module = sys.modules[__name__]

あなたの文脈で:

import sys, inspect
def print_classes():
    for name, obj in inspect.getmembers(sys.modules[__name__]):
        if inspect.isclass(obj):
            print(obj)

そしてさらに良い:

clsmembers = inspect.getmembers(sys.modules[__name__], inspect.isclass)

inspect.getmembers()は述語を取るからです。

341
Nadia Alramli

どう?

g = globals().copy()
for name, obj in g.iteritems():

18
Krab

それを行うための「適切な」方法があるかどうかはわかりませんが、スニペットは正しい軌道に乗っています:foo.pyにimport fooを追加し、inspect.getmembers(foo)を実行すれば、うまくいくはずです。

13
int3

dir ビルトインプラス getattr から必要なものすべてを取得できました。

# Works on pretty much everything, but be mindful that 
# you get lists of strings back

print dir(myproject)
print dir(myproject.mymodule)
print dir(myproject.mymodule.myfile)
print dir(myproject.mymodule.myfile.myclass)

# But, the string names can be resolved with getattr, (as seen below)

しかし、それは毛玉のように見えます:

def list_supported_platforms():
    """
        List supported platforms (to match sys.platform)

        @Retirms:
            list str: platform names
    """
    return list(itertools.chain(
        *list(
            # Get the class's constant
            getattr(
                # Get the module's first class, which we wrote
                getattr(
                    # Get the module
                    getattr(platforms, item),
                    dir(
                        getattr(platforms, item)
                    )[0]
                ),
                'SYS_PLATFORMS'
            )
            # For each include in platforms/__init__.py 
            for item in dir(platforms)
            # Ignore magic, ourselves (index.py) and a base class.
            if not item.startswith('__') and item not in ['index', 'base']
        )
    ))
8
ThorSummoner
import pyclbr
print(pyclbr.readmodule(__name__).keys())

StdlibのPythonクラスブラウザーモジュールは静的ソース分析を使用するため、実際の.pyファイルによってサポートされているモジュールに対してのみ機能することに注意してください。

6
ncoghlan

現在のモジュールに属するすべてのクラスが必要な場合は、これを使用できます。

import sys, inspect
def print_classes():
    is_class_member = lambda member: inspect.isclass(member) and member.__module__ == __name__
    clsmembers = inspect.getmembers(sys.modules[__name__], is_class_member)

Nadiaの回答を使用し、モジュールに他のクラスをインポートしていた場合、そのクラスもインポートされます。

そのため、member.__module__ == __name__で使用される述語にis_class_memberが追加されています。このステートメントは、クラスが本当にモジュールに属していることを確認します。

述語は、ブール値を返す関数(呼び出し可能)です。

4
Benjy Malca

Python 2および3で機能する別のソリューション:

#foo.py
import sys

class Foo(object):
    pass

def print_classes():
    current_module = sys.modules[__name__]
    for key in dir(current_module):
        if isinstance( getattr(current_module, key), type ):
            print(key)

# test.py
import foo
foo.print_classes()
3
Florian

これは、現在のモジュールで定義されている(つまりインポートされていない)すべてのクラスを取得するために使用する行です。 PEP-8によると少し長いですが、必要に応じて変更できます。

import sys
import inspect

classes = [name for name, obj in inspect.getmembers(sys.modules[__name__], inspect.isclass) 
          if obj.__module__ is __name__]

これにより、クラス名のリストが表示されます。クラスオブジェクト自体が必要な場合は、代わりにobjを保持します。

classes = [obj for name, obj in inspect.getmembers(sys.modules[__name__], inspect.isclass)
          if obj.__module__ is __name__]

これは私の経験ではもっと便利です。

2

このようなことができると思います。

class custom(object):
    __custom__ = True
class Alpha(custom):
    something = 3
def GetClasses():
    return [x for x in globals() if hasattr(globals()[str(x)], '__custom__')]
print(GetClasses())`

独自のクラスが必要な場合

0
Rain0Ash