SelectionTool: asDomainQuery: New method

product/ERP5Form/Tool/SelectionTool.py

@@ -46,6 +46,7 @@ from urlparse import urlsplit, urlunsplit
 from zLOG import LOG, INFO, WARNING
 from Acquisition import aq_base
 from Products.ERP5Type.Message import translateString
+from Products.ZSQLCatalog.SQLCatalog import SimpleQuery, ComplexQuery
 import warnings
 
 
@@ -1332,6 +1333,36 @@ class SelectionTool( BaseTool, SimpleItem ):
         # Return the search dialog
         return getattr(o, dialog_id)(REQUEST=REQUEST)
 
+    security.declarePublic('asDomainSelection')
+    def asDomainQuery(self, domain, strict_membership=False):
+      if isinstance(domain, DomainSelection):
+        warnings.warn("To pass a DomainSelection instance is deprecated.\n"
+                      "Please use a domain dict instead.",
+                      DeprecationWarning)
+      else:
+        domain = DomainSelection(domain).__of__(self)
+      relation_dict = {}
+      query_list = []
+      append = query_list.append
+      domain_item_dict = domain.asDomainItemDict()
+      # XXX: why even put Nones in domain if they are ignored ?
+      domain_item_dict.pop(None, None)
+      for key, value in domain_item_dict.iteritems():
+        if getattr(aq_base(value), 'isPredicate', 0):
+          append(
+            value.asQuery(strict_membership=strict_membership),
+          )
+        else:
+          relation_dict[key] = [value]
+      for key, value in self.getPortalObject().portal_catalog.getCategoryValueDictParameterDict(
+        relation_dict,
+        strict_membership=strict_membership,
+      ).iteritems():
+        append(SimpleQuery(**{key: value}))
+      if query_list:
+        return ComplexQuery(query_list)
+      return SimpleQuery(uid=0, comparison_operator='>')
+
     security.declarePublic('buildSQLJoinExpressionFromDomainSelection')
     def buildSQLJoinExpressionFromDomainSelection(self, selection_domain,
                                                   domain_id=None,