Django: Como posso encontrar uma lista de modelos que o ORM conhece?

86

No Django, há um lugar onde posso obter uma lista ou pesquisar os modelos que o ORM conhece?

interstar
fonte

Respostas:

174

Solução simples:

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

Por padrão apps.get_models(), não inclui

  • modelos criados automaticamente para relações muitos-para-muitos sem uma tabela intermediária explícita
  • modelos que foram trocados.

Se você quiser incluí-los também,

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

Antes do Django 1.7, use:

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

O include_auto_createdparâmetro garante que as tabelas criadas implicitamente por ManyToManyFields também sejam recuperadas.

Daniel Roseman
fonte
2
Usando o mesmo método, você também pode obter todos os modelos em apenas um aplicativo: stackoverflow.com/a/8702854/117268
Emil Stenström
from django.apps.apps import get_modelsproduz ImportError: No module named 'django.apps.apps'... alguma ideia?
aljabear
3
from django.apps import apps>>apps.get_models
Dingo
11

Liste modelos usando 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
fonte
7

Se você quiser um dicionário com todos os modelos, pode usar:

from django.apps import apps

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

Se você quiser brincar e não usar a boa solução , pode brincar um pouco com a introspecção 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

Nota: este é um pedaço de código bastante grosseiro; ele assumirá que todos os modelos estão definidos em "models.py" e que eles herdam de django.db.models.Model.

roubar
fonte
0

Se você registrar seus modelos com o aplicativo de administração, poderá ver todos os atributos dessas classes na documentação do administrador.

Bardo Técnico
fonte
0

Esta é uma maneira simples de localizar e excluir quaisquer permissões que existam no banco de dados, mas não existam nas definições do modelo 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
fonte