fix: Respect ?preview for menu views

This is a hot fix for #69: Respect the ?preview flag in menu views by manually setting toolbar.preview_mode_active accordingly.

Author: metaforx

djangocms_rest/views.py

@@ -328,7 +328,7 @@ def get_menu_structure(
 
         request.api_endpoint = api_endpoint
         request.LANGUAGE_CODE = language
-        request.current_page = get_object(self.site, path)
+        request.current_page = get_object(self.site, path)  # Used to identify the current page in menus
         self.check_object_permissions(request, request.current_page)
         menu_renderer = menu_pool.get_renderer(request)
         menu_renderer.site = self.site

djangocms_rest/views_base.py

@@ -24,7 +24,8 @@
             )
         ]
     )
-except ImportError: # pragma: no cover
+except ImportError:  # pragma: no cover
+
     class OpenApiTypes:
         BOOL = "boolean"
 
@@ -40,11 +41,13 @@ def __init__(self, *args, **kwargs):
     def extend_schema(*_args, **_kwargs):  # pragma: no cover
         def _decorator(obj: T) -> T:
             return obj
+
         return _decorator
 
-    def preview_schema(obj: T) -> T: # pragma: no cover
+    def preview_schema(obj: T) -> T:  # pragma: no cover
         return obj
 
+
 @preview_schema
 class BaseAPIMixin:
     """
@@ -62,9 +65,12 @@ def site(self):
         return site if site is not None else get_current_site(self.request)
 
     def _preview_requested(self):
-        return "preview" in self.request.GET and self.request.GET.get(
-            "preview", ""
-        ).lower() not in ("0", "false")
+        preview_mode = "preview" in self.request.GET and self.request.GET.get("preview", "").lower() not in (
+            "0",
+            "false",
+        )
+        self.request.toolbar.preview_mode_active = preview_mode
+        return preview_mode
 
     @property
     def content_getter(self):