NDOJ/judge/fulltext.py

# From: http://www.mercurytide.co.uk/news/article/django-full-text-search/

from django.db import connection, models
from django.db.models.query import QuerySet


class SearchQuerySet(QuerySet):
    DEFAULT = ""
    BOOLEAN = " IN BOOLEAN MODE"
    NATURAL_LANGUAGE = " IN NATURAL LANGUAGE MODE"
    QUERY_EXPANSION = " WITH QUERY EXPANSION"

    def __init__(self, fields=None, **kwargs):
        super(SearchQuerySet, self).__init__(**kwargs)
        self._search_fields = fields

    def _clone(self, *args, **kwargs):
        queryset = super(SearchQuerySet, self)._clone(*args, **kwargs)
        queryset._search_fields = self._search_fields
        return queryset

    def search(self, query, mode=DEFAULT):
        meta = self.model._meta

        # Get the table name and column names from the model
        # in `table_name`.`column_name` style
        columns = [meta.get_field(name).column for name in self._search_fields]
        full_names = [
            "%s.%s"
            % (
                connection.ops.quote_name(meta.db_table),
                connection.ops.quote_name(column),
            )
            for column in columns
        ]

        # Create the MATCH...AGAINST expressions
        fulltext_columns = ", ".join(full_names)
        match_expr = "MATCH(%s) AGAINST (%%s%s)" % (fulltext_columns, mode)

        # Add the extra SELECT and WHERE options
        return self.extra(
            select={"relevance": match_expr},
            select_params=[query],
            where=[match_expr],
            params=[query],
        )


class SearchManager(models.Manager):
    def __init__(self, fields=None):
        super(SearchManager, self).__init__()
        self._search_fields = fields

    def get_queryset(self):
        if self._search_fields is not None:
            return SearchQuerySet(model=self.model, fields=self._search_fields)
        return super(SearchManager, self).get_queryset()

    def search(self, *args, **kwargs):
        return self.get_queryset().search(*args, **kwargs)