exactly same options for 2 different functions in python click

Question

I'm writing a tool using Python 2 and click which reads/writes registers in hardware. I have two functions that accept exactly the same options. The difference is that they process input and direct output to a different devices.

Here is what I have so far:

@cli.command()
@click.option('--r0', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r1', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r2', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
def mydevice1(r0, r1, r2):
    # Handle inputs for device 1
    click.echo('myfunc1')

@cli.command()   
@click.option('--r0', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r1', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r2', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
def mydevice2(r0, r1, r2):
    # Handle inputs for device 2
    click.echo('myfunc2')

Both of the functions will handle inputs in the same way, the only difference is that they will pass handled info to different devices. In other words what I'd like to have is

@click.option('--r0', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r1', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
@click.option('--r2', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
def handle_common_options(r0, r1, r2):
    # Handle common options
    pass

@cli.command()
def mydevice1():
    handle_common_options()
    # pass processed options to device 1

@cli.command()
def mydevice2():
    handle_common_options()
    # pass processed options to device 2

Is this possible?

Stephane Martin · Accepted Answer · 2016-11-24 00:20:56Z

2

sure.

@decorator
def f():
    pass

means

def f():
    pass
f = decorator(f)

so:

decorator0 = cli.command()
decorator1 = click.option('--r0', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
decorator2 = click.option('--r1', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)
decorator3 = click.option('--r2', type=click.IntRange(-2, 0xffffffff, clamp=False), default=-2)

common_decorator = lambda f: decorator0(decorator1(decorator2(decorator3(f))))

@common_decorator
def mydevice1(r0, r1, r2):
    click.echo('myfunc1')

@common_decorator
def mydevice2(r0, r1, r2):
    click.echo('myfunc2')

without a lambda:

def common_decorator(f):
    return decorator0(decorator1(decorator2(decorator3(f))))

edited Nov 24, 2016 at 0:20

answered Nov 23, 2016 at 2:20

Stephane Martin

1,6521 gold badge19 silver badges26 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

flashburn Over a year ago

Is there a way to do it without a lambda?

flashburn Over a year ago

Would you mind showing me how? I've been at it for several hours and still can't figure out it.

Stephane Martin Over a year ago

the proposed answer was edited too. def f(x): return g(x) is equivalent to f = lambda x: g(x). Also please accept the proposed answer if you consider it good.

Collectives™ on Stack Overflow

exactly same options for 2 different functions in python click

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related