我有很多使用相同装饰器的函数。有没有办法将所有这些都放在一个,然后只在所有函数中应用一个?
答案 0 :(得分:1)
我花了一段时间,但我已经弄清楚了。这是一个例子:
import click
@click.group()
def cli():
pass
@cli.command()
@click.option('--number', help='Integer number')
@click.option('--letter', help='Single character')
def show(number, letter):
print "number {} and letter {}".format(number, letter)
@cli.command()
@click.option('--number', help='Integer number')
@click.option('--letter', help='Single character')
def flip(number, letter):
print "flipped number {} and letter {}".format(-int(number), letter.swapcase())
if __name__ == '__main__':
cli()
您可以使用一个装饰器来装饰所有命令,这些装饰器将应用所有其他类似的装置:
import click, functools
def the_one(f):
@cli.command()
@click.option('--number', help='Integer number')
@click.option('--letter', help='Single character')
@functools.wraps(f)
def wrapper(*args, **kwds):
return f(*args, **kwds)
return wrapper
@click.group()
def cli():
pass
@the_one
def show(number, letter):
print "number {} and letter {}".format(number, letter)
@the_one
def flip(number, letter):
print "flipped number {} and letter {}".format(-int(number), letter.swapcase())
if __name__ == '__main__':
cli()
这是在行动:
$ python example.py flip --number 2 --letter B
flipped number -2 and letter b