如何查看pickle/cloudpickle加载的Lambda及对象的Python源码？

我来帮你搞定这两个关于序列化函数源码查看的问题～

Lambda函数被pickle序列化后，加载出来的对象本身并没有直接存储原始源码，但我们可以通过解析它的字节码来还原出近似的源码：

• 第一步，先查看函数的字节码逻辑：用dis模块反编译字节码，能帮你快速理解函数的执行流程：

  import dis
  new_squared = pickle.loads(...)  # 替换成你的加载代码
  dis.dis(new_squared)
  

• 如果想要更接近原始写法的源码，可以用第三方库uncompyle6来反编译字节码。先安装库：

  pip install uncompyle6
  

  然后执行以下代码还原源码：

  import uncompyle6
  import io
  
  code_obj = new_squared.__code__
  output_buffer = io.StringIO()
  # 反编译字节码到输出流
  uncompyle6.decompile(code_obj.co_argcount, code_obj, output_buffer)
  print(output_buffer.getvalue())
  

注意：反编译出来的源码可能会丢失注释、变量名可能和原始略有不同，但核心逻辑是完全一致的。

cloudpickle比标准pickle支持更多类型的函数序列化，查看源码的思路和pickle类似，但有一些额外的情况需要考虑：

• 如果加载的是原本定义在某个模块里的普通函数，你可以直接用inspect模块获取源码（前提是该模块还能被正常导入）：

  import inspect
  new_squared = pickle.loads(...)  # 替换成你的加载代码
  try:
      print(inspect.getsource(new_squared))
  except TypeError:
      # 若函数是lambda或动态生成的，就用uncompyle6反编译
      import uncompyle6
      import io
      output_buffer = io.StringIO()
      uncompyle6.decompile(new_squared.__code__.co_argcount, new_squared.__code__, output_buffer)
      print(output_buffer.getvalue())
  

• 如果是lambda或者动态生成的函数，同样依赖uncompyle6来反编译字节码，操作和问题1里的方法一致。

另外要提醒的是，如果函数依赖了外部的变量或闭包内容，反编译的源码里不会包含这些依赖的定义，需要结合你序列化时的上下文来理解。

内容的提问来源于stack exchange，提问作者user3458988