Click is prematurely exiting my script with exit_code 0

[jrmidkiff]

I'm having trouble using click==8.1.7, python==3.8 with a package I wrote - unfortunately I can't share the package as it's private, but I'll provide as much detail as I can. The package never reaches the line 1/0 always exiting the script altogether with an exit code of 0 after the line return engine

import sqlalchemy as sa
import citygeo_secrets

def create_engine(creds: dict, prod: bool) -> sa.engine: 
    '''Create engine and confirm that a connection is possible'''
    if prod: 
        url_object = sa.URL.create(<stuff_from_creds>)
    else: 
        url_object = sa.URL.create(<other_stuff_from_creds>)
    
    engine = sa.create_engine(url_object)
    engine.connect() # Ensure the information passed is correct by making test conn
    return engine

@click.command()
@click.option('--prod/--test', default=False)
def get_engine(prod: bool) -> sa.engine: 
    ''''''
    engine = citygeo_secrets.connect_with_secrets(
        create_engine, 'databridge-v2/citygeo', 
        'databridge-v2/hostname-testing', 'databridge-v2/hostname', 
        prod=prod) 
    return engine # Click exits the entire script after here
    

if __name__ == '__main__': 
    ''' Run this script via `python script.py --prod/--test` '''
    engine = get_engine()
    1/0 # This line is never reached as the application exits with exit code 0 

All that citygeo_secrets does is essentially the following, and the function is working successfully with or without Click

def connect_with_secrets(func: Callable[[dict], Any], *secret_names: str, **kwargs): 
    ''' Use secret names to connect to host, automatically retrieving newest secrets 
    and retrying once if an exception is raised.

    - func: User function that accepts a dictionary input and returns a 
    database/server/API/etc. connection
    - secret_names: Names of secrets to gather
    - **kwargs: keyword-arguments to pass to func'''

    secrets = get_secrets(*secret_names)
    try: 
        return func(secrets, **kwargs)

Some other things I noticed after the return statement:

• Context._depth == 2 initially
• Once Context._depth == 1, then in BaseCommand.main, It seemed like standalone_mode == False when the context is trying to teardown and trying to return my SQLAlchemy Engine, so that is when it exits with exit code 0

Any thoughts? Thank you so much!

[jrmidkiff]

It seems that by design nothing will run in Click after the function decorated with click.command() has been called, hence the exit_code 0. I don't know if I missed this in the documentation, or if it should be clearer in there