improve tag generation from viewset name

This commit is contained in:
Dhaval Mehta 2020-02-09 13:14:52 +05:30
parent 44c1c25bde
commit a5eec91196

View File

@ -590,12 +590,14 @@ class AutoSchema(ViewInspector):
return self._tags
# Extract tag from viewset name
# UserViewSet tags = [User]
# UserView tags = [User]
# User tags = [User]
if hasattr(self.view, 'action'):
name = self.view.__class__.__name__
if name.lower().endswith('viewset'):
name = name[:-7] # remove trailing `viewset` from name
if name.endswith('APIView') or name.endswith('ViewSet'):
name = name[:-7]
elif name.endswith('View'):
name = name[:-4]
return [name]
# First element of a specific path could be valid tag. This is a fallback solution.