Базовий та повнотекстовий пошук із Django та Postgres

Якщо реляційні бази даних стандартизовані, то повнотекстовий пошук — досі ні. Існує кілька його варіантів відкритим кодом, наприклад ElasticSearch, Solr та Xapian. ElasticSearch — чи не найпопулярніше рішення, однак його складно налаштовувати та підтримувати. Крім того, якщо ви не користуєтеся певними додатковими функціями ElasticSearch, вам варто звернутись до можливостей повнотекстового пошуку, які пропонують багато реляційних (як-от Postgres, MySQL, SQLite) та нереляційних баз даних (як-от MongoDB та CouchDB). Зокрема Postgres добре підходить для повнотекстового пошуку. Django також підтримує його одразу з коробки. Для більшості Django-застосунків вам слід принаймні почати з повнотекстового пошуку з Postgres. Після цього вже можна шукати функціональніше рішення, таке як ElasticSearch або Solr. Поки що ми розповімо, як додати базовий та повнотекстовий пошук до Django-застосунку за допомогою Postgres.

Мета

У цій статті ми навчимося:

1. Налаштовувати базовий пошук у Django-застосунку з об'єктом Q.
2. Додавати повнотекстовий пошук до Django-застосунку.
3. Впорядковувати результати повнотекстового пошуку за відповідністю (через стемінг, ранжування та визначення ваги).

Налаштування проєкту та огляд

Клонуйте гілку base із репозиторію django-search:

$ git clone https://github.com/testdrivenio/django-search --branch base --single-branch
$ cd django-search

Ми користуватимемося Docker для простішого налаштування та запуску Postgres разом з Django. З кореневої директорії проєкту створіть образи та запустіть контейнери Docker:

$ docker-compose up -d --build

Далі виконайте міграції та створіть суперкористувача:

$ docker-compose exec web python manage.py migrate
$ docker-compose exec web python manage.py createsuperuser

Тоді перейдіть до http://127.0.0.1:8011/quotes/, щоб переконатися, чи застосунок працює як слід. Ви маєте побачити таке:

1. I am in the middle.
2. You don't like middle school. Ось що повертатиме кожен тип пошуку:

1. I am a pony.
2. You don't like ponies Ось що повертатиме кожен тип пошуку:

INSTALLED_APPS = [
    "django.contrib.admin",
    "django.contrib.auth",
    "django.contrib.contenttypes",
    "django.contrib.sessions",
    "django.contrib.messages",
    "django.contrib.staticfiles",
    "quotes.apps.QuotesConfig",
    "debug_toolbar",
    "django.contrib.postgres",  # new
]

Тепер гляньмо на два короткі приклади повнотекстового пошуку, у одному полі та у кількох полях.

Пошук в одному полі

Оновлюємо SearchResultsList:

class SearchResultsList(ListView):
    model = Quote
    context_object_name = "quotes"
    template_name = "search.html"
    def get_queryset(self):
        query = self.request.GET.get("q")
        return Quote.objects.filter(quote__search=query)

Тут ми шукаємо лише цитату з поля пошуку.