PicklableFile implementation

file.py

from StringIO import StringIO

class UnpickledStringIO(StringIO):
    def __setstate__(self, state):
        self.__dict__.update(state)


def unpickle_factory(*a, **kw):
    return UnpickledStringIO(*a, **kw)


class PicklableFile(object):
    '''
    Wrapper around files that allows them to be pickled.
    When unpickled, you'll have an UnpickledStringIO
    instance, which is like a regular StringIO obj
    except it'll have a name property

    Be aware of large files!
    '''
    def __init__(self, file):
        self.file = file

    def __reduce__(self):
        self.file.seek(0)
        data = self.file.read()
        return (unpickle_factory, (data,), {
            'name': self.file.name
        })