Comment puis-je obtenir une liste de toutes les classes du module actuel en Python ?

Question

J'ai vu beaucoup d'exemples de personnes qui extraient toutes les classes d'un module, généralement quelque chose comme :

# foo.py
class Foo:
    pass

# test.py
import inspect
import foo

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

Génial.

Mais je n'arrive pas à trouver comment récupérer toutes les classes de la base de données. actuel module.

# 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()

C'est probablement quelque chose de très évident, mais je n'ai rien trouvé. Quelqu'un peut-il m'aider ?

Answer 1

Essayez ça :

import sys
current_module = sys.modules[__name__]

Dans votre contexte :

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

Et encore mieux :

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

Parce que inspect.getmembers() prend un prédicat.

Answer 2

Je ne sais pas s'il y a une façon "correcte" de le faire, mais votre extrait est sur la bonne voie : ajoutez simplement import foo dans foo.py, faites inspect.getmembers(foo) et cela devrait fonctionner correctement.

Answer 3

Qu'en est-il

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

?

Answer 4

import pyclbr
print(pyclbr.readmodule(__name__).keys())

Notez que le module de navigation des classes Python de la stdlib utilise l'analyse statique des sources, et ne fonctionne donc que pour les modules qui sont soutenus par un réel .py fichier.

Answer 5

Il y avait un [PEP] pour une fonctionnalité de ce type, mais il a été rejeté.