From 52b4df4aecabb72970e5ce3ec8c8206498c31471 Mon Sep 17 00:00:00 2001
From: rdb <git@rdb.name>
Date: Fri, 1 Jan 2021 17:41:34 +0100
Subject: [PATCH] stdpy: Expose DEFAULT_PROTOCOL and HIGHEST_PROTOCOL in
 direct.stdpy.pickle

---
 direct/src/stdpy/pickle.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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