diff --git a/src/agenda_culturel/celery.py b/src/agenda_culturel/celery.py index bd76442..2ede094 100644 --- a/src/agenda_culturel/celery.py +++ b/src/agenda_culturel/celery.py @@ -66,10 +66,10 @@ def close_import_task(taskid, success, error_message, importer): task.status = ( BatchImportation.STATUS.SUCCESS if success else BatchImportation.STATUS.FAILED ) - task.nb_initial = importer.get_nb_events() - task.nb_imported = importer.get_nb_imported_events() - task.nb_updated = importer.get_nb_updated_events() - task.nb_removed = importer.get_nb_removed_events() + task.nb_initial = 0 if importer is None else importer.get_nb_events() + task.nb_imported = 0 if importer is None else importer.get_nb_imported_events() + task.nb_updated = 0 if importer is None else importer.get_nb_updated_events() + task.nb_removed = 0 if importer is None else importer.get_nb_removed_events() fields = ["status", "nb_initial", "nb_updated", "nb_imported", "nb_removed"] if not success: @@ -260,7 +260,7 @@ def run_recurrent_import(self, pklist): raise self.retry(countdown=120) except MaxRetriesExceededError as e: logger.error(e) - close_import_task(self.request.id, False, e, importer) + close_import_task(self.request.id, False, e, None) return pklist[1:] if is_list else False