IPython/kernel/pickleutil.py

# encoding: utf-8

"""Pickle related utilities."""

__docformat__ = "restructuredtext en"

#-------------------------------------------------------------------------------
#  Copyright (C) 2008  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
#-------------------------------------------------------------------------------

from types import FunctionType

class CannedObject(object):
    pass
    
class CannedFunction(CannedObject):
    
    def __init__(self, f):
        self._checkType(f)    
        self.code = f.func_code
    
    def _checkType(self, obj):
        assert isinstance(obj, FunctionType), "Not a function type"
    
    def getFunction(self, g=None):
        if g is None:
            g = globals()
        newFunc = FunctionType(self.code, g)
        return newFunc

def can(obj):
    if isinstance(obj, FunctionType):
        return CannedFunction(obj)
    else:
        return obj

def canDict(obj):
    if isinstance(obj, dict):
        for k, v in obj.iteritems():
            obj[k] = can(v)
        return obj
    else:
        return obj

def canSequence(obj):
    if isinstance(obj, (list, tuple)):
        t = type(obj)
        return t([can(i) for i in obj])
    else:
        return obj

def uncan(obj, g=None):
    if isinstance(obj, CannedFunction):
        return obj.getFunction(g)
    else:
        return obj

def uncanDict(obj, g=None):
    if isinstance(obj, dict):
        for k, v in obj.iteritems():
            obj[k] = uncan(v,g)
        return obj
    else:
        return obj

def uncanSequence(obj, g=None):
    if isinstance(obj, (list, tuple)):
        t = type(obj)
        return t([uncan(i,g) for i in obj])
    else:
        return obj


def rebindFunctionGlobals(f, glbls):
    return FunctionType(f.func_code, glbls)