如何用一个替换多个python装饰器?

时间:2018-05-12 04:30:32

标签: python

我有很多使用相同装饰器的函数。有没有办法将所有这些都放在一个,然后只在所有函数中应用一个?

1 个答案:

答案 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