Move celery to tasks

hypothesis · Jan 31, 2025 · 4375704 · 4375704
1 parent 1338b6b
commit 4375704
Show file tree

Hide file tree

Showing 9 changed files with 10 additions and 10 deletions.
diff --git a/h/cli/commands/celery.py b/h/cli/commands/celery.py
@@ -1,6 +1,6 @@
 import click
 
-from h.celery import start
+from h.tasks.celery import start
 
 
 @click.command(

diff --git a/h/tasks/annotations.py b/h/tasks/annotations.py
@@ -1,7 +1,7 @@
-from h.celery import celery, get_task_logger
 from h.db.types import URLSafeUUID
 from h.models import Annotation
 from h.services.annotation_write import AnnotationWriteService
+from h.tasks.celery import celery, get_task_logger
 
 log = get_task_logger(__name__)
 

diff --git a/h/celery.py → h/tasks/celery.py b/h/celery.py → h/tasks/celery.py
diff --git a/h/tasks/cleanup.py b/h/tasks/cleanup.py
@@ -4,7 +4,7 @@
 from sqlalchemy import func, select
 
 from h import models
-from h.celery import celery, get_task_logger
+from h.tasks.celery import celery, get_task_logger
 
 log = get_task_logger(__name__)
 

diff --git a/h/tasks/indexer.py b/h/tasks/indexer.py
@@ -3,8 +3,8 @@
 import newrelic
 from celery import Task
 
-from h.celery import celery, get_task_logger
 from h.services import AnnotationSyncService
+from h.tasks.celery import celery, get_task_logger
 
 log = get_task_logger(__name__)
 

diff --git a/h/tasks/job_queue.py b/h/tasks/job_queue.py
@@ -1,4 +1,4 @@
-from h.celery import celery, get_task_logger
+from h.tasks.celery import celery, get_task_logger
 
 log = get_task_logger(__name__)
 

diff --git a/h/tasks/mailer.py b/h/tasks/mailer.py
@@ -9,7 +9,7 @@
 import pyramid_mailer
 import pyramid_mailer.message
 
-from h.celery import celery, get_task_logger
+from h.tasks.celery import celery, get_task_logger
 
 __all__ = ("send",)
 

diff --git a/h/tasks/url_migration.py b/h/tasks/url_migration.py
@@ -1,4 +1,4 @@
-from h.celery import celery
+from h.tasks.celery import celery
 
 
 @celery.task(rate_limit="10/m")

diff --git a/tests/unit/h/celery_test.py b/tests/unit/h/celery_test.py
@@ -2,7 +2,7 @@
 
 from billiard.einfo import ExceptionInfo
 
-from h import celery
+from h.tasks import celery
 
 
 class TestCelery:
@@ -45,7 +45,7 @@ def test_transaction_abort_aborts_request_transaction(self):
         sender.app.request.tm.abort.assert_called_once_with()
 
     def test_report_failure_reports_failure_in_debug_mode(self, patch):
-        log = patch("h.celery.log")
+        log = patch("h.tasks.celery.log")
         sender = mock.Mock(spec=["app"])
         sender.name = "wibble"
         sender.app.request.debug = True
@@ -62,7 +62,7 @@ def test_report_failure_reports_failure_in_debug_mode(self, patch):
         assert log.error.called
 
     def test_report_failure_skipped_when_not_in_debug_mode(self, patch):
-        log = patch("h.celery.log")
+        log = patch("h.tasks.celery.log")
         sender = mock.Mock(spec=["app"])
         sender.name = "wibble"
         sender.app.request.debug = False