add-ons: Fix autotranslate invocation

WeblateOrg · Oct 27, 2021 · 30c76a1 · 30c76a1
1 parent 0cc5b2c
commit 30c76a1
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/weblate/trans/tasks.py b/weblate/trans/tasks.py
@@ -425,12 +425,11 @@ def auto_translate_component(
     auto_source: str,
     engines: List[str],
     threshold: int,
-    component: Optional[Component] = None,
+    component: Optional[int],
 ):
-    if component is None:
-        component = Component.objects.get(pk=component_id)
+    component_obj = Component.objects.get(pk=component_id)
 
-    for translation in component.translation_set.iterator():
+    for translation in component_obj.translation_set.iterator():
         if translation.is_source:
             continue
 
@@ -440,15 +439,15 @@ def auto_translate_component(
             mode,
             filter_type,
             auto_source,
-            component.id,
+            component,
             engines,
             threshold,
             translation=translation,
             component_wide=True,
         )
-    component.update_source_checks()
-    component.run_batched_checks()
-    return {"component": component.id}
+    component_obj.update_source_checks()
+    component_obj.run_batched_checks()
+    return {"component": component_obj.id}
 
 
 @app.task(trail=False)