diff options
| author | Erik Romijn <erik@erik.io> | 2016-07-03 15:55:14 +0200 |
|---|---|---|
| committer | Florian Apolloner <apollo13@users.noreply.github.com> | 2016-07-03 15:55:14 +0200 |
| commit | 8db889eaf7dce0cb715b075be32047c1b1b316da (patch) | |
| tree | 13a52c3c21615067a22906a00b829bd70df951d1 /django | |
| parent | f7e00b40c558967d697529b7ed55c5fb82396e6d (diff) | |
Fixed #18682 -- Expanded explanation in stale content type deletion. (#6869)
Diffstat (limited to 'django')
| -rw-r--r-- | django/contrib/contenttypes/management.py | 46 |
1 files changed, 37 insertions, 9 deletions
diff --git a/django/contrib/contenttypes/management.py b/django/contrib/contenttypes/management.py index 0bdf8ed9fa..591f96705b 100644 --- a/django/contrib/contenttypes/management.py +++ b/django/contrib/contenttypes/management.py @@ -1,5 +1,6 @@ from django.apps import apps as global_apps from django.db import DEFAULT_DB_ALIAS, migrations, router, transaction +from django.db.models.deletion import Collector from django.db.utils import IntegrityError from django.utils import six from django.utils.six.moves import input @@ -141,21 +142,39 @@ def update_contenttypes(app_config, verbosity=2, interactive=True, using=DEFAULT print("Adding content type '%s | %s'" % (ct.app_label, ct.model)) # Confirm that the content type is stale before deletion. + using = router.db_for_write(ContentType) if to_remove: if interactive: - content_type_display = '\n'.join( - ' %s | %s' % (ct.app_label, ct.model) - for ct in to_remove - ) - ok_to_delete = input("""The following content types are stale and need to be deleted: + ct_info = [] + for ct in to_remove: + ct_info.append(' - Content type for %s.%s' % (ct.app_label, ct.model)) + collector = NoFastDeleteCollector(using=using) + collector.collect([ct]) + + for obj_type, objs in collector.data.items(): + if objs == {ct}: + continue + ct_info.append(' - %s object%s of type %s.%s:' % ( + len(objs), + 's' if len(objs) != 1 else '', + obj_type._meta.app_label, + obj_type._meta.model_name) + ) + + content_type_display = '\n'.join(ct_info) + print("""Some content types in your database are stale and can be deleted. +Any objects that depend on these content types will then also be deleted. +The content types, and the dependent objects that would be deleted, are: %s -Any objects related to these content types by a foreign key will also -be deleted. Are you sure you want to delete these content types? -If you're unsure, answer 'no'. +This list does not include data that might be in your database +outside of Django's models. - Type 'yes' to continue, or 'no' to cancel: """ % content_type_display) +Are you sure you want to delete these content types? +If you're unsure, answer 'no'. + """ % content_type_display) + ok_to_delete = input("Type 'yes' to continue, or 'no' to cancel: ") else: ok_to_delete = False @@ -167,3 +186,12 @@ If you're unsure, answer 'no'. else: if verbosity >= 2: print("Stale content types remain.") + + +class NoFastDeleteCollector(Collector): + def can_fast_delete(self, *args, **kwargs): + """ + We always want to load the objects into memory so that we can display + them to the user when asking confirmation. + """ + return False |
