Skip to content

Commit 623c886

Browse files
committed
ext: add celery signals ep
* closes inveniosoftware/invenio-jobs#67
1 parent 869c18d commit 623c886

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

Diff for: invenio_celery/ext.py

+7
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,15 @@ def init_app(self, app, entry_point_group="invenio_celery.tasks", **kwargs):
3636
self.init_config(app)
3737
self.celery = FlaskCeleryExt(app).celery
3838
self.entry_point_group = entry_point_group
39+
self.load_celery_signals()
3940
app.extensions["invenio-celery"] = self
4041

42+
def load_celery_signals(self):
43+
"""Load Celery signals."""
44+
# Import Celery signals via entry points
45+
for ep in pkg_resources.iter_entry_points("invenio_celery.signals"):
46+
__import__(ep.module_name) # Just import the module to trigger registration
47+
4148
def load_entry_points(self):
4249
"""Load tasks from entry points."""
4350
if self.entry_point_group:

0 commit comments

Comments
 (0)