better main for python

main.py

import atexit
import sys

def main(f):
    if f.__module__ == "__main__":
        atexit.register(f)


sys.modules['main'] = main  # noqa

program.py

import main


@main
def my_main_function():
    print("This is my main function")


print("this runs first")