openstack/deb-python-kafka
Code Issues Proposed changes

Handle python3 in kafka.protocol.pickle

Browse Source
This commit is contained in:
Dana Powers
2015-12-10 16:26:45 -08:00
parent b3bd4cf09b
commit 4d9f8971e1
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
kafka/protocol/pickle.py
View File

@@ -1,6 +1,10 @@
from __future__ import absolute_import from __future__ import absolute_import
import copy_reg try:
import copyreg
except ImportError:
import copy_reg as copyreg # python2
import types import types
@@ -22,4 +26,4 @@ def _unpickle_method(func_name, obj, cls):
return func.__get__(obj, cls) return func.__get__(obj, cls)
# https://bytes.com/topic/python/answers/552476-why-cant-you-pickle-instancemethods # 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)