python中load函数_有没有办法在python的class被load的时候触发一个函数

postimport.py

# -*- coding: utf-8 -*-

import importlib

import sys

from collections import defaultdict

_post_import_hooks = defaultdict(list)

class PostImportFinder:

def __init__(self):

self._skip = set()

def find_module(self, fullname, path=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 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

def when_imported(fullname):

def decorate(func):

if fullname in sys.modules:

func(sys.modules[fullname])

else:

_post_import_hooks[fullname].append(func)

return func

return decorate

sys.meta_path.insert(0, PostImportFinder())

调用方：

myTest.py

from postimport import when_imported

@when_imported("PyQt4")

def warn(mod):

print "are you sure you want to import " + str(mod)

import PyQt4

这样当PyQt4模块被导入的时候，会触发被装饰器when_imported修饰的函数warn