search preference settings

cookbook/migrations/0124_build_full_text_index.py

@@ -1,14 +1,14 @@
 # Generated by Django 3.1.7 on 2021-04-07 20:00
+import annoying.fields
 from django.conf import settings
 from django.contrib.postgres.indexes import GinIndex
 from django.contrib.postgres.search import SearchVectorField, SearchVector
-from django.db import migrations
+from django.db import migrations, models
+from django.db.models import deletion
 from django_scopes import scopes_disabled
 from django.utils import translation
 from cookbook.managers import DICTIONARY
-from cookbook.models import Recipe, Step, Index
-
-
+from cookbook.models import Recipe, Step, Index, PermissionModelMixin, nameSearchField, allSearchFields
 
 
 def set_default_search_vector(apps, schema_editor):
@@ -17,6 +17,7 @@ def set_default_search_vector(apps, schema_editor):
     language = DICTIONARY.get(translation.get_language(), 'simple')
     with scopes_disabled():
         # TODO this approach doesn't work terribly well if multiple languages are in use
+        # I'm also uncertain about forcing unaccent here
         Recipe.objects.all().update(
             name_search_vector=SearchVector('name__unaccent', weight='A', config=language),
             desc_search_vector=SearchVector('description__unaccent', weight='B', config=language)
@@ -26,7 +27,8 @@ def set_default_search_vector(apps, schema_editor):
 
 class Migration(migrations.Migration):
     dependencies = [
-        ('cookbook', '0121_auto_20210518_1638'),
+        ('auth', '0012_alter_user_first_name_max_length'),
+        ('cookbook', '0123_invitelink_email'),
     ]
     operations = [
         migrations.AddField(
@@ -80,6 +82,28 @@ class Migration(migrations.Migration):
             model_name='viewlog',
             index=Index(fields=['recipe', '-created_at'], name='cookbook_vi_recipe__5cd178_idx'),
         ),
+        migrations.CreateModel(
+            name='SearchFields',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=32, unique=True)),
+                ('field', models.CharField(max_length=64, unique=True)),
+            ],
+            bases=(models.Model, PermissionModelMixin),
+        ),
+        migrations.CreateModel(
+            name='SearchPreference',
+            fields=[
+                ('user', annoying.fields.AutoOneToOneField(on_delete=deletion.CASCADE, primary_key=True, serialize=False, to='auth.user')),
+                ('search', models.CharField(choices=[('PLAIN', 'Plain'), ('PHRASE', 'Phrase'), ('WEBSEARCH', 'Web'), ('RAW', 'Raw')], default='SIMPLE', max_length=32)),
+                ('fulltext', models.ManyToManyField(blank=True, related_name='fulltext_fields', to='cookbook.SearchFields')),
+                ('icontains', models.ManyToManyField(blank=True, default=nameSearchField, related_name='icontains_fields', to='cookbook.SearchFields')),
+                ('istartswith', models.ManyToManyField(blank=True, related_name='istartswith_fields', to='cookbook.SearchFields')),
+                ('trigram', models.ManyToManyField(blank=True, related_name='trigram_fields', to='cookbook.SearchFields')),
+                ('unaccent', models.ManyToManyField(blank=True, default=allSearchFields, related_name='unaccent_fields', to='cookbook.SearchFields')),
+            ],
+            bases=(models.Model, PermissionModelMixin),
+        ),
         migrations.RunPython(
             set_default_search_vector
         ),

cookbook/migrations/0125_create_searchfields.py.stop

@@ -0,0 +1,23 @@
+from cookbook.models import SearchFields
+from django.db import migrations
+
+
+def create_searchfields(apps, schema_editor):
+    SearchFields.objects.create(name='Name', field='name')
+    SearchFields.objects.create(name='Description', field='description')
+    SearchFields.objects.create(name='Instructions', field='steps__instruction')
+    SearchFields.objects.create(name='Ingredients', field='steps__ingredients__food__name')
+    SearchFields.objects.create(name='Keywords', field='keywords__name')
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('cookbook', '0124_build_full_text_index'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            create_searchfields
+        ),
+    ]