générique Python métaclasse pour garder la trace des sous-classes

Question

Je suis en train d'écrire un métaclasse générique pour les sous-classes de suivi

Depuis que je veux que ce soit générique, je ne voulais pas coder en dur un nom de classe dans cette métaclasse, donc Je suis venu avec une fonction qui génère la métaclasse appropriée, quelque chose comme:

def make_subtracker(root): 
    class SubclassTracker(type): 
     def __init__(cls, name, bases, dct): 
      print('registering %s' % (name,)) 
      root._registry.append(cls) 
      super(SubclassTracker, cls).__init__(name, bases, dct) 
    return SubclassTracker 

de cette façon, je pouvais l'appeler pour générer un métaclasse pour une classe racine spécifique:

__metaclass__ = make_subtracker(Root) 

Voici où je tombe sur un problème. Je ne peux pas le faire:

class Root(object): 
    _registry = [] 
    __metaclass__ = make_subtracker(Root) 

... parce que Root ne définit pas encore quand je l'utilise make_subtracker(Root). J'ai essayé d'ajouter le métaclasse attribut plus tard, de sorte qu'au moins il peut être appliqué dans les sous-classes:

class Root(object): 
    _registry = [] 

Root.__metaclass__ = make_subtracker(Root) 

... mais cela ne fonctionne pas. métaclasse a un traitement spécial lorsque la définition de la classe est lu, tel que défini dans http://docs.python.org/reference/datamodel.html#customizing-class-creation

Je cherche des suggestions afin de le faire (soit changer une classe métaclasse lors de l'exécution d'une manière qu'il est appliqué à ses sous-classes, ou toute autre alternative).

Answer 1

Python fait automatiquement pour les classes de type nouveau, comme mentionné dans ce answer à la queston How can I find all subclasses of a given class in Python? similaire ici.

Answer 2

Je pense que vous voulez quelque chose comme ça (non testé):

class SubclassTracker(type): 
    def __init__(cls, name, bases, dct): 
     if not hasattr(cls, '_registry'): 
      cls._registry = [] 
     print('registering %s' % (name,)) 
     cls._registry.append(cls) 
     super(SubclassTracker, cls).__init__(name, bases, dct) 

Ensuite, pour Python 2, vous pouvez l'appeler comme:

class Root(object): 
    __metaclass__ = SubclassTracker 

pour Python 3

class Root(object, metaclass=SubclassTracker): 

Notez que vous n'avez pas besoin de coller l'attribut _registry là-bas parce que ce genre de choses est ce que metaclas ses sont pour. Puisque vous en avez déjà un autour ...;)

Notez également que vous pouvez déplacer le code d'enregistrement dans une clause else afin que la classe ne s'enregistre pas en tant que sous-classe.

Answer 3

est ici quelque chose que je jouais avec (qui fonctionne):