Update views.py as per comments

FidelityInternational · Mar 6, 2024 · 804242b · 804242b
1 parent 9c85627
commit 804242b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/djangocms_navigation/views.py b/djangocms_navigation/views.py
@@ -9,6 +9,7 @@
 from cms.utils import get_current_site, get_language_from_request
 
 from djangocms_navigation.utils import is_model_supported, supported_models
+from djangocms_versioning.constants import ARCHIVED, UNPUBLISHED
 
 
 class ContentObjectSelect2View(View):
@@ -38,8 +39,7 @@ def get(self, request, *args, **kwargs):
         if model == Page:
             queryset_data = [
                 page for page in queryset_data
-                if getattr(page.get_title_obj().versions.first(), "state", None)
-                != 'unpublished'
+                if not getattr(page.get_title_obj().versions.first(), "state", None) in [ ARCHIVED, UNPUBLISHED ]
             ]
 
         data = {