SearchResultSet

Complexity

Total Complexity

6

Size/Duplication

Total Lines	21
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__getitem__()	0	2	1
A	add()	0	3	1
A	__iter__()	0	3	1
A	__init__()	0	3	1
A	get_sorted()	0	3	2

from simple_forums.utils import get_setting, string_to_class


def get_search_class():
    """ Get the search backend class """
    backend_info = get_setting('search_backend', {})

    return string_to_class(
        backend_info.get(
            'search_class',
            'simple_forums.backends.search.SimpleSearch'))


class SearchResultSet:
    """ Contains objects that are the results of a search """

    def __init__(self):
        """ Create blank list of results """
        self.results = []

    def __getitem__(self, key):
        return self.results[key]

    def __iter__(self):
        """ Iterate over results """
        return iter(self.results)

    def add(self, obj, score=0):
        """ Add the given object to the result set """
        self.results.append((obj, score))

    def get_sorted(self):
        """ Get the result set ordered by score, descending """
        return sorted(self.results, key=lambda item: item[1], reverse=True)


# Make module classes easily accessible
from simple_forums.backends.search.base_search import BaseSearch        # noqa
from simple_forums.backends.search.elastic_search import ElasticSearch  # noqa
from simple_forums.backends.search.simple_search import SimpleSearch    # noqa