Main function decorator (for Python 2 and 3).

automain.py

import inspect

def main(function):

    locale = inspect.stack()[1][0].f_locals
    module = locale.get("__name__", None)

    if module == "__main__":
        locale[function.__name__] = function
        function()

    return function

The following modifications work fine for me under 3.10.5

from inspect import stack

def automain( func ) :
  locale = stack()[ 1 ].frame.f_locals
  module = locale.get( '__name__', None )
  if module == '__main__' :
    locale[ func.__name__ ] = func
    func()
  return func