1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Fix handling of ProtectedError exceptions

This commit is contained in:
Jeremy Stretch
2020-07-16 13:45:02 -04:00
parent bdf41451eb
commit 1dbf776279
2 changed files with 9 additions and 24 deletions

View File

@ -7,31 +7,17 @@ def handle_protectederror(obj, request, e):
"""
Generate a user-friendly error message in response to a ProtectedError exception.
"""
try:
dep_class = e.protected_objects[0]._meta.verbose_name_plural
except IndexError:
raise e
# Grammar for single versus multiple triggering objects
if type(obj) in (list, tuple):
err_message = "Unable to delete the requested {}. The following dependent {} were found: ".format(
obj[0]._meta.verbose_name_plural,
dep_class,
)
else:
err_message = "Unable to delete {} {}. The following dependent {} were found: ".format(
obj._meta.verbose_name,
obj,
dep_class,
)
protected_objects = list(e.protected_objects)
err_message = f"Unable to delete {obj._meta.verbose_name} <strong>{obj}</strong>. " \
f"{len(protected_objects)} dependent objects were found: "
# Append dependent objects to error message
dependent_objects = []
for obj in e.protected_objects:
for dependent in protected_objects:
if hasattr(obj, 'get_absolute_url'):
dependent_objects.append('<a href="{}">{}</a>'.format(obj.get_absolute_url(), escape(obj)))
dependent_objects.append(f'<a href="{dependent.get_absolute_url()}">{escape(dependent)}</a>')
else:
dependent_objects.append(str(obj))
dependent_objects.append(str(dependent))
err_message += ', '.join(dependent_objects)
messages.error(request, mark_safe(err_message))