Page MenuHomec4science
Files F96457376

api.py
No OneTemporary
Actions

File Metadata

Created
Thu, Dec 26, 22:36

api.py
View Options

# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2014, 2015 CERN.
#
# Invenio 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 2 of the
# License, or (at your option) any later version.
#
# Invenio 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 Invenio; if not, write to the Free Software Foundation, Inc.,
# 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
"""Search engine API."""
import pypeg2
from flask_login import current_user
from invenio_query_parser.parser import Main
from werkzeug.utils import cached_property
from .walkers.terms import Terms
from .walkers.match_unit import MatchUnit
class SearchEngine(object):
"""Search engine implemetation."""
def __init__(self, query):
"""Initialize with search query."""
self._query = query
@cached_property
def query(self):
"""Parse query string using given grammar."""
from invenio_query_parser.walkers.pypeg_to_ast import \
PypegConverter
tree = pypeg2.parse(self._query, Main, whitespace="")
return tree.accept(PypegConverter())
def search(self, user_info=None, collection=None):
"""Search records."""
user_info = user_info or current_user
from .searchext.engines.native import search
return search(self, user_info=user_info, collection=collection)
def match(self, record, user_info=None):
"""Return True if record match the query."""
return self.query.accept(MatchUnit(record))
def terms(self, keywords=None):
"""Return list of terms for given keywords in query pattern."""
return self.query.accept(Terms(keywords=keywords))

Event Timeline

c4science · Help