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?

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.


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


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

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.



0

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


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))
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.