Pythonの基本クラスから継承されたすべてのクラスを取得する実用的なアプローチが必要です。
新しいスタイルのクラス(つまり、Python 3のデフォルトであるobject
のサブクラス)には、サブクラスを返す__subclasses__
メソッドがあります。
class Foo(object): pass
class Bar(Foo): pass
class Baz(Foo): pass
class Bing(Bar): pass
サブクラスの名前は次のとおりです。
print([cls.__for cls in Foo.__subclasses__()])
# ['Bar', 'Baz']
サブクラス自体は次のとおりです。
print(Foo.__subclasses__())
# [<class '__main__.Bar'>, <class '__main__.Baz'>]
サブクラスが実際にベースとしてFoo
をリストすることの確認:
for cls in Foo.__subclasses__():
print(cls.__base__)
# <class '__main__.Foo'>
# <class '__main__.Foo'>
サブサブクラスが必要な場合は、再帰する必要があることに注意してください。
def all_subclasses(cls):
return set(cls.__subclasses__()).union(
[s for c in cls.__subclasses__() for s in all_subclasses(c)])
print(all_subclasses(Foo))
# {<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>}
サブクラスのクラス定義がまだ実行されていない場合-たとえば、サブクラスのモジュールがまだインポートされていない場合-そのサブクラスはまだ存在せず、__subclasses__
はそれを見つけられないことに注意してください。
あなたは「その名前を与えられた」と述べました。 Pythonクラスはファーストクラスオブジェクトであるため、クラスまたはそのようなものの代わりにクラス名の文字列を使用する必要はありません。クラスを直接使用するだけで、おそらくそうする必要があります。
クラスの名前を表す文字列があり、そのクラスのサブクラスを検索する場合は、2つの手順があります。名前が指定されたクラスを検索し、上記の__subclasses__
でサブクラスを検索します。
名前からクラスを見つける方法は、どこで見つけるかによって異なります。クラスを見つけようとしているコードと同じモジュールでそれを見つけることを期待しているなら、
cls = globals()[name]
仕事をするか、地元の人でそれを見つけることを期待しているというまれなケースでは、
cls = locals()[name]
クラスが任意のモジュール内にある可能性がある場合、名前文字列には、完全に修飾された名前('pkg.module.Foo'
ではなく'Foo'
など)を含める必要があります。 importlib
を使用してクラスのモジュールをロードし、対応する属性を取得します。
import importlib
modname, _, clsname = name.rpartition('.')
mod = importlib.import_module(modname)
cls = getattr(mod, clsname)
ただし、クラスを見つけた場合、cls.__subclasses__()
はそのサブクラスのリストを返します。
直接サブクラスが必要な場合は、.__subclasses__()
が正常に機能します。すべてのサブクラス、サブクラスのサブクラスなどが必要な場合、それを行うための関数が必要になります。
これは、特定のクラスのすべてのサブクラスを再帰的に検索する、シンプルで読み取り可能な関数です。
def get_all_subclasses(cls):
all_subclasses = []
for subclass in cls.__subclasses__():
all_subclasses.append(subclass)
all_subclasses.extend(get_all_subclasses(subclass))
return all_subclasses
一般的な形式の最も簡単なソリューション:
def get_subclasses(cls):
for subclass in cls.__subclasses__():
yield from get_subclasses(subclass)
yield subclass
そして、あなたが継承する単一のクラスがある場合のクラスメソッド:
@classmethod
def get_subclasses(cls):
for subclass in cls.__subclasses__():
yield from subclass.get_subclasses()
yield subclass
__init_subclass__
python 3.6以降は __subclasses__
メソッドをオーバーライドすることでこの属性の作成を変更できるため、他の回答で言及したように、__init_subclass__
属性をチェックしてサブクラスのリストを取得できます。
class PluginBase:
subclasses = []
def __init_subclass__(cls, **kwargs):
super().__init_subclass__(**kwargs)
cls.subclasses.append(cls)
class Plugin1(PluginBase):
pass
class Plugin2(PluginBase):
pass
このようにして、あなたが何をしているのかがわかっている場合、__subclasses__
の動作をオーバーライドし、このリストからサブクラスを省略/追加できます。
FWIW、これが私が意味することです @ unutbu's answer ローカルで定義されたクラスでのみ動作します— eval()
の代わりにvars()
を使用すると、アクセス可能なクラスだけでなく、現在のスコープで定義されているもの。
eval()
の使用を嫌う人のために、それを避ける方法も示されています。
最初に、vars()
の使用に伴う潜在的な問題を示す具体的な例を示します。
class Foo(object): pass
class Bar(Foo): pass
class Baz(Foo): pass
class Bing(Bar): pass
# unutbu's approach
def all_subclasses(cls):
return cls.__subclasses__() + [g for s in cls.__subclasses__()
for g in all_subclasses(s)]
print(all_subclasses(vars()['Foo'])) # Fine because Foo is in scope
# -> [<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>]
def func(): # won't work because Foo class is not locally defined
print(all_subclasses(vars()['Foo']))
try:
func() # not OK because Foo is not local to func()
except Exception as e:
print('calling func() raised exception: {!r}'.format(e))
# -> calling func() raised exception: KeyError('Foo',)
print(all_subclasses(eval('Foo'))) # OK
# -> [<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>]
# using eval('xxx') instead of vars()['xxx']
def func2():
print(all_subclasses(eval('Foo')))
func2() # Works
# -> [<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>]
これはeval('ClassName')
を定義された関数に移動することで改善できます。これにより、eval()
はコンテキスト依存ではないvars()
を使用することで得られる追加の一般性を失うことなく使いやすくなります:
# easier to use version
def all_subclasses2(classname):
direct_subclasses = eval(classname).__subclasses__()
return direct_subclasses + [g for s in direct_subclasses
for g in all_subclasses2(s.__name__)]
# pass 'xxx' instead of eval('xxx')
def func_ez():
print(all_subclasses2('Foo')) # simpler
func_ez()
# -> [<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>]
最後に、セキュリティ上の理由でeval()
を使用しないようにすることが可能であり、場合によっては重要になることもあります。そのため、次のバージョンがあります。
def get_all_subclasses(cls):
""" Generator of all a class's subclasses. """
try:
for subclass in cls.__subclasses__():
yield subclass
for subclass in get_all_subclasses(subclass):
yield subclass
except TypeError:
return
def all_subclasses3(classname):
for cls in get_all_subclasses(object): # object is base of all new-style classes.
if cls.__name__.split('.')[-1] == classname:
break
else:
raise ValueError('class %s not found' % classname)
direct_subclasses = cls.__subclasses__()
return direct_subclasses + [g for s in direct_subclasses
for g in all_subclasses3(s.__name__)]
# no eval('xxx')
def func3():
print(all_subclasses3('Foo'))
func3() # Also works
# -> [<class '__main__.Bar'>, <class '__main__.Baz'>, <class '__main__.Bing'>]
すべてのサブクラスのリストを取得するためのはるかに短いバージョン:
from itertools import chain
def subclasses(cls):
return list(
chain.from_iterable(
[list(chain.from_iterable([[x], subclasses(x)])) for x in cls.__subclasses__()]
)
)
これは、@ unutbuが言及している特別なbuilt_in__subclasses__()
classメソッドを使用するのと同じくらい良い答えではないので、単に演習として提示します。定義されたsubclasses()
functionは、すべてのサブクラス名をサブクラス自体にマップする辞書を返します。
def traced_subclass(baseclass):
class _SubclassTracer(type):
def __new__(cls, classname, bases, classdict):
obj = type(classname, bases, classdict)
if baseclass in bases: # sanity check
attrname = '_%s__derived' % baseclass.__name__
derived = getattr(baseclass, attrname, {})
derived.update( {classname:obj} )
setattr(baseclass, attrname, derived)
return obj
return _SubclassTracer
def subclasses(baseclass):
attrname = '_%s__derived' % baseclass.__name__
return getattr(baseclass, attrname, None)
class BaseClass(object):
pass
class SubclassA(BaseClass):
__metaclass__ = traced_subclass(BaseClass)
class SubclassB(BaseClass):
__metaclass__ = traced_subclass(BaseClass)
print subclasses(BaseClass)
出力:
{'SubclassB': <class '__main__.SubclassB'>,
'SubclassA': <class '__main__.SubclassA'>}
名前を指定してクラスのすべてのサブクラスを見つけるにはどうすればよいですか?
確かに、オブジェクト自体へのアクセスがあれば、これを簡単に行うことができます。
同じ名前で複数のクラスがあり、同じモジュールで定義されていることもあるため、単純に名前を指定するのは適切ではありません。
別の answer の実装を作成しましたが、この質問に答えており、ここにある他のソリューションよりも少しエレガントなので、ここにあります:
def get_subclasses(cls):
"""returns all subclasses of argument, cls"""
if issubclass(cls, type):
subclasses = cls.__subclasses__(cls)
else:
subclasses = cls.__subclasses__()
for subclass in subclasses:
subclasses.extend(get_subclasses(subclass))
return subclasses
使用法:
>>> import pprint
>>> list_of_classes = get_subclasses(int)
>>> pprint.pprint(list_of_classes)
[<class 'bool'>,
<enum 'IntEnum'>,
<enum 'IntFlag'>,
<class 'sre_constants._NamedIntConstant'>,
<class 'subprocess.Handle'>,
<enum '_ParameterKind'>,
<enum 'Signals'>,
<enum 'Handlers'>,
<enum 'RegexFlag'>]
再帰のないバージョンは次のとおりです。
def get_subclasses_gen(cls):
def _subclasses(classes, seen):
while True:
subclasses = sum((x.__subclasses__() for x in classes), [])
yield from classes
yield from seen
found = []
if not subclasses:
return
classes = subclasses
seen = found
return _subclasses([cls], [])
これは、元のクラスを返すという点で他の実装とは異なります。これは、コードが簡単になり、次の理由によります。
class Ham(object):
pass
assert(issubclass(Ham, Ham)) # True
Get_subclasses_genが少し奇妙に見える場合、それは末尾再帰実装をループジェネレーターに変換することによって作成されたためです。
def get_subclasses(cls):
def _subclasses(classes, seen):
subclasses = sum(*(frozenset(x.__subclasses__()) for x in classes))
found = classes + seen
if not subclasses:
return found
return _subclasses(subclasses, found)
return _subclasses([cls], [])
現実のユースケースを想像することはできませんが、堅牢な方法(Python 2つの古いスタイルのクラスであっても)はグローバル名前空間をスキャンすることです。
def has_children(cls):
g = globals().copy() # use a copy to make sure it will not change during iteration
g.update(locals()) # add local symbols
for k, v in g.items(): # iterate over all globals object
try:
if (v is not cls) and issubclass(v, cls): # found a strict sub class?
return True
except TypeError: # issubclass raises a TypeError if arg is not a class...
pass
return False
Python 2つの新しいスタイルクラスとPython 3クラス、およびPython 2 classicクラスで動作します。