ext: add celery signals ep

CHANGES.rst

@@ -9,6 +9,10 @@
 Changes
 =======
 
+Version 2.1.0 (released 2025-04-28)
+
+- ext: add celery signals entrypoint
+
 Version 2.0.0 (released 2024-12-02)
 
 - setup: bump invenio dependencies

invenio_celery/__init__.py

@@ -91,6 +91,6 @@ def sum(x, y):
 
 from .ext import InvenioCelery
 
-__version__ = "2.0.0"
+__version__ = "2.1.0"
 
 __all__ = ("__version__", "InvenioCelery")

invenio_celery/ext.py

@@ -36,8 +36,15 @@ def init_app(self, app, entry_point_group="invenio_celery.tasks", **kwargs):
         self.init_config(app)
         self.celery = FlaskCeleryExt(app).celery
         self.entry_point_group = entry_point_group
+        self.load_celery_signals()
         app.extensions["invenio-celery"] = self
 
+    def load_celery_signals(self):
+        """Load Celery signals."""
+        # Import Celery signals via entry points
+        for ep in pkg_resources.iter_entry_points("invenio_celery.signals"):
+            __import__(ep.module_name)  # Just import the module to trigger registration
+
     def load_entry_points(self):
         """Load tasks from entry points."""
         if self.entry_point_group:

tests/test_invenio_celery.py

@@ -27,6 +27,7 @@ def _mock_entry_points(group, name=None):
             EntryPoint("bpackage_2", "bpackage.second_tasks"),
             EntryPoint("apackage", "apackage.third_tasks"),
         ],
+        "invenio_celery.signals": [],
     }
     assert name is None
     for entry_point in data[group]: