在 python-3.4+ 中实现导入后挂钩的正确方法

Question

我想修改导入行为。我在 David Beazley 和 Brian K. Jones 的“Python Cookbook”一书中找到了一个带有 post import hook 的示例，它应该适合我的问题。

由于最新版本是在python-3.3 发布的，因此提供的示例已过时。我不得不自己修改它，以便代码在python-3.4之后与importlib兼容。在PostImportFinder 的原始版本中定义了classfind_module(self, fullname, path = None)，并在PostImportLoader 中定义了create_module()，而不是load_module()。 这是一个可重现的小例子：

#postimport.py
import
import importlib
import sys
from collections import defaultdict

_post_import_hooks = defaultdict(list)

class PostImportFinder:
    def __init__(self):
        self._skip=set()
    
    def find_spec(self, fullname, path = None, target = None):
        if fullname in self._skip:
            return None
        self._skip.add(fullname)
        return PostImportLoader(self)


class PostImportLoader:
    def __init__(self, finder):
        self._finder = finder
    
    def create_module(self, spec):
        importlib.import_module(spec.name)
        module = sys.modules[spec.name]
        for func in _post_import_hooks[spec.name]:
            func(module)
        self._finder._skip.remove(spec.name)
        return module
    
def when_imported(names):
    def decorate(func):
        for fullname in names:
            if fullname in sys.modules:
                print(f'importing {fullname}')
                func(sys.modules[fullname])
            else:
                _post_import_hooks[fullname].append(func)
        return func
    return decorate

sys.meta_path.insert(0,PostImportFinder)
 

#postimportfunc.py
from inspect import getmembers, isfunction, isclass
from postimport import when_imported


list_of_module_names = ['simple']
#Some decorator. For example purposes pretty simple
@when_imported(list_of_module_names)
def decorate(mod):
    # Decorate classes
    print(f'module {mod} imported')

#simple.py
class A:
    def __init__(self):
        self.x=42
    
    def bar(self):
        print(self.x)

#start.py
import postimportfunc
from simple import A


foo = A()
foo.bar()

当我运行start.py 时出现以下错误：

    runfile('/home/user/reproduce/start.py', wdir='/home/user/reproduce')

  File "/usr/lib/python3/dist-packages/spyder_kernels/customize/spydercustomize.py", line 827, in runfile
    execfile(filename, namespace)

  File "/usr/lib/python3/dist-packages/spyder_kernels/customize/spydercustomize.py", line 110, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "/home/user/reproduce/start.py", line 2, in <module>
    import postimportfunc

  File "<frozen importlib._bootstrap>", line 991, in _find_and_load

  File "<frozen importlib._bootstrap>", line 971, in _find_and_load_unlocked

  File "<frozen importlib._bootstrap>", line 914, in _find_spec

  File "/home/user/reproduce/postimport.py", line 12, in find_spec
    if fullname in self._skip:

AttributeError: 'str' object has no attribute '_skip' 

所以我的问题很明显：我做错了什么？我根本看不出_skip 和str 是如何相关的，因为我将_skip 初始化为一个集合。 是否有另一种/更好的方法来更改指定模块的导入后行为？

更新：P.S.我忘了提，我目前使用的是 python-3.7 。

Answer 1

主要原因是postimport.py 的最后一行出现了一个愚蠢的错字。 而不是：

sys.meta_path.insert(0,PostImportFinder)

应该是：

sys.meta_path.insert(0,PostImportFinder()) #DO NOT FORGET PARENTHESIS

但这不是唯一的问题。 我还必须更改以下内容：

在PostImportFinder 中将find_spec(self, fullname, path = None, targer = None) 重命名为find_module(self, fullname, path = None) 并在 PostImportLoader 替换 find_spec()with:

def load_module(self, fullname):
        importlib.import_module(fullname)
        module = sys.modules[fullname]
        for func in _post_import_hooks[fullname]:
            func(module)
        self._finder._skip.remove(fullname)
        return module

所以，总结一下：我应该坚持问题中提到的书中的食谱。