# -*- coding: utf-8 -*- # Copyright (C) 2011-2017 RhodeCode GmbH # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License, version 3 # (only), as published by the Free Software Foundation. # # 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 Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. # # This program is dual-licensed. If you wish to learn more about the # RhodeCode Enterprise Edition, including its added features, Support services, # and proprietary license terms, please see https://rhodecode.com/licenses/ import logging import urllib from pyramid.view import view_config from webhelpers.util import update_params from rhodecode.apps._base import BaseAppView, RepoAppView from rhodecode.lib.auth import (LoginRequired, HasRepoPermissionAnyDecorator) from rhodecode.lib.helpers import Page from rhodecode.lib.utils2 import safe_str from rhodecode.lib.index import searcher_from_config from rhodecode.model import validation_schema from rhodecode.model.validation_schema.schemas import search_schema log = logging.getLogger(__name__) def search(request, tmpl_context, repo_name): searcher = searcher_from_config(request.registry.settings) formatted_results = [] execution_time = '' schema = search_schema.SearchParamsSchema() search_params = {} errors = [] try: search_params = schema.deserialize( dict(search_query=request.GET.get('q'), search_type=request.GET.get('type'), search_sort=request.GET.get('sort'), page_limit=request.GET.get('page_limit'), requested_page=request.GET.get('page')) ) except validation_schema.Invalid as e: errors = e.children def url_generator(**kw): q = urllib.quote(safe_str(search_query)) return update_params( "?q=%s&type=%s" % (q, safe_str(search_type)), **kw) c = tmpl_context search_query = search_params.get('search_query') search_type = search_params.get('search_type') search_sort = search_params.get('search_sort') if search_params.get('search_query'): page_limit = search_params['page_limit'] requested_page = search_params['requested_page'] try: search_result = searcher.search( search_query, search_type, c.auth_user, repo_name, requested_page, page_limit, search_sort) formatted_results = Page( search_result['results'], page=requested_page, item_count=search_result['count'], items_per_page=page_limit, url=url_generator) finally: searcher.cleanup() if not search_result['error']: execution_time = '%s results (%.3f seconds)' % ( search_result['count'], search_result['runtime']) elif not errors: node = schema['search_query'] errors = [ validation_schema.Invalid(node, search_result['error'])] c.perm_user = c.auth_user c.repo_name = repo_name c.sort = search_sort c.url_generator = url_generator c.errors = errors c.formatted_results = formatted_results c.runtime = execution_time c.cur_query = search_query c.search_type = search_type c.searcher = searcher class SearchView(BaseAppView): def load_default_context(self): c = self._get_local_tmpl_context() self._register_global_c(c) return c @LoginRequired() @view_config( route_name='search', request_method='GET', renderer='rhodecode:templates/search/search.mako') def search(self): c = self.load_default_context() search(self.request, c, repo_name=None) return self._get_template_context(c) class SearchRepoView(RepoAppView): def load_default_context(self): c = self._get_local_tmpl_context() self._register_global_c(c) return c @LoginRequired() @HasRepoPermissionAnyDecorator( 'repository.read', 'repository.write', 'repository.admin') @view_config( route_name='search_repo', request_method='GET', renderer='rhodecode:templates/search/search.mako') def search_repo(self): c = self.load_default_context() search(self.request, c, repo_name=self.db_repo_name) return self._get_template_context(c)