altf4arnold/datatracker
1
0
Fork 0
Code Issues Pull requests Actions3 Packages Projects Releases Wiki Activity

Updated the referential integrity checking comand.

Browse source 
- Legacy-Id: 13234
This commit is contained in:
Henrik Levkowetz 2017-04-18 20:21:58 +00:00
parent 8e8d2c359f
commit 0b730c9554
1 changed files with 17 additions and 9 deletions

26
ietf/utils/management/commands/check_referential_integrity.py
View file

@ -2,6 +2,7 @@
import django
django.setup()
from django.apps import apps
from django.core.management.base import BaseCommand #, CommandError
from django.core.exceptions import FieldError
from django.db import models
@ -13,11 +14,14 @@ class Command(BaseCommand):
help = "Check all models for referential integrity."
def handle(self, *args, **options):
verbosity = options.get("verbosity", "1")
verbosity = int(verbosity) if verbosity.isdigit() else 1
verbosity = options.get("verbosity", 1)
def check_field(field):
foreign_model = field.related.parent_model
try:
foreign_model = field.related_model
except Exception:
debug.pprint('dir(field)')
raise
if verbosity > 1:
print " %s -> %s.%s" % (field.name,foreign_model.__module__,foreign_model.__name__),
used = set(field.model.objects.values_list(field.name,flat=True))
@ -33,11 +37,15 @@ class Command(BaseCommand):
print "%s.%s.%s -> %s.%s Bad key values:" % (model.__module__,model.__name__,field.name,foreign_model.__module__,foreign_model.__name__),list(used - exists)
def check_reverse_field(field):
foreign_model = field.related.parent_model
try:
foreign_model = field.related_model
except Exception:
debug.pprint('dir(field)')
raise
if foreign_model == field.model:
return
foreign_field_name = field.related.var_name
foreign_accessor_name = field.related.get_accessor_name()
foreign_field_name = field.rel.name
foreign_accessor_name = field.rel.get_accessor_name()
if verbosity > 1:
print " %s <- %s -> %s.%s" % (field.model.__name__, field.rel.through._meta.db_table, foreign_model.__module__, foreign_model.__name__),
try:
@ -58,10 +66,10 @@ class Command(BaseCommand):
if used - exists:
print "%s.%s <- %s -> %s.%s ** Bad key values:\n " % (field.model.__module__, field.model.__name__, field.rel.through._meta.db_table, foreign_model.__module__, foreign_model.__name__), list(used - exists)
for app in [a for a in models.get_apps() if a.__name__.startswith('ietf.')]:
for conf in [ c for c in apps.get_app_configs() if c.name.startswith('ietf.')]:
if verbosity > 1:
print "Checking",app.__name__
for model in models.get_models(app):
print "Checking", conf.name
for model in conf.get_models():
if model._meta.proxy:
continue
if verbosity > 1: