src/pyams_content/root/zmi/search.py
changeset 1395 8184dadefc8b
parent 1189 26b86361e20b
child 1407 85e615286c7a
--- a/src/pyams_content/root/zmi/search.py	Wed Jul 01 18:05:38 2020 +0200
+++ b/src/pyams_content/root/zmi/search.py	Fri Jul 03 12:27:49 2020 +0200
@@ -23,6 +23,7 @@
 from zope.schema import Choice, Datetime, List
 
 from pyams_catalog.query import CatalogResultSet
+from pyams_content.component.theme import ICollectionsManager
 from pyams_content.component.theme.interfaces import ITagsManager
 from pyams_content.profile.interfaces import IAdminProfile
 from pyams_content.root import ISiteRoot
@@ -177,6 +178,9 @@
     tags = ThesaurusTermsListField(title=_("Tags"),
                                    required=False)
 
+    collections = ThesaurusTermsListField(title=_("Collections"),
+                                          required=False)
+
 
 @template_config(template='templates/advanced-search.pt', layer=IPyAMSLayer)
 @implementer(IInnerPage)
@@ -192,6 +196,7 @@
 
     fields = field.Fields(ISiteRootAdvancedSearchFields)
     fields['tags'].widgetFactory = ThesaurusTermsTreeFieldWidget
+    fields['collections'].widgetFactory = ThesaurusTermsTreeFieldWidget
     ajax_handler = 'advanced-search-results.html'
 
     def updateWidgets(self, prefix=None):
@@ -201,6 +206,11 @@
             manager = ITagsManager(self.request.root)
             widget.thesaurus_name = manager.thesaurus_name
             widget.extract_name = manager.extract_name
+        if 'collections' in self.widgets:
+            widget = self.widgets['collections']
+            manager = ICollectionsManager(self.request.root)
+            widget.thesaurus_name = manager.thesaurus_name
+            widget.extract_name = manager.extract_name
 
 
 @adapter_config(context=(ISiteRoot, IPyAMSLayer, SiteRootAdvancedSearchForm), provides=IContentSearch)
@@ -223,14 +233,16 @@
             else:
                 query_params = Eq(catalog['oid'], sequence.get_full_oid(query))
                 negotiator = get_utility(INegotiator)
-                for lang in {self.request.registry.settings.get('pyramid.default_locale_name', 'en'),
+                for lang in {self.request.registry.settings.get('pyramid.default_locale_name',
+                                                                'en'),
                              self.request.locale_name,
                              negotiator.server_language} | negotiator.offered_languages:
                     index_name = 'title:{0}'.format(lang)
                     if index_name in catalog:
                         index = catalog[index_name]
                         if index.check_query(query):
-                            query_params |= Contains(index, ' and '.join((w+'*' for w in query.split())))
+                            query_params |= Contains(index, ' and '.join(
+                                (w + '*' for w in query.split())))
                 params &= query_params
         if data.get('owner'):
             params &= Eq(catalog['role:owner'], data['owner'])
@@ -245,6 +257,9 @@
         if data.get('tags'):
             tags = [intids.register(term) for term in data['tags']]
             params &= Any(catalog['tags'], tags)
+        if data.get('collections'):
+            tags = [intids.register(term) for term in data['collections']]
+            params &= Any(catalog['collections'], tags)
         return unique(map(get_last_version,
                           CatalogResultSet(CatalogQuery(catalog).query(params,
                                                                        sort_index='modified_date',