Merge pull request #17 from CybercentreCanada/hotfix/update_tool_version

Update tool version only if needed
CybercentreCanada · Jul 27, 2021 · 1d489d5 · 1d489d5
2 parents 7bc555f + 19d9fa7
commit 1d489d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/assemblyline_service_client/task_handler.py b/assemblyline_service_client/task_handler.py
@@ -390,8 +390,9 @@ def handle_task_result(self, result_json_path: str, task: ServiceTask):
             file.pop('path', None)
 
         new_tool_version = result.get('response', {}).get('service_tool_version', None)
-        if new_tool_version is not None:
-            self.session.headers.update({'service_tool_version': str(new_tool_version)})
+        if new_tool_version is not None and self.service_tool_version != new_tool_version:
+            self.service_tool_version = new_tool_version
+            self.session.headers.update({'service_tool_version': self.service_tool_version})
 
         data = dict(task=task.as_primitives(), result=result, freshen=True)
         try: