implemented async flatpages

Author: amirreza8002

django_async_extensions/acontrib/flatpages/__init__.py

@@ -0,0 +1,11 @@
+from django.urls import path
+
+from django_async_extensions.acontrib.flatpages import views
+
+urlpatterns = [
+    path(
+        "<path:url>",
+        views.flatpage,
+        name="django_async_extensions.acontrib.flatpages.views.flatpage",
+    ),
+]

django_async_extensions/acontrib/flatpages/urls.py

@@ -0,0 +1,74 @@
+from asgiref.sync import sync_to_async
+from django.conf import settings
+from django.contrib.sites.shortcuts import get_current_site
+from django.http import Http404, HttpResponse, HttpResponsePermanentRedirect
+from django.shortcuts import aget_object_or_404
+from django.template import loader
+from django.utils.safestring import mark_safe
+from django.views.decorators.csrf import csrf_protect
+
+from django_async_extensions.acontrib.flatpages.models import AsyncFlatPage
+
+DEFAULT_TEMPLATE = "flatpages/default.html"
+
+# This view is called from AsyncFlatpageFallbackMiddleware.process_response
+# when a 404 is raised, which often means CsrfViewMiddleware.process_view
+# has not been called even if CsrfViewMiddleware is installed. So we need
+# to use @csrf_protect, in case the template needs {% csrf_token %}.
+# However, we can't just wrap this view; if no matching flatpage exists,
+# or a redirect is required for authentication, the 404 needs to be returned
+# without any CSRF checks. Therefore, we only
+# CSRF protect the internal implementation.
+
+
+async def flatpage(request, url):
+    """
+    Public interface to the flat page view.
+
+    Models: `flatpages.flatpages`
+    Templates: Uses the template defined by the ``template_name`` field,
+        or :template:`flatpages/default.html` if template_name is not defined.
+    Context:
+        flatpage
+            `flatpages.flatpages` object
+    """
+    if not url.startswith("/"):
+        url = "/" + url
+    site = await sync_to_async(get_current_site)(request)
+    site_id = site.id
+    try:
+        f = await aget_object_or_404(AsyncFlatPage, url=url, sites=site_id)
+    except Http404:
+        if not url.endswith("/") and settings.APPEND_SLASH:
+            url += "/"
+            f = await aget_object_or_404(AsyncFlatPage, url=url, sites=site_id)
+            return HttpResponsePermanentRedirect("%s/" % request.path)
+        else:
+            raise
+    return await render_flatpage(request, f)
+
+
+@csrf_protect
+async def render_flatpage(request, f):
+    """
+    Internal interface to the flat page view.
+    """
+    # If registration is required for accessing this page, and the user isn't
+    # logged in, redirect to the login page.
+    user = await request.auser()
+    if f.registration_required and not user.is_authenticated:
+        from django.contrib.auth.views import redirect_to_login
+
+        return redirect_to_login(request.path)
+    if f.template_name:
+        template = loader.select_template((f.template_name, DEFAULT_TEMPLATE))
+    else:
+        template = loader.get_template(DEFAULT_TEMPLATE)
+
+    # To avoid having to always use the "|safe" filter in flatpage templates,
+    # mark the title and content as already safe (since they are raw HTML
+    # content in the first place).
+    f.title = mark_safe(f.title)  # noqa: S308
+    f.content = mark_safe(f.content)  # noqa: S308
+
+    return HttpResponse(template.render({"flatpage": f}, request))