Browse Source

Added ability to change task name generation behavior

Dmitry Malinovsky 10 years ago
parent
commit
d618ee2681
1 changed files with 4 additions and 1 deletions
  1. 4 1
      celery/app/base.py

+ 4 - 1
celery/app/base.py

@@ -268,7 +268,7 @@ class Celery(object):
     def _task_from_fun(self, fun, name=None, base=None, bind=False, **options):
         if not self.finalized and not self.autofinalize:
             raise RuntimeError('Contract breach: app not finalized')
-        name = name or gen_task_name(self, fun.__name__, fun.__module__)
+        name = name or self.gen_task_name(self, fun.__name__, fun.__module__)
         base = base or self.Task
 
         if name not in self._tasks:
@@ -286,6 +286,9 @@ class Celery(object):
             task = self._tasks[name]
         return task
 
+    def gen_task_name(self, name, module):
+        return gen_task_name(self, name, module)
+
     def finalize(self, auto=False):
         with self._finalize_mutex:
             if not self.finalized: