src/search/utils.py

   1 from django.conf import settings
   2 from django.db.models import Func
   3 from django.contrib.postgres.search import SearchQuery, SearchVectorField
   4
   5
   6 class UnaccentSearchQuery(SearchQuery):
   7     '''
   8     The idea is to run unaccent *after* the query is already passed through the language dictionary.
   9     '''
  10     def as_sql(self, *args, **kwargs):
  11         sql, params = super().as_sql(*args, **kwargs)
  12         if settings.SEARCH_USE_UNACCENT:
  13             sql = f'unaccent({sql}::text)::tsquery'
  14         return sql, params
  15
  16
  17 class UnaccentSearchVector(Func):
  18     '''
  19     We do the indexing twice, to account for non-diacritic versions.
  20     For example: user enters 'róże' -> stem to 'róża' -> unaccent to 'roza'.
  21     But user enters 'roze' -> stem leaves it as is, so we need original form in the vector.
  22     '''
  23     function='to_tsvector'
  24     if settings.SEARCH_USE_UNACCENT:
  25         template = f'''unaccent(
  26         %(function)s('{settings.SEARCH_CONFIG}', %(expressions)s)::text)::tsvector ||
  27         to_tsvector(
  28         '{settings.SEARCH_CONFIG_SIMPLE}',
  29         unaccent(%(expressions)s)
  30         )'''
  31     output_field = SearchVectorField()