Changelog:

* Auto update model i18n field , delete / update languages

Author: dorin.musteata

README.md

@@ -85,6 +85,9 @@ class Blog(models.Model):  # noqa
 
     # DRF Localize model specific constant, which is used to store translations (json)
     LOCALIZE_FIELD = 'i18n'
+    
+    # DRF Localize model specific constant, which is used to auto-update or remove translations for languages
+    LOCALIZE_AUTO_UPDATE = False # not required , defaults to False
 
     # Your custom model fields
     title = models.CharField(max_length=254, null=True)

drf_localize/core/__init__.py

@@ -7,12 +7,6 @@
 from django.db.models.base import ModelBase
 from contextlib import suppress
 from zipfile import ZipFile
-from django.db.models import (
-    F,
-    Func,
-    Value,
-    JSONField
-)
 
 # Import your package here.
 
@@ -451,28 +445,48 @@ def _signal(self, model=None):
 
         translate, field, auto_update = self._model_set(model=model)
 
-        if not auto_update:
+        if not all([translate, field, auto_update]):
             return self
 
         languages = self.get_languages()
+        objects = []
+        queryset = model.objects.all()
+
+        for element in queryset:
+            i18n = getattr(element, field, None)
+
+            if not i18n:
+                continue
 
-        x = model.objects.values_list(field, 'id')
-        for i18n, _id in x:
             keys = list(i18n.keys())
-            difference = list(set(languages).difference(keys))
+            difference_to_update = list(set(languages).difference(keys))
+            difference_to_delete = list(set(keys).difference(languages))
 
-            if not difference:
+            if not difference_to_update and not difference_to_delete:
                 continue
 
-            payload = {}
-            for language in difference:
-                payload.update({language: {}})
+            for language in difference_to_delete:
+                i18n.pop(language, None)
 
-                for translate_field in translate:
-                    payload[language].update({translate_field: ''})
+            for language in difference_to_update:
+                i18n.update({language: {}})
 
-            i18n.update(**payload)
-            model.objects.filter(id=_id).update(**{field: i18n})
+                for translate_field in translate:
+                    i18n[language].update({translate_field: ''})
+
+            # Set updatable field
+            setattr(element, field, i18n)
+            objects.append(element)
+
+        # Bulk update model objects
+        if length := len(objects):
+            model.objects.bulk_update(
+                objs=objects,
+                fields=[
+                    field,
+                ],
+                batch_size=length,
+            )
 
 
 # Export localize instance

drf_localize/serializers.py

@@ -32,7 +32,7 @@ def __init__(self, **kwargs):
         self.context = kwargs.pop('context', None)
         self.localize_namespace = kwargs.pop('namespace', False)
         self.localize_translate, self.localize_field, self.localize_auto_update = localize._model_set(model=self.localize_model)  # noqa
-        localize._signal(model=self.localize_model)
+        localize._signal(model=self.localize_model)  # noqa
         super(I18N, self).__init__(**kwargs)
 
     def to_representation(self, instance):

drf_localize_test/settings.py

@@ -84,5 +84,5 @@
 DEFAULT_AUTO_FIELD = 'django.db.models.AutoField'
 
 DRF_LOCALIZE = {
-    'LANGUAGES': ['en', 'ru', 'ro', 'pt', 'bg'],
+    'LANGUAGES': ['en', 'ro', 'bg'],
 }

drf_localize_test/test/models.py

@@ -12,7 +12,7 @@ class Blog(models.Model):  # noqa
     LOCALIZE_FIELD = 'i18n'
 
     # DRF Localize model specific constant, which is used to auto-update or remove translations for new languages
-    LOCALIZE_AUTO_UPDATE = True
+    LOCALIZE_AUTO_UPDATE = False
 
     # Your custom model fields
     title = models.CharField(max_length=254, null=True)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "drf-localize"
-version = "0.1.4"
+version = "0.1.5"
 description = "Package to provide localization experiences for mobile and api applications."
 authors = ["Dorin Musteața"]
 readme = "README.md"

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="drf-localize",
-    version="0.1.4",
+    version="0.1.5",
     author="Dorin Musteața",
     description="Package to provide localization experiences for mobile and api applications.",
     long_description=README,