Also trap cases where service service completely crash dues to what w…

…e sent it.
CybercentreCanada · May 13, 2020 · 6cc8c02 · 6cc8c02
1 parent da04be1
commit 6cc8c02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/assemblyline_service_client/task_handler.py b/assemblyline_service_client/task_handler.py
@@ -396,7 +396,7 @@ def handle_task_result(self, result_json_path: str, task: ServiceTask):
 
                     data['freshen'] = False
                     r = self.request_with_retries('post', self._path('task'), json=data)
-        except ServiceServerException as e:
+        except (ServiceServerException, requests.HTTPError) as e:
             self.handle_task_error(task, message=str(e), error_type='EXCEPTION', status='FAIL_NONRECOVERABLE')
 
     def handle_task_error(self, task: ServiceTask, error_json_path: Optional[str] = None,