Python从内存中使用编译后的模块

在Windows编程的时候，有些时候，我们经常会要使用一些非常规的方法，比如说从内存中加载DLL，然后使用DLL中的函数。于是就思索在用Python的时候是否能够将几个编译好的Pyc合并成一个，然后使用动态的读取这个文件，然后根据标记进行划分，获得不同的模块的pyc内容，然后动态作为一个新的模块引用到我们的程序中去，这就涉及到一个问题，如何将一个pyc整到内存，然后从内存中获取变成一个新模块进行引入处理。之后找了一些资料之后，发现一种方法，就是通过PyCodeObject这个代码对象来进行处理，也就是用compile编译之后的内容，然后读取形成codeobject对象，然后用types.ModuleType建立一个新的模块，然后将这个新模块加入到sys.modules字典中去，之后在这这个新建的模块环境中执行前面读取的codeObject对象，那么久可以使用这个新的导入模块了，方式如下：

#代码如下：

PycContext = open('test.pyo', 'rb').read()

import marshal

#可以查看PyCodeObject数据结构前面8个位是一个4字节MagicNum和4字节的时间戳，所以从第八位开始

PyCodeObject = marshal.loads(b[8:])

import types

#建立一个名字叫testSimple的新模块

newModule = types.ModuleType('testSimple')

import sys

sys.modules['testSimple'] = newModule

#这个时候已经可以用import testSimple了

#但是运行时候会发现，这个新模块什么功能函数都没有，因为还没有和

#上面的PyCodeObject关联起来，此时需要关联

#就是使用exec在本模块环境执行一次，则可

exec c in newModule.__dict__

#这样，上面的codeObject和新模块就关联起来了，然后就可以使用里面的函数了

另外记录一个外国的资料代码如下

def load_compiled_from_memory(name, filename, data, ispackage=False):

    if data[:4]!=imp.get_magic():

        raise ImportError('Bad magic number in %s' % filename)

    # Ignore timestamp in data[4:8]

    code = marshal.loads(data[8:])

    imp.acquire_lock() # Required in threaded applications

    try:

        mod = imp.new_module(name)

        sys.modules[name] = mod # To handle circular and submodule imports

                                # it should come before exec.

        try:

            mod.__file__ = filename # Is not so important.

            # For package you have to set mod.__path__ here.

            # Here I handle simple cases only.

            if ispackage:

                mod.__path__ = [name.replace('.', '/')]

            exec code in mod.__dict__

        except:

            del sys.modules[name]

            raise

    finally:

        imp.release_lock()

    return mod

另外需要说明一下的是，使用Python import自动生成的编译后的文件一般都带有魔数和时间戳，也就是说读取codeObject的时候需要移动8位，但是有些用Python的API生成的不一定带有这个魔数和时间戳的，那么这个时候就不用移位，而直接读取生成codeobject

国外参考资料

巴特西

Python从内存中使用编译后的模块

最新文章

热门文章