セロリで条件付きサブタスクのチェーンを作成するにはどうすればよいですか？

Question

私は、実行する一連のタスクを作成するアプリケーションを作成していますが、チェーンは、そのピースが追加されるというユーザーの希望に応じてタスクを実行します。ユーザーがstart_booしたい場合

例えば、チェーンは次のようになります。fooとbazがすでに開始されている場合

def start_boo(): 
    chain = start_foo.s() | start_bar.s() | start_baz.s() 
    chain() 

しかし、我々はこれを行うにはしたくないでしょう。代わりに次のようなものを好む：

def start_boo(foo=True, bar=True, baz=True): 
    if not (foo or bar or baz): 
     raise Exception("At least one should be true...") 
    chain = None 
    if foo: 
     chain |= start_foo.s() 
    if bar: 
     chain |= start_bar.s() 
    if baz: 
     chain |= start_baz.s() 
    chain() 

start_boo(foo=False, baz=False) 

ただし、これはさまざまな理由で機能しません。

このようなことを行うイディオムはありますか？

Answer 1

イディオムはfunctoolsのreduceの機能です。あなたは次のことができます：

def start_boo(foo=True, bar=True, baz=True): 
    if not (foo or bar or baz): 
     raise Exception("At least one should be true...") 

    todo_tasks = [foo, bar, baz] 
    start_tasks = [start_foo, start_bar, start_baz] 

    # tasks contains start tasks which should be done per the options. 
    # if it's False in todo_tasks, its associated start_task isn't added 
    tasks = [start_task for todo, start_task in zip(todo_tasks, start_tasks) if todo] 
    first_task, rest = *tasks 

    # start with the first task to be completed and chain it with remaining tasks 
    chain = functools.reduce(lambda x, y: x | y.s(), rest, first_task.s()) 
    chain()