attributs Céleri personnalisé fonction de tâche

Question

J'ai une fonction de tâche de céleri qui ressemble this-

@task(base=MyBaseTask) 
@my_custom_decorator 
def my_task(*args, **kwargs): 
    my_task.ltc.some_func() #fails - attribute ltc doesn't exist on the object 

et my_custom_decorator ressemble à ce

def my_custom_decorator (f): 
    from functools import wraps 
    ltc = SomeClass() 
    @wraps(f) 
    def _inner(*args, **kwargs): 
     ret_obj = None 
     try: 
      f.task_cache = ltc 
      ret_obj = f(*args, **kwargs) 
     except Exception, e: 
      raise 
     return ret_obj 
    _inner.ltc = ltc 
    return _inner 

Je vois que cela est parce que l'objet appelable réel qui est invoqué pour exécuter la tâche est un objet de type classe de tâche céleri. Comment puis-je conserver mon attribut 'ltc' sur cet objet pour pouvoir y accéder à partir de la tâche comme indiqué ci-dessus i.e - my_task.ltc.some_func()?

Merci,

Answer 1

Je pense que d'une façon facile de le faire serait d'introduire ltc comme paramètre mot-clé ..

@task(base=MyBaseTask) 
@my_custom_decorator 
def my_task(*args, **kwargs): 
    ltc = kwargs['ltc'] 
    ltc.some_func() 

peut-être de cette façon:

def my_custom_decorator (f): 
    from functools import wraps 
    ltc = SomeClass() 
    @wraps(f) 
    def _inner(*args, **kwargs): 
     ret_obj = None 
     try: 
      f.task_cache = ltc 
      kwargs['ltc'] = ltc 
      ret_obj = f(*args, **kwargs) 
     except Exception, e: 
      raise 
     return ret_obj 
    _inner.ltc = ltc 
    return _inner 

Je ne Je ne sais pas s'il y a une tâche pour le céleri. J'espère que cela vous aide.