在执行之前将模块上的__name__设置为'__main__'

Question

I currently have code that checks if __name__ == '__main__' and then calls some function if that is the case. Currently, I execute this code by spawning another python subprocess, but importing it as a module is cleaner. The problem is that there is no main function in these modules is exactly what is executed under __name__ == '__main__' varies. The only option that I really have is to set the __name__ attribute before executing the module. What is the best way to do this?

Answer 1

What I ended up doing is the following:

imp_new_module = type(sys)
new_module = imp_new_module(module_name)
new_module.__dict__['__name__'] = '__main__'
exec(open(scriptname).read(), new_module.__dict__)

This is similar to what the importlib internals do, but does skip some attributes.

Answer 2

使用运行库。

runpy.run_module(module_name, run_name= "__main__")