python click subcommand unified error handling

Question

In the case where there are command groups and every sub-command may raise exceptions, how can I handle them all together in one place?

Given the example below:

import click


@click.group()
def cli():
    pass

@cli.command()
def foo():
    pass

if __name__ == '__main__':
    cli()

Both cli and foo may raise. I know that one possible solution is to place try-except around cli() in the if clause. But that does not work when you distribute a package. In setup.py, you have to specify an entry point (in this case, cli). The if clause will not be executed.

Stephen Rauch · Accepted Answer · 2017-06-03 19:24:19Z

12

You can create a custom click.Group by inheriting from same. The custom group can be used by passing it as the cls parameter to the click.group() decorator. If you override the __call__ method, you can insert an exception handler like:

Code:

class CatchAllExceptions(click.Group):

    def __call__(self, *args, **kwargs):
        try:
            return self.main(*args, **kwargs)
        except Exception as exc:
            click.echo('We found %s' % exc)

Test Code:

import click

@click.group(cls=CatchAllExceptions)
def cli():
    pass

@cli.command()
def foo():
    raise Exception('an exception!')

if __name__ == '__main__':
    cli('foo'.split())

Results:

We found an exception!

answered Jun 3, 2017 at 19:24

Stephen Rauch♦

50.1k32 gold badges118 silver badges143 bronze badges

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

1 Comment

laker93 Over a year ago

This doesn't work when attempting to capture the exception in a unit test with the CliRunner. Invoking the custom group as follows: ``` runner = CliRunner() result = runner.invoke(cli, 'foo') print(result)` ``` prints: "an exception!"

Collectives™ on Stack Overflow

python click subcommand unified error handling

1 Answer 1

Code:

Test Code:

Results:

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Code:

Test Code:

Results:

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related