便利な機能をコマンドとして使いたい。そのために、click
ライブラリをテストしています。3 つの元の関数を定義し、次のように装飾しましたclick.command
。
import click
import os, sys
@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=True)
def add_name(content, to_stdout=False):
if not content:
content = ''.join(sys.stdin.readlines())
result = content + "\n\tadded name"
if to_stdout is True:
sys.stdout.writelines(result)
return result
@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=True)
def add_surname(content, to_stdout=False):
if not content:
content = ''.join(sys.stdin.readlines())
result = content + "\n\tadded surname"
if to_stdout is True:
sys.stdout.writelines(result)
return result
@click.command()
@click.argument('content', required=False)
@click.option('--to_stdout', default=False)
def add_name_and_surname(content, to_stdout=False):
result = add_surname(add_name(content))
if to_stdout is True:
sys.stdout.writelines(result)
return result
このようにして、3 つのコマンドを生成add_name
し、ファイルをadd_surname
使用add_name_and_surname
してパイプすることができます。setup.py
pip install --editable .
$ echo "original content" | add_name | add_surname
original content
added name
added surname
ただし、さまざまなクリック コマンドを関数として作成する場合、解決しなければならない小さな問題が 1 つあります。
$echo "original content" | add_name_and_surname
Usage: add_name_and_surname [OPTIONS] [CONTENT]
Error: Got unexpected extra arguments (r i g i n a l c o n t e n t
)
なぜ機能しないのかわかりません。このコマンドをコマンドとしてではなく関数としてadd_name_and_surname
呼び出す必要があります。そうしないと、関数をライブラリ関数とコマンドの両方として使用するという当初の目的が無効になります。add_name
add_surname