swagger ui own tag implemented

Author: PRITAM CHAKRABORTY

blogapp/swagger_schema.py

@@ -0,0 +1,12 @@
+from drf_yasg.inspectors import SwaggerAutoSchema
+
+
+class CustomAutoSchema(SwaggerAutoSchema):
+
+    def get_tags(self, operation_keys=None):
+        tags = self.overrides.get('tags', None) or getattr(
+            self.view, 'my_tags', [])
+        if not tags:
+            tags = [operation_keys[0]]
+
+        return tags

blogapp/views.py

@@ -46,6 +46,7 @@ class BloggerViewSet(ModelViewSet):
     permission_classes = [IsAdminUser | (IsAuthenticated & IsSelf)]
     queryset = Blogger.objects.all()
     serializer_class = BloggerAdminSerializer
+    my_tags = ["Blogger"]
 
     def get_permissions(self):
         method = self.request.method
@@ -71,6 +72,7 @@ class OwnBlogViewSet(ModelViewSet):
     ordering_fields = ["title", "created_at"]
     serializer_class = BlogReadSerializer
     permission_classes = [IsAdminUser | (IsAuthenticated & IsBlogOwner)]
+    my_tags = ["Blogger-Blog"]
     # queryset = Blog.objects.prefetch_related(
     #     'comments').select_related('creator').all()
 
@@ -104,10 +106,12 @@ class AllBlogVSet(ListModelMixin, GenericViewSet):
     queryset = Blog.objects.select_related(
         'creator').prefetch_related('comments').all()
     serializer_class = BlogReadSerializer
+    my_tags = ["All Blogs"]
 
 
 class CommentVSet(ModelViewSet):
     http_method_names = ["get", "post", "patch", "delete", "option", "head"]
+    my_tags = ["Blog-Comments"]
 
     def get_permissions(self):
         if self.request.method == 'GET':

core/settings/common.py

@@ -62,6 +62,9 @@
 
 ROOT_URLCONF = 'core.urls'
 
+
+SWAGGER_SETTINGS = {"DEFAULT_AUTO_SCHEMA_CLASS": "blogapp.swagger_schema.CustomAutoSchema"}
+
 TEMPLATES = [
     {
         'BACKEND': 'django.template.backends.django.DjangoTemplates',