mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-12-08 18:32:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Fix issue #4374: Pickle tests fail w/o _pickle extension.

Browse source 
Add an initialization check to mimic the interface of _pickle.
This commit is contained in:
Alexandre Vassalotti 2008-12-27 09:30:39 +00:00
parent 44531cb2db
commit 3cfcab9592
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
Lib/pickle.py
View file

@ -222,6 +222,11 @@ class _Pickler:
def dump(self, obj): def dump(self, obj):
"""Write a pickled representation of obj to the open file.""" """Write a pickled representation of obj to the open file."""
# Check whether Pickler was initialized correctly. This is
# only needed to mimic the behavior of _pickle.Pickler.dump().
if not hasattr(self, "write"):
raise PicklingError("Pickler.__init__() was not called by "
"%s.__init__()" % (self.__class__.__name__,))
if self.proto >= 2: if self.proto >= 2:
self.write(PROTO + bytes([self.proto])) self.write(PROTO + bytes([self.proto]))
self.save(obj) self.save(obj)
@ -789,6 +794,11 @@ class _Unpickler:
Return the reconstituted object hierarchy specified in the file. Return the reconstituted object hierarchy specified in the file.
""" """
# Check whether Unpickler was initialized correctly. This is
# only needed to mimic the behavior of _pickle.Unpickler.dump().
if not hasattr(self, "read"):
raise UnpicklingError("Unpickler.__init__() was not called by "
"%s.__init__()" % (self.__class__.__name__,))
self.mark = object() # any new unique object self.mark = object() # any new unique object
self.stack = [] self.stack = []
self.append = self.stack.append self.append = self.stack.append