diff --git a/application/api/user/routes.py b/application/api/user/routes.py index 5110149..443fadd 100644 --- a/application/api/user/routes.py +++ b/application/api/user/routes.py @@ -229,7 +229,7 @@ def upload_remote(): def task_status(): """Get celery job status.""" task_id = request.args.get("task_id") - from application.celery import celery + from application.celery_init import celery task = celery.AsyncResult(task_id) task_meta = task.info diff --git a/application/api/user/tasks.py b/application/api/user/tasks.py index 4602bf8..862b6dc 100644 --- a/application/api/user/tasks.py +++ b/application/api/user/tasks.py @@ -1,5 +1,5 @@ from application.worker import ingest_worker, remote_worker -from application.celery import celery +from application.celery_init import celery @celery.task(bind=True) def ingest(self, directory, formats, name_job, filename, user): diff --git a/application/app.py b/application/app.py index e646ffb..fe8efd1 100644 --- a/application/app.py +++ b/application/app.py @@ -1,6 +1,6 @@ import platform import dotenv -from application.celery import celery +from application.celery_init import celery from flask import Flask, request, redirect from application.core.settings import settings from application.api.user.routes import user diff --git a/application/celery.py b/application/celery_init.py similarity index 100% rename from application/celery.py rename to application/celery_init.py diff --git a/tests/test_celery.py b/tests/test_celery.py index f4b2244..f4ec6a0 100644 --- a/tests/test_celery.py +++ b/tests/test_celery.py @@ -1,9 +1,9 @@ from unittest.mock import patch from application.core.settings import settings -from application.celery import make_celery +from application.celery_init import make_celery -@patch('application.celery.Celery') +@patch('application.celery_init.Celery') def test_make_celery(mock_celery): # Arrange app_name = 'test_app_name'