ClickをつかってPython製CLIツールをスマートにする その３：yes/no

shuzo-kino.hateblo.jp
シリーズの３回目。
今回はCLIツールではよく見かけるyes/noの選択肢を表示するオプションのはなし。

実際のところ

何か文字列を表示し、yes/noを選択させるアプリを考えてみます。

import click

def abort_if_false(ctx, param, value):
    if not value:
        ctx.abort()

@click.command()
@click.option('--yes', is_flag=True, callback=abort_if_false,
              expose_value=False,
              prompt='Are you sure you want to drop the db?')
def dropdb():
    click.echo('Dropped all tables!')

def filelist():
    click.echo("some function")

if __name__ == '__main__':
   filelist()
   dropdb()

動かすとこんな感じ。
noを選択した場合でも"ctx.abort()" で中止してくれます。便利ですねぇ

$ python sample_click_yes.py
some function
Are you sure you want to drop the db? [y/N]: n
Aborted!

$ python sample_click_yes.py
some function
Are you sure you want to drop the db? [y/N]: y
Dropped all tables!

"--yes"オプションで全部yesになります。
yesコマンド使わないでよいって事かもしれませんが、まぁあんまりオススメしたくないですねコレ。

$ python sample_click_yes.py --yes
some function
Dropped all tables!

参考もと

Options — Click Documentation (6.0)