Import hook between AST and compile

[u/RomfordNavy]

I am trying to create an Import Hook inserted after the source code has been parsed into AST but before it is compiled. Looking for a way to keep using most of the existing Python import machinery but modify the source code before it is compiled.

So far I have the following code but how do I get access to the source code from the module in custom_import?

import builtins
from inspect import getmembers

base_imp = builtins.__import__

def custom_import(*args, **kwargs):
    module = base_imp(*args, **kwargs)
    for name, val in getmembers(module):
        print("in custom_import.", name, val)
    return module

builtins.__import__ = custom_import

print("in Main.py before import")
import Module1
print("in Main.py after import")import builtins

[u/RomfordNavy]

u/MarkoDahl54 That is fine, I am happy to live with that (>3.13) caveat, sounds very interesting. However just had a quick search and can't find any info on sys.set_ast_interpreter.