Verilen bir modülde belirli bir sınıfa ait alt sınıflar üzerinde yineleyin

Question

Python'da, X modülünü ve Y sınıfını verdiğinizde, X modülünde var olan Y alt sınıflarının bir listesini nasıl oluşturabilirim veya oluşturabilirim?

Answer 1

İşte bunu yapmak için tek yolu şudur:

import inspect 

def get_subclasses(mod, cls): 
    """Yield the classes in module ``mod`` that inherit from ``cls``""" 
    for name, obj in inspect.getmembers(mod): 
     if hasattr(obj, "__bases__") and cls in obj.__bases__: 
      yield obj 

Answer 2

Chris Atlee ve zacherates gelen cevapların ne gereklerini yerine getirmek olduğunu önerebilir

class foo(object): pass 
class bar(foo): pass 
class baz(foo): pass 

class grar(Exception): pass 

def find_subclasses(module, clazz): 
    for name in dir(module): 
     o = getattr(module, name) 

     try: 
      if issubclass(o, clazz): 
      yield name, o 
     except TypeError: pass 

>>> import foo 
>>> list(foo.find_subclasses(foo, foo.foo)) 
[('bar', <class 'foo.bar'>), ('baz', <class 'foo.baz'>), ('foo', <class 'foo.foo'>)] 
>>> list(foo.find_subclasses(foo, object)) 
[('bar', <class 'foo.bar'>), ('baz', <class 'foo.baz'>), ('foo', <class 'foo.foo'>), ('grar', <class 'foo.grar'>)] 
>>> list(foo.find_subclasses(foo, Exception)) 
[('grar', <class 'foo.grar'>)] 

Answer 3

foo.py modül Verilen? Ben zacerates cevap bu değişiklik olduğunu düşünüyorum daha iyi:

def find_subclasses(module, clazz): 
    for name in dir(module): 
     o = getattr(module, name) 
     try: 
      if (o != clazz) and issubclass(o, clazz): 
       yield name, o 
     except TypeError: pass 

Bana verilen cevaplar katılmıyorum nedeni ilk verilen sınıfın uzak bir alt sınıfıdır sınıfları üretmez ve ikinci verilmiş içerir olmasıdır sınıf.

Answer 4

Quamrana'nın önerisi iyi çalışıyor olsa da, daha fazla pythonic yapmayı önermek istiyorum birkaç olası iyileştirmeler var. Standart kütüphaneden inceleme modülünü kullanmaya güveniyorlar.

1. Sen inspect.getmembers()
2. deneme/yakalama kullanarak getattr çağrıyı önleyebilirsiniz olanlar ile inspect.isclass()

kullanılarak önlenebilir, sen eğer tek bir liste anlama Tüm olayı azaltabilir benzeri:

def find_subclasses(module, clazz): 
    return [ 
     cls 
      for name, cls in inspect.getmembers(module) 
       if inspect.isclass(cls) and issubclass(cls, clazz) 
    ]