fix(dref): Change relateable models instances to starting langauge

- Refactor dref translation related models

Author: susilnem

dref/serializers.py

@@ -38,7 +38,7 @@
 from main.writable_nested_serializers import NestedCreateMixin, NestedUpdateMixin
 from utils.file_check import validate_file_type
 
-from .tasks import send_dref_email
+from .tasks import _translate_related_objects, send_dref_email
 
 
 class RiskSecuritySerializer(ModelSerializer):
@@ -1087,6 +1087,14 @@ def create(self, validated_data):
             operational_update.users.add(*dref_operational_update.users.all())
             operational_update.risk_security.add(*dref_operational_update.risk_security.all())
             operational_update.source_information.add(*dref_operational_update.source_information.all())
+
+        # NOTE: Sync related models with the starting language
+        if starting_langauge != "en":
+            _translate_related_objects(
+                instance=operational_update,
+                auto_translate=False,
+                language=starting_langauge,
+            )
         return operational_update
 
     def update(self, instance, validated_data):
@@ -1541,6 +1549,14 @@ def create(self, validated_data):
             # also update is_final_report_created for dref
             dref.is_final_report_created = True
             dref.save(update_fields=["is_final_report_created"])
+
+        # NOTE: Sync related models with the starting language
+        if starting_langauge != "en":
+            _translate_related_objects(
+                instance=dref_final_report,
+                auto_translate=False,
+                language=starting_langauge,
+            )
         return dref_final_report
 
     def update(self, instance, validated_data):

dref/tasks.py

@@ -78,7 +78,25 @@ def process_dref_translation(model_name, instance_pk):
         return False
 
 
-def _translate_related_objects(instance, visited=None):
+def _translate_related_objects(
+    instance,
+    visited=None,
+    auto_translate=True,
+    language="en",
+):
+    """
+    Sync the relateable translation fields for the given model instance.
+    This function ensures that the translation fields are updated correctly
+    based on the current language settings.
+
+    Args:
+        instance: The model instance whose related objects need to be translated.
+        visited: A set to keep track of visited instances to avoid infinite recursion.
+        auto_translate: A boolean indicating whether to auto-translate related objects.
+        language: The language code to set for the original language field.
+
+    """
+
     if visited is None:
         visited = set()
 
@@ -102,15 +120,17 @@ def _translate_related_objects(instance, visited=None):
         if not field.many_to_many:
             if hasattr(related_value, TRANSLATOR_ORIGINAL_LANGUAGE_FIELD_NAME):
                 model_name = get_model_name(type(related_value))
-                translate_model_fields(model_name, related_value.pk)
-                related_value.translation_module_original_language = "en"
+                if auto_translate:
+                    translate_model_fields(model_name, related_value.id)
+                related_value.translation_module_original_language = language
                 related_value.save(update_fields=["translation_module_original_language"])
-                _translate_related_objects(related_value, visited)
+                _translate_related_objects(related_value, visited, auto_translate, language)
         else:
             for related_obj in related_value.all():
                 if hasattr(related_obj, TRANSLATOR_ORIGINAL_LANGUAGE_FIELD_NAME):
                     model_name = get_model_name(type(related_obj))
-                    translate_model_fields(model_name, related_obj.pk)
-                    related_obj.translation_module_original_language = "en"
+                    if auto_translate:
+                        translate_model_fields(model_name, related_obj.id)
+                    related_obj.translation_module_original_language = language
                     related_obj.save(update_fields=["translation_module_original_language"])
-                    _translate_related_objects(related_obj, visited)
+                    _translate_related_objects(related_obj, visited, auto_translate, language)