From 6c43d87d220ba623a5f8093e291f60c40d7d3db6 Mon Sep 17 00:00:00 2001 From: Abhimanyu Saharan Date: Mon, 10 Apr 2023 22:06:35 +0530 Subject: [PATCH] fixed model name --- netbox/core/api/schema.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/netbox/core/api/schema.py b/netbox/core/api/schema.py index cada59c64..ed10c6cf5 100644 --- a/netbox/core/api/schema.py +++ b/netbox/core/api/schema.py @@ -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."