search.py

# -*- coding: UTF-8 -*-
# Copyright (C) 2009 Sylvain Taverne <sylvain@itaapy.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# Import from itools
from itools.datatypes import Unicode, Enumerate
from itools.gettext import MSG
from itools.handlers import checkid
from itools.xapian import OrQuery, PhraseQuery, AndQuery, split_unicode
from itools.xapian import RangeQuery
from itools.web import get_context

# Import from ikaaro
from ikaaro.folder import Folder
from ikaaro.registry import register_resource_class

# Import from itws
from itws.bar import SideBarAware

# Import from shop
from categories_views import Category_View


class Shop_CategoriesEnumerate(Enumerate):

    @classmethod
    def get_options(cls):
        options = [{'name': '*', 'value': MSG(u'The whole site')}]
        context = get_context()
        resource = context.resource
        root = context.root
        site_root = resource.get_site_root()
        categories = site_root.get_resource('categories')
        query = [PhraseQuery('format', 'category'),
                 PhraseQuery('parent_path', str(categories.get_abspath()))]
        for brain in root.search(AndQuery(*query)).get_documents():
            categorie = root.get_resource(brain.abspath)
            nb_products = categorie.get_nb_products(only_public=True)
            if nb_products == 0:
                continue
            small_title = categorie.get_property('breadcrumb_title')
            options.append({'name': brain.abspath,
                            'value': small_title or categorie.get_title()})
        return options



class Shop_ProductSearch(Category_View):

    access = True
    title = MSG(u'Search')

    search_schema = {
        'product_search_text': Unicode,
        'category': Shop_CategoriesEnumerate(default='*'),
    }

    context_menus = []

    nb_results = 0


    def get_sub_categories_namespace(self, resource, context):
        namespace = []
        search_word = context.query['product_search_text']
        query = AndQuery(PhraseQuery('format', 'category'),
                         PhraseQuery('title', search_word))
        for brain in context.root.search(query).get_documents():
            cat = context.root.get_resource(brain.abspath)
            nb_products = cat.get_nb_products()
            if nb_products == 0:
                continue
            img = cat.get_property('image_category')
            path_cat = resource.get_pathto(cat)
            namespace.append(
                {'name': cat.name,
                 'link': context.get_link(cat),
                 'title': cat.get_title(),
                 'breadcrumb_title': cat.get_property('breadcrumb_title'),
                 'css': None,
                 'nb_products': nb_products,
                 'img': str(path_cat.resolve2(img)) if img else None})
        if namespace:
            namespace[0]['css'] = 'start'
            namespace[-1]['css'] = 'end'
        return namespace



    def get_items(self, resource, context):
        site_root = resource.get_site_root()
        shop = site_root.get_resource('shop')
        abspath = site_root.get_canonical_path()
        query = [PhraseQuery('parent_paths', str(abspath)),
                 PhraseQuery('format', shop.product_class.class_id),
                 PhraseQuery('workflow_state', 'public')]
        category = context.query['category']
        if category and category != '*':
            query.append(PhraseQuery('parent_paths', category))
        search_word = context.query['product_search_text']
        if search_word:
            for word in split_unicode(search_word):
                alternative = (word + u"s" if not word.endswith(u's')
                                            else word[:-1])
                plain_text = OrQuery(
                                # By reference
                                #PhraseQuery('reference', word.upper()), #encode('utf-8')), #.upper()),
                                # On product
                                PhraseQuery('title', word),
                                PhraseQuery('description', word),
                                PhraseQuery('data', word),
                                PhraseQuery('text', word),
                                # XXX Hack manufacturer
                                #PhraseQuery('manufacturer', checkid(word)),
                                # Alternative
                                PhraseQuery('title', alternative),
                                PhraseQuery('description', alternative),
                                PhraseQuery('data', alternative),
                                PhraseQuery('text', alternative))
                query.append(plain_text)
        # Add query of filter
        for key, datatype in self.get_query_schema().items():
            value = context.query[key]
            if key == 'stored_price' and value:
                query.append(RangeQuery('stored_price', value[0], value[1]))
            # TODO Add other filters
        results = context.root.search(AndQuery(*query))
        # XXX Hack results
        self.nb_results = len(results)
        return results



class ShopSearch(SideBarAware, Folder):

    class_id = 'shop-search'
    class_title = MSG(u'Rechercher')
    class_views = ['view', 'edit'] + SideBarAware.class_views

    view = Shop_ProductSearch()


    @staticmethod
    def _make_resource(cls, folder, name, *args,  **kw):
        Folder._make_resource(cls, folder, name, **kw)
        _cls = SideBarAware
        _cls._make_resource(_cls, folder, name, **kw)


register_resource_class(ShopSearch)