argparse after `__name__ == "__main__"`

Dear autoprogram team,

If I have a code like this:

if __name__ == "__main__":
    logger.setLevel("DEBUG")

    parser = argparse.ArgumentParser(
        description="""random"""
    )

then the autoprogram is unable to see the parser object for that specific file. Moving parser up before the if works, but require all the add_argument to be above the if __name__ == "__main__".

I usually declare the parser only if this is executed as a script or with the -m option of python, and tend to avoid global objects otherwise.

Is there a way to make it work like this?

Thank you,

Start by reproducing the issue with the example where the ArgumentParser is created inside if __name__ == "__main__":, including execution with the -m option. Trace how autoprogram discovers the parser and its add_argument calls; done means it can recognize this parser without requiring those declarations above the guard.