FilesExpand file tree

 master

/

codeutil.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

43 lines (33 loc) · 1.58 KB

 master

/

codeutil.py

Copy path

Top

File metadata and controls

• Code
• Blame

43 lines (33 loc) · 1.58 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

# encoding: utf-8

"""Utilities to enable code objects to be pickled.

Any process that import this module will be able to pickle code objects. This

includes the func_code attribute of any function. Once unpickled, new

functions can be built using new.function(code, globals()). Eventually

we need to automate all of this so that functions themselves can be pickled.

Reference: A. Tremols, P Cogolo, "Python Cookbook," p 302-305

"""

__docformat__ = "restructuredtext en"

#-------------------------------------------------------------------------------

# Copyright (C) 2008-2011 The IPython Development Team

#

# Distributed under the terms of the BSD License. The full license is in

# the file COPYING, distributed as part of this software.

#-------------------------------------------------------------------------------

#-------------------------------------------------------------------------------

# Imports

#-------------------------------------------------------------------------------

import sys

import types, copy_reg

def code_ctor(*args):

return types.CodeType(*args)

def reduce_code(co):

if co.co_freevars or co.co_cellvars:

raise ValueError("Sorry, cannot pickle code objects with closures")

args = [co.co_argcount, co.co_nlocals, co.co_stacksize,

co.co_flags, co.co_code, co.co_consts, co.co_names,

co.co_varnames, co.co_filename, co.co_name, co.co_firstlineno,

co.co_lnotab]

if sys.version_info[0] >= 3:

args.insert(1, co.co_kwonlyargcount)

return code_ctor, tuple(args)

copy_reg.pickle(types.CodeType, reduce_code)