从Python AST生成.pyc？

Question

如何从Python AST生成.pyc文件，以便从Python导入该文件？

我已经使用compile创建了一个代码对象，然后将co_code属性写到了一个文件中，但是当我尝试从Python中导入该文件时，我得到了一个ImportError: Bad magic number in output.pyc。

Answer 1

该解决方案可以从py_compile模块改编而来：

import marshal
import py_compile
import time
import ast

codeobject = compile(ast.parse('print "Hello World"'), '<string>', 'exec')

with open('output.pyc', 'wb') as fc:
    fc.write('\0\0\0\0')
    py_compile.wr_long(fc, long(time.time()))
    marshal.dump(codeobject, fc)
    fc.flush()
    fc.seek(0, 0)
    fc.write(py_compile.MAGIC)

​

Answer 2

compile标准函数为Python2.x和Python3.x提供了此函数。但是，您会发现2.x和3.x之间的AST表示有很大的不同，所以要做好准备。