2016-03-01 11:23:03 -05:00
|
|
|
from django.contrib import messages
|
2023-11-01 13:47:14 -04:00
|
|
|
from django.db.models import ProtectedError, RestrictedError
|
2017-05-02 11:43:11 -04:00
|
|
|
from django.utils.html import escape
|
|
|
|
from django.utils.safestring import mark_safe
|
2023-11-01 13:47:14 -04:00
|
|
|
from django.utils.translation import gettext_lazy as _
|
2016-03-01 11:23:03 -05:00
|
|
|
|
|
|
|
|
2020-08-20 09:44:45 -04:00
|
|
|
def handle_protectederror(obj_list, request, e):
|
2016-03-01 11:23:03 -05:00
|
|
|
"""
|
2023-11-01 13:47:14 -04:00
|
|
|
Generate a user-friendly error message in response to a ProtectedError or RestrictedError exception.
|
2016-03-01 11:23:03 -05:00
|
|
|
"""
|
2023-11-01 13:47:14 -04:00
|
|
|
if type(e) is ProtectedError:
|
|
|
|
protected_objects = list(e.protected_objects)
|
|
|
|
elif type(e) is RestrictedError:
|
|
|
|
protected_objects = list(e.restricted_objects)
|
|
|
|
else:
|
|
|
|
raise e
|
|
|
|
|
|
|
|
# Formulate the error message
|
|
|
|
err_message = _("Unable to delete <strong>{objects}</strong>. {count} dependent objects were found: ").format(
|
|
|
|
objects=', '.join(str(obj) for obj in obj_list),
|
|
|
|
count=len(protected_objects) if len(protected_objects) <= 50 else _('More than 50')
|
|
|
|
)
|
2016-05-12 16:10:55 -04:00
|
|
|
|
|
|
|
# Append dependent objects to error message
|
|
|
|
dependent_objects = []
|
2020-08-20 09:44:45 -04:00
|
|
|
for dependent in protected_objects[:50]:
|
|
|
|
if hasattr(dependent, 'get_absolute_url'):
|
2020-07-16 13:45:02 -04:00
|
|
|
dependent_objects.append(f'<a href="{dependent.get_absolute_url()}">{escape(dependent)}</a>')
|
2016-05-12 16:10:55 -04:00
|
|
|
else:
|
2020-07-16 13:45:02 -04:00
|
|
|
dependent_objects.append(str(dependent))
|
2017-05-24 11:33:11 -04:00
|
|
|
err_message += ', '.join(dependent_objects)
|
2016-05-12 16:10:55 -04:00
|
|
|
|
2017-05-02 11:43:11 -04:00
|
|
|
messages.error(request, mark_safe(err_message))
|