From a5eec911960e743929fb59ed94f0de40c023c582 Mon Sep 17 00:00:00 2001
From: Dhaval Mehta <20968146+dhaval-mehta@users.noreply.github.com>
Date: Sun, 9 Feb 2020 13:14:52 +0530
Subject: [PATCH] improve tag generation from viewset name

---
 rest_framework/schemas/openapi.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/rest_framework/schemas/openapi.py b/rest_framework/schemas/openapi.py
index d49a7e14e..8bdbdfe13 100644
--- a/rest_framework/schemas/openapi.py
+++ b/rest_framework/schemas/openapi.py
@@ -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.