Is there a way to prevent the program from exiting after command handler

[Mufanc]

I want to do some extra work after the command handler, however, the program exits after main, like the following:

import click


@click.command()
def main():
    pass


if __name__ == '__main__':
    main()  # program exited here! 
    print('do something')

My current solution is to catch SystemExit, but that makes my code ugly and less readable, and I'm wondering if there's some configuration that keeps the click from exiting after the command handler