询问用户输入，运行指定功能，然后返回询问

Question

I am making a CLI in python using click. What i want to have is a kind of "idle state" which the program starts in. The script asks for user input, and depending on the input a specific function should be run. This sub-function can also ask for user input and do different things.

What im having trouble with is making the program return to the "idle state" after the function has completed running.

@click.command()    
def cli():
     print "Welcome. What do you want to do?"
     choice = click.prompt('')
     if choice == 'add':
            add()
     else:
            subtract()

So after running eg add() i want to program to once again print "Welcome" and prompt the user for input. Wrapping the cli()-function in a while-loop didn't work - im suspecting click breaks it somehow.

Answer 1

you can try this:

def cli():
    print "Welcome. What do you want to do?"
    choice = click.prompt('')
    if choice == 'add':
        add()
        cli()
    else:
        subtract()