wip

Author: jobselko

pulp_python/app/pypi/views.py

@@ -3,7 +3,7 @@
 
 from aiohttp.client_exceptions import ClientError
 from rest_framework.viewsets import ViewSet
-from rest_framework.renderers import JSONRenderer, TemplateHTMLRenderer
+from rest_framework.renderers import BrowsableAPIRenderer, JSONRenderer, TemplateHTMLRenderer
 from rest_framework.response import Response
 from django.core.exceptions import ObjectDoesNotExist
 from django.shortcuts import redirect
@@ -18,7 +18,6 @@
     HttpResponseBadRequest,
     StreamingHttpResponse,
     HttpResponse,
-    JsonResponse,
 )
 from drf_spectacular.utils import extend_schema
 from dynaconf import settings
@@ -263,32 +262,34 @@ class SimpleView(PackageUploadMixin, ViewSet):
         ],
     }
 
-    renderer_classes = [
-        TemplateHTMLRenderer,
-        PyPISimpleHTMLRenderer,
-        PyPISimpleJSONRenderer,
-    ]
+    def get_renderers(self):
+        """Defines custom renderers."""
+        if self.action in ["list", "retrieve"]:
+            # Use PyPI renderers for simple API endpoints
+            return [
+                TemplateHTMLRenderer(),
+                PyPISimpleHTMLRenderer(),
+                PyPISimpleJSONRenderer(),
+            ]
+        else:
+            # Use standard DRF renderers
+            return [JSONRenderer(), BrowsableAPIRenderer()]
 
     @extend_schema(summary="Get index simple page")
     def list(self, request, path):
         """Gets the simple api html page for the index."""
-        content_type = _select_content_type(request)
-        if content_type is None:
-            return HttpResponse("Not Acceptable Content-Type", status=406)
-
         repo_version, content = self.get_rvc()
         if self.should_redirect(repo_version=repo_version):
             return redirect(urljoin(self.base_content_url, f"{path}/simple/"))
-        names = content.order_by("name").values_list("name", flat=True).distinct().iterator()
 
-        if content_type == PYPI_SIMPLE_V1_JSON:
-            names_list = list(names)
-            data_dict = write_simple_index_json(names_list)
+        names = content.order_by("name").values_list("name", flat=True).distinct()
+
+        if request.accepted_renderer.media_type == PYPI_SIMPLE_V1_JSON:
+            data_dict = write_simple_index_json(list(names))
             headers = {"X-PyPI-Last-Serial": str(PYPI_SERIAL_CONSTANT)}
-            response = JsonResponse(data_dict, content_type=content_type, headers=headers)
-            return response
+            return Response(data_dict, headers=headers)
         else:
-            return StreamingHttpResponse(write_simple_index(names, streamed=True))
+            return StreamingHttpResponse(write_simple_index(names.iterator(), streamed=True))
 
     def pull_through_package_simple(self, package, path, remote):
         """Gets the package's simple page from remote."""
@@ -327,10 +328,6 @@ def parse_package(release_package):
     @extend_schema(operation_id="pypi_simple_package_read", summary="Get package simple page")
     def retrieve(self, request, path, package):
         """Retrieves the simple api html page for a package."""
-        content_type = _select_content_type(request)
-        if content_type is None:
-            return HttpResponse("Not Acceptable Content-Type", status=406)
-
         repo_ver, content = self.get_rvc()
         # Should I redirect if the normalized name is different?
         normalized = canonicalize_name(package)
@@ -352,12 +349,11 @@ def retrieve(self, request, path, package):
             name = present[2]
         releases = ((f, urljoin(self.base_content_url, f"{path}/{f}"), d) for f, d, _ in packages)
 
-        if content_type == PYPI_SIMPLE_V1_JSON:
+        if request.accepted_renderer.media_type == PYPI_SIMPLE_V1_JSON:
             releases_list = list(releases)
             data_dict = write_simple_detail_json(name, releases_list)
             headers = {"X-PyPI-Last-Serial": str(PYPI_SERIAL_CONSTANT)}
-            response = JsonResponse(data_dict, content_type=content_type, headers=headers)
-            return response
+            return Response(data_dict, headers=headers)
         else:
             return StreamingHttpResponse(write_simple_detail(name, releases, streamed=True))