Django: Comment puis-je trouver une liste de modèles que l'ORM connaît?

86

Dans Django, y a-t-il un endroit où je peux obtenir une liste ou rechercher les modèles que l'ORM connaît?

interstar
la source

Réponses:

174

Solution simple:

import django.apps
django.apps.apps.get_models()

Par défaut, apps.get_models()ne pas inclure

  • modèles créés automatiquement pour les relations plusieurs à plusieurs sans table intermédiaire explicite
  • modèles qui ont été échangés.

Si vous souhaitez les inclure également,

django.apps.apps.get_models(include_auto_created=True, include_swapped=True)

Avant Django 1.7, utilisez à la place:

from django.db import models
models.get_models(include_auto_created=True)

Le include_auto_createdparamètre garantit que les tables créées implicitement par ManyToManyFields seront également récupérées.

Daniel Roseman
la source
2
En utilisant la même méthode, vous pouvez également obtenir tous les modèles dans une seule application: stackoverflow.com/a/8702854/117268
Emil Stenström
from django.apps.apps import get_modelsproduit ImportError: No module named 'django.apps.apps'... une idée?
aljabear
3
from django.apps import apps>>apps.get_models
Dingo
11

Liste des modèles à l'aide de http://docs.djangoproject.com/en/dev/ref/contrib/contenttypes/

from django.contrib.contenttypes.models import ContentType

for ct in ContentType.objects.all():
    m = ct.model_class()
    print "%s.%s\t%d" % (m.__module__, m.__name__, m._default_manager.count())
andreipak
la source
7

Si vous voulez un dictionnaire avec tous les modèles, vous pouvez utiliser:

from django.apps import apps

models = {
    model.__name__: model for model in apps.get_models()
}
Myzel394
la source
4

Si vous voulez jouer, et ne pas utiliser la bonne solution , vous pouvez jouer un peu avec l'introspection python:

import settings
from django.db import models

for app in settings.INSTALLED_APPS:
  models_name = app + ".models"
  try:
    models_module = __import__(models_name, fromlist=["models"])
    attributes = dir(models_module)
    for attr in attributes:
      try:
        attrib = models_module.__getattribute__(attr)
        if issubclass(attrib, models.Model) and attrib.__module__== models_name:
          print "%s.%s" % (models_name, attr)
      except TypeError, e:
        pass
  except ImportError, e:
    pass

Remarque: il s'agit d'un morceau de code assez grossier; il supposera que tous les modèles sont définis dans "models.py" et qu'ils héritent de django.db.models.Model.

Rob
la source
0

Si vous enregistrez vos modèles avec l'application d'administration, vous pouvez voir tous les attributs de ces classes dans la documentation d'administration.

Barde technique
la source
0

Voici un moyen simple de rechercher et de supprimer toutes les autorisations qui existent dans la base de données mais qui n'existent pas dans les définitions de modèle ORM:

from django.apps import apps
from django.contrib.auth.management import _get_all_permissions
from django.contrib.auth.models import Permission
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    def handle(self, *args, **options):
        builtins = []
        for klass in apps.get_models():
            for perm in _get_all_permissions(klass._meta):
                builtins.append(perm[0])
        builtins = set(builtins)

        permissions = set(Permission.objects.all().values_list('codename', flat=True))
        to_remove = permissions - builtins
        res = Permission.objects.filter(codename__in=to_remove).delete()
        self.stdout.write('Deleted records: ' + str(res))
Aidan
la source