Merge pull request #994 from xucailiang/fix-celery-import-error

rename celery.py
arc53 · Jun 19, 2024 · c80678a · c80678a
2 parents eae49d2 + 1754570
commit c80678a
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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 → application/celery_init.py b/application/celery.py → application/celery_init.py
diff --git 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'