diff --git a/server/surveys/models.py b/server/surveys/models.py index e0e5d6fc..454cc8d7 100644 --- a/server/surveys/models.py +++ b/server/surveys/models.py @@ -2,15 +2,21 @@ from django.contrib.auth import get_user_model from django.db import models from django.db.models import JSONField from wagtail.snippets.models import register_snippet +from wagtail.search import index @register_snippet -class Survey(models.Model): +class Survey(models.Model, index.Indexed): title = models.CharField(max_length=255) module = models.ForeignKey('books.Module', related_name='surveys', on_delete=models.CASCADE, null=True, blank=True) data = JSONField() + search_fields = [ + index.SearchField('title', partial_match=True), + index.SearchField('module__meta_title', partial_match=True), + ] + def __str__(self): return self.title