# HG changeset patch
# User Marcin Kuzminski <marcin@rhodecode.com>
# Date 2017-12-19 14:32:52
# Node ID 0e246bbee51f34fbf0832bfdd4a54d4de3409292
# Parent  d10039efd3fefbd9b2ce3b825cdc0f1e683cbbdd

celery: register EE tasks if we have EE enabled

diff --git a/rhodecode/lib/celerylib/loader.py b/rhodecode/lib/celerylib/loader.py
--- a/rhodecode/lib/celerylib/loader.py
+++ b/rhodecode/lib/celerylib/loader.py
@@ -28,6 +28,7 @@ Celery loader, run with::
 """
 import os
 import logging
+import importlib
 
 from celery import Celery
 from celery import signals
@@ -77,11 +78,21 @@ def get_logger(obj):
     return custom_log
 
 
+imports = ['rhodecode.lib.celerylib.tasks']
+
+try:
+    # try if we have EE tasks available
+    importlib.import_module('rc_ee')
+    imports.append('rc_ee.lib.celerylib.tasks')
+except ImportError:
+    pass
+
+
 base_celery_config = {
     'result_backend': 'rpc://',
     'result_expires': 60 * 60 * 24,
     'result_persistent': True,
-    'imports': [],
+    'imports': imports,
     'worker_max_tasks_per_child': 100,
     'accept_content': ['json_ext'],
     'task_serializer': 'json_ext',