Browse Source

stdpy: Expose DEFAULT_PROTOCOL and HIGHEST_PROTOCOL in direct.stdpy.pickle

rdb 5 years ago
parent
commit
52b4df4aec
1 changed files with 6 additions and 1 deletions
  1. 6 1
      direct/src/stdpy/pickle.py

+ 6 - 1
direct/src/stdpy/pickle.py

@@ -22,7 +22,7 @@ Unfortunately, cPickle cannot be supported, because it does not
 support extensions of this nature. """
 support extensions of this nature. """
 
 
 __all__ = ["PickleError", "PicklingError", "UnpicklingError", "Pickler",
 __all__ = ["PickleError", "PicklingError", "UnpicklingError", "Pickler",
-           "Unpickler", "dump", "dumps", "load", "loads"]
+           "Unpickler", "dump", "dumps", "load", "loads", "HIGHEST_PROTOCOL"]
 
 
 import sys
 import sys
 from panda3d.core import BamWriter, BamReader, TypedObject
 from panda3d.core import BamWriter, BamReader, TypedObject
@@ -36,11 +36,16 @@ else:
 # with the local pickle.py.
 # with the local pickle.py.
 pickle = __import__('pickle')
 pickle = __import__('pickle')
 
 
+HIGHEST_PROTOCOL = pickle.HIGHEST_PROTOCOL
+
 PickleError = pickle.PickleError
 PickleError = pickle.PickleError
 PicklingError = pickle.PicklingError
 PicklingError = pickle.PicklingError
 UnpicklingError = pickle.UnpicklingError
 UnpicklingError = pickle.UnpicklingError
 
 
 if sys.version_info >= (3, 0):
 if sys.version_info >= (3, 0):
+    DEFAULT_PROTOCOL = pickle.DEFAULT_PROTOCOL
+    __all__.append("DEFAULT_PROTOCOL")
+
     BasePickler = pickle._Pickler
     BasePickler = pickle._Pickler
     BaseUnpickler = pickle._Unpickler
     BaseUnpickler = pickle._Unpickler
 else:
 else: