diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-10 16:26:45 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-10 18:37:03 -0800 |
| commit | 4d9f8971e114f38999592ccd149d1a59fcbb3883 (patch) | |
| tree | 0e97645b7e60def235bdb763c5eb2542d6739949 /kafka/protocol/pickle.py | |
| parent | b3bd4cf09baf7d46a3ef21b19d91b9e37679b26c (diff) | |
| download | kafka-python-4d9f8971e114f38999592ccd149d1a59fcbb3883.tar.gz | |
Handle python3 in kafka.protocol.pickle
Diffstat (limited to 'kafka/protocol/pickle.py')
| -rw-r--r-- | kafka/protocol/pickle.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/kafka/protocol/pickle.py b/kafka/protocol/pickle.py index af0d1ee..2265efd 100644 --- a/kafka/protocol/pickle.py +++ b/kafka/protocol/pickle.py @@ -1,6 +1,10 @@ from __future__ import absolute_import -import copy_reg +try: + import copyreg +except ImportError: + import copy_reg as copyreg # python2 + import types @@ -22,4 +26,4 @@ def _unpickle_method(func_name, obj, cls): return func.__get__(obj, cls) # https://bytes.com/topic/python/answers/552476-why-cant-you-pickle-instancemethods -copy_reg.pickle(types.MethodType, _pickle_method, _unpickle_method) +copyreg.pickle(types.MethodType, _pickle_method, _unpickle_method) |