fix: DEV-3839: Avoid creation of pg_trgm in postgres if it's already existing (#3395)

fix: DEV-3839: Avoid creation of pg_trgm in postgres if it's already existing (#3384)

fix: DEV-3839: Avoid creation of pg_trgm in postgres if it's already exists

Author: makseq

label_studio/core/utils/common.py

@@ -35,7 +35,9 @@
 from django_filters.rest_framework import DjangoFilterBackend
 from drf_yasg.inspectors import CoreAPICompatInspector, NotHandled
 from collections import defaultdict
+from django.contrib.postgres.operations import TrigramExtension
 
+from core.utils.params import get_env
 from datetime import datetime
 from functools import wraps
 from pkg_resources import parse_version
@@ -658,3 +660,19 @@ def __exit__(self, type_, value, traceback):
                 sender=sender,
                 dispatch_uid=dispatch_uid
             )
+
+
+def trigram_migration_operations(next_step):
+    ops = [
+        TrigramExtension(),
+        next_step,
+    ]
+    SKIP_TRIGRAM_EXTENSION = get_env('SKIP_TRIGRAM_EXTENSION', None)
+    if SKIP_TRIGRAM_EXTENSION == '1' or SKIP_TRIGRAM_EXTENSION == 'yes' or SKIP_TRIGRAM_EXTENSION == 'true':
+        ops = [
+            next_step
+        ]
+    if SKIP_TRIGRAM_EXTENSION == 'full':
+        ops = []
+
+    return ops

label_studio/tasks/migrations/0009_auto_20210913_0739.py

@@ -2,7 +2,7 @@
 import logging
 
 from django.db import migrations
-from django.contrib.postgres.operations import TrigramExtension
+from core.utils.common import trigram_migration_operations
 
 logger = logging.getLogger(__name__)
 
@@ -30,7 +30,4 @@ class Migration(migrations.Migration):
 
     dependencies = [('tasks', '0008_auto_20210903_1332')]
 
-    operations = [
-        TrigramExtension(),
-        migrations.RunPython(forwards, backwards),
-    ]
+    operations = trigram_migration_operations(migrations.RunPython(forwards, backwards))

label_studio/tasks/migrations/0017_new_index_anno_result.py

@@ -2,7 +2,7 @@
 import logging
 
 from django.db import migrations
-from django.contrib.postgres.operations import TrigramExtension
+from core.utils.common import trigram_migration_operations
 
 logger = logging.getLogger(__name__)
 
@@ -26,14 +26,11 @@ def backwards(apps, schema_editor):
         return
 
     schema_editor.execute('drop index tasks_annotations_result_idx2;')
-
+    
 
 class Migration(migrations.Migration):
     atomic = False
 
     dependencies = [('tasks', '0016_auto_20220414_1408')]
 
-    operations = [
-        TrigramExtension(),
-        migrations.RunPython(forwards, backwards),
-    ]
+    operations = trigram_migration_operations(migrations.RunPython(forwards, backwards))