python模块动态加载

󰃭 2016-09-01

python 提供了 exec 用于在程序中执行一段 python 代码,官方说明:

exec_stmt ::=  "exec" or_expr ["in" expression ["," expression]]

该语句可以使用exec()函数进行替代。

先举个简单的例子

In [1]: exec "print('hello')"
hello

这里就执行了print语句,打印出了hello。然而这个并没有什么实际的效果,在实际编码中,我们更希望能够把加载文件中的方法、变量等取出使用,这时候就用到了exec中的in。

in的作用是将执行代码中的变量、函数或者类放入到一个字典中。

In [11]: m = dict()

In [12]: exec "a=123" in m

In [14]: print m["a"]
123

In [15]: print type(m)
<type 'dict'>

exec "a=123" in m这句话等同于exec("a=123", m)

根据这个思路,构造自己的模块加载方法

def load_conf(module_name, path):
    """
    加载模块
    :param module_name: 加载后的模块名
    :param path: 文件路径
    :return:
    """
    try:
        module = types.ModuleType(module_name)
        exec open(path).read() in module.__dict__
        return module
    except:
        print("Load module [path %s] error: %s" % (path, traceback.format_exc()))
        return None

下面给出测试文件

TEST_LIST = ['111222', '2223333']
TEST_DICT = {'a': 'testaaaaa', 'b': 'testbbbbb'}

def test():
    print 1

测试代码

m = load_conf("test", "/tmp/test.inc")
print dir(m)
print m.__name__
print m.TEST_LIST
print m.TEST_DICT.get('a')
print m.TEST_DICT.get('c', 'default')
m.test()

结果

['TEST_LIST', 'TEST_DICT', '__builtins__', '__doc__', '__name__', 'test']
test
['111222', '2223333']
testaaaaa
default
1