tutorial_2_decisions

Complexity

Total Complexity

5

Size/Duplication

Total Lines	49
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	is_name_set()	0	2	1
A	some_action()	0	2	1
A	ask_for_name()	0	7	2
A	say_hello()	0	2	1

"""Tutorial 2 - how to use decisions"""
import curio
import async_btree as bt
import contextvars


async def some_action():
    print("continue here...")


async def say_hello():    
    print(f"Hello {name.get()}")    

async def is_name_set():    
    return name.get() != ""

async def ask_for_name():
    new_name = input("Hello, whats your name? \n")        
    if new_name != "":
        name.set(new_name)
        return bt.definition.SUCCESS
    else:
        return bt.definition.FAILURE


name = contextvars.ContextVar('name',default="")

greet_with_name = bt.decision(condition=is_name_set,
        success_tree=bt.always_success(say_hello),
        failure_tree=bt.sequence([ask_for_name,bt.always_success(say_hello)])            
        )

b_tree = bt.sequence(
    children=[
        greet_with_name,        
        bt.always_success(child=bt.action(target=some_action)),        
    ]
)

if __name__ == '__main__':

    name = contextvars.ContextVar('name',default="")

    curio.run(b_tree)

    # You can take a look at the final behavior tree
    abstract_tree_tree_1 = bt.analyze(b_tree) 
    print(bt.stringify_analyze(abstract_tree_tree_1))