python(内置函数, 模块)打补丁, 兼容py2、3

print(unicode, type(unicode))
#  输出
"""
Traceback (most recent call last):
  File "/Users/msw/Desktop/tools/py3_test.py", line 4, in <module>
    print(unicode, type(unicode))
NameError: name 'unicode' is not defined
"""

import sys
if sys.version_info[0] >= 3:
    PY3 = True
else:
    PY3 = False


def patch_default_type():
    if not PY3:
        return
    __builtins__["unicode"] = __builtins__["basestring"] = str
    __builtins__["long"] = int

from py2_to_py3 import patch_default_type
patch_default_type()

print(unicode, type(unicode))

# py3 输出 <class 'str'> <class 'type'>
# py2 输出 (<type 'unicode'>, <type 'type'>)

import sys
if sys.version_info[0] >= 3:
    PY3 = True
else:
    PY3 = False

def patch_modules(copy_reg=False, itertools_imap=False):
    if not PY3:
        return
　　 if copy_reg:
        import copyreg
        sys.modules["copy_reg"] = copyreg
    if itertools_imap:
        import itertools
        setattr(itertools, "imap", map)
 

from py2_to_py3 import patch_modules
patch_modules(copy_reg=True, itertools_imap=True)

import copy_reg
from itertools import imap


print(copy_reg)
print(imap)

# py3 输出
# <module 'copyreg' from '/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/copyreg.py'>
# <class 'map'>

# py2 输出
# <module 'copy_reg' from '/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/copy_reg.pyc'>
# <type 'itertools.imap'>from py2_to_py3 import patch_modules

from six.moves import map, copyreg

print(copyreg)
print(map)

# py3 输出
# <module 'copyreg' from '/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/copyreg.py'>
# <class 'map'>

# py2 输出
# <module 'copy_reg' from '/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/copy_reg.pyc'>
# <type 'itertools.imap'>