From: Emilie <zawadzki@ircam.fr>
Date: Wed, 12 Oct 2016 07:54:10 +0000 (+0200)
Subject: Search: only page
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=dcc3085b0942e43c3032fc7483a228d8057d771c;p=mezzo.git

Search: only page
---

diff --git a/app/local_settings.py b/app/local_settings.py
index b1fb9bf7..e9de2f15 100644
--- a/app/local_settings.py
+++ b/app/local_settings.py
@@ -129,6 +129,7 @@ DASHBOARD_TAGS = ( ("mezzanine_tags.app_list",), (), ("mezzanine_tags.recent_act
 GRAPPELLI_ADMIN_TITLE = 'IRCAM Admin'
 
 SEARCH_MODEL_CHOICES = None # all objects
+SEARCH_MODEL_CHOICES = ('pages.Page',)
 SEARCH_PER_PAGE = 10
 MAX_PAGING_LINKS = 10
 
diff --git a/app/organization/core/views.py b/app/organization/core/views.py
index 0095baca..4813431c 100644
--- a/app/organization/core/views.py
+++ b/app/organization/core/views.py
@@ -53,12 +53,24 @@ class CustomSearchView(TemplateView):
         # count objects
         filter_dict = dict()
         for result in results:
-            if result.__class__.__name__ in filter_dict:
-                filter_dict[result.__class__.__name__]['count'] += 1
+            # print('---------------------------')
+            parent_class = result._meta.get_parent_list()[0]
+            print(parent_class.__class__.__name__)
+            # print(parent_class == "<class 'mezzanine.pages.models.Page'>")
+            # print('---------------------------')
+
+            if result._meta.get_parent_list() == 'mezzanine.pages.models.Page':
+                classname = 'Page'
+            else :
+                classname = result.__class__.__name__
+
+            if classname in filter_dict:
+                filter_dict[classname]['count'] += 1
             else:
-                filter_dict[result.__class__.__name__] = {'count' : 1}
-                filter_dict[result.__class__.__name__].update({'app_label' : result._meta.app_label})
+                filter_dict[classname] = {'count' : 1}
+                filter_dict[classname].update({'app_label' : result._meta.app_label})
 
+        print(filter_dict)
         # get url param
         current_query = QueryDict(mutable=True)
         current_query = request.GET.copy()