src/catalogue/models/snippet.py

   1 from django.db import models
   2 from django.contrib.postgres.indexes import GinIndex
   3 from django.contrib.postgres.search import SearchVector, SearchVectorField
   4 from search.utils import UnaccentSearchVector
   5
   6
   7 class Snippet(models.Model):
   8     book = models.ForeignKey('Book', models.CASCADE)
   9     sec = models.IntegerField()
  10     text = models.TextField()
  11     search_vector = SearchVectorField()
  12
  13     class Meta:
  14         indexes = [
  15             GinIndex('search_vector', name='search_vector_idx'),
  16         ]
  17
  18     def save(self, *args, **kwargs):
  19         super().save(*args, **kwargs)
  20         if not self.search_vector:
  21             self.update()
  22
  23     def update(self):
  24         self.search_vector = UnaccentSearchVector('text')
  25         self.save()
  26
  27     @classmethod
  28     def update_all(cls):
  29         cls.objects.all().update(search_vector = build_search_vector('text'))