fixed model name

This commit is contained in:
Abhimanyu Saharan 2023-04-10 22:06:35 +05:30
parent be81143a26
commit 6c43d87d22

View File

@ -255,10 +255,8 @@ class NetBoxAutoSchema(AutoSchema):
else:
is_list = True
description = self.view.__class__.__name__
if description.endswith('ViewSet'):
description = description[:-len('ViewSet')]
model_name = self.view.queryset.model._meta.verbose_name
if is_list:
return f"{self.method.capitalize()} a list of {description} objects."
return f"{self.method.capitalize()} a {description} object."
return f"{self.method.capitalize()} a list of {model_name} objects."
return f"{self.method.capitalize()} a {model_name} object."