1 # This module is used to map the old Python 2 names to the new names used in 2 # Python 3 for the pickle module. This needed to make pickle streams 3 # generated with Python 2 loadable by Python 3. 4 5 # This is a copy of lib2to3.fixes.fix_imports.MAPPING. We cannot import 6 # lib2to3 and use the mapping defined there, because lib2to3 uses pickle. 7 # Thus, this could cause the module to be imported recursively. 8 IMPORT_MAPPING = { 9 '__builtin__' : 'builtins', 10 'copy_reg': 'copyreg', 11 'Queue': 'queue', 12 'SocketServer': 'socketserver', 13 'ConfigParser': 'configparser', 14 'repr': 'reprlib', 15 'tkFileDialog': 'tkinter.filedialog', 16 'tkSimpleDialog': 'tkinter.simpledialog', 17 'tkColorChooser': 'tkinter.colorchooser', 18 'tkCommonDialog': 'tkinter.commondialog', 19 'Dialog': 'tkinter.dialog', 20 'Tkdnd': 'tkinter.dnd', 21 'tkFont': 'tkinter.font', 22 'tkMessageBox': 'tkinter.messagebox', 23 'ScrolledText': 'tkinter.scrolledtext', 24 'Tkconstants': 'tkinter.constants', 25 'Tix': 'tkinter.tix', 26 'ttk': 'tkinter.ttk', 27 'Tkinter': 'tkinter', 28 'markupbase': '_markupbase', 29 '_winreg': 'winreg', 30 'thread': '_thread', 31 'dummy_thread': '_dummy_thread', 32 'dbhash': 'dbm.bsd', 33 'dumbdbm': 'dbm.dumb', 34 'dbm': 'dbm.ndbm', 35 'gdbm': 'dbm.gnu', 36 'xmlrpclib': 'xmlrpc.client', 37 'SimpleXMLRPCServer': 'xmlrpc.server', 38 'httplib': 'http.client', 39 'htmlentitydefs' : 'html.entities', 40 'HTMLParser' : 'html.parser', 41 'Cookie': 'http.cookies', 42 'cookielib': 'http.cookiejar', 43 'BaseHTTPServer': 'http.server', 44 'test.test_support': 'test.support', 45 'commands': 'subprocess', 46 'urlparse' : 'urllib.parse', 47 'robotparser' : 'urllib.robotparser', 48 'urllib2': 'urllib.request', 49 'anydbm': 'dbm', 50 '_abcoll' : 'collections.abc', 51 } 52 53 54 # This contains rename rules that are easy to handle. We ignore the more 55 # complex stuff (e.g. mapping the names in the urllib and types modules). 56 # These rules should be run before import names are fixed. 57 NAME_MAPPING = { 58 ('__builtin__', 'xrange'): ('builtins', 'range'), 59 ('__builtin__', 'reduce'): ('functools', 'reduce'), 60 ('__builtin__', 'intern'): ('sys', 'intern'), 61 ('__builtin__', 'unichr'): ('builtins', 'chr'), 62 ('__builtin__', 'unicode'): ('builtins', 'str'), 63 ('__builtin__', 'long'): ('builtins', 'int'), 64 ('itertools', 'izip'): ('builtins', 'zip'), 65 ('itertools', 'imap'): ('builtins', 'map'), 66 ('itertools', 'ifilter'): ('builtins', 'filter'), 67 ('itertools', 'ifilterfalse'): ('itertools', 'filterfalse'), 68 ('itertools', 'izip_longest'): ('itertools', 'zip_longest'), 69 ('UserDict', 'IterableUserDict'): ('collections', 'UserDict'), 70 ('UserList', 'UserList'): ('collections', 'UserList'), 71 ('UserString', 'UserString'): ('collections', 'UserString'), 72 ('whichdb', 'whichdb'): ('dbm', 'whichdb'), 73 ('_socket', 'fromfd'): ('socket', 'fromfd'), 74 ('_multiprocessing', 'Connection'): ('multiprocessing.connection', 'Connection'), 75 ('multiprocessing.process', 'Process'): ('multiprocessing.context', 'Process'), 76 ('multiprocessing.forking', 'Popen'): ('multiprocessing.popen_fork', 'Popen'), 77 ('urllib', 'ContentTooShortError'): ('urllib.error', 'ContentTooShortError'), 78 ('urllib', 'getproxies'): ('urllib.request', 'getproxies'), 79 ('urllib', 'pathname2url'): ('urllib.request', 'pathname2url'), 80 ('urllib', 'quote_plus'): ('urllib.parse', 'quote_plus'), 81 ('urllib', 'quote'): ('urllib.parse', 'quote'), 82 ('urllib', 'unquote_plus'): ('urllib.parse', 'unquote_plus'), 83 ('urllib', 'unquote'): ('urllib.parse', 'unquote'), 84 ('urllib', 'url2pathname'): ('urllib.request', 'url2pathname'), 85 ('urllib', 'urlcleanup'): ('urllib.request', 'urlcleanup'), 86 ('urllib', 'urlencode'): ('urllib.parse', 'urlencode'), 87 ('urllib', 'urlopen'): ('urllib.request', 'urlopen'), 88 ('urllib', 'urlretrieve'): ('urllib.request', 'urlretrieve'), 89 ('urllib2', 'HTTPError'): ('urllib.error', 'HTTPError'), 90 ('urllib2', 'URLError'): ('urllib.error', 'URLError'), 91 } 92 93 PYTHON2_EXCEPTIONS = ( 94 "ArithmeticError", 95 "AssertionError", 96 "AttributeError", 97 "BaseException", 98 "BufferError", 99 "BytesWarning", 100 "DeprecationWarning", 101 "EOFError", 102 "EnvironmentError", 103 "Exception", 104 "FloatingPointError", 105 "FutureWarning", 106 "GeneratorExit", 107 "IOError", 108 "ImportError", 109 "ImportWarning", 110 "IndentationError", 111 "IndexError", 112 "KeyError", 113 "KeyboardInterrupt", 114 "LookupError", 115 "MemoryError", 116 "NameError", 117 "NotImplementedError", 118 "OSError", 119 "OverflowError", 120 "PendingDeprecationWarning", 121 "ReferenceError", 122 "RuntimeError", 123 "RuntimeWarning", 124 # StandardError is gone in Python 3, so we map it to Exception 125 "StopIteration", 126 "SyntaxError", 127 "SyntaxWarning", 128 "SystemError", 129 "SystemExit", 130 "TabError", 131 "TypeError", 132 "UnboundLocalError", 133 "UnicodeDecodeError", 134 "UnicodeEncodeError", 135 "UnicodeError", 136 "UnicodeTranslateError", 137 "UnicodeWarning", 138 "UserWarning", 139 "ValueError", 140 "Warning", 141 "ZeroDivisionError", 142 ) 143 144 try: 145 WindowsError 146 except NameError: 147 pass 148 else: 149 PYTHON2_EXCEPTIONS += ("WindowsError",) 150 151 for excname in PYTHON2_EXCEPTIONS: 152 NAME_MAPPING[("exceptions", excname)] = ("builtins", excname) 153 154 MULTIPROCESSING_EXCEPTIONS = ( 155 'AuthenticationError', 156 'BufferTooShort', 157 'ProcessError', 158 'TimeoutError', 159 ) 160 161 for excname in MULTIPROCESSING_EXCEPTIONS: 162 NAME_MAPPING[("multiprocessing", excname)] = ("multiprocessing.context", excname) 163 164 # Same, but for 3.x to 2.x 165 REVERSE_IMPORT_MAPPING = dict((v, k) for (k, v) in IMPORT_MAPPING.items()) 166 assert len(REVERSE_IMPORT_MAPPING) == len(IMPORT_MAPPING) 167 REVERSE_NAME_MAPPING = dict((v, k) for (k, v) in NAME_MAPPING.items()) 168 assert len(REVERSE_NAME_MAPPING) == len(NAME_MAPPING) 169 170 # Non-mutual mappings. 171 172 IMPORT_MAPPING.update({ 173 'cPickle': 'pickle', 174 '_elementtree': 'xml.etree.ElementTree', 175 'FileDialog': 'tkinter.filedialog', 176 'SimpleDialog': 'tkinter.simpledialog', 177 'DocXMLRPCServer': 'xmlrpc.server', 178 'SimpleHTTPServer': 'http.server', 179 'CGIHTTPServer': 'http.server', 180 # For compatibility with broken pickles saved in old Python 3 versions 181 'UserDict': 'collections', 182 'UserList': 'collections', 183 'UserString': 'collections', 184 'whichdb': 'dbm', 185 'StringIO': 'io', 186 'cStringIO': 'io', 187 }) 188 189 REVERSE_IMPORT_MAPPING.update({ 190 '_bz2': 'bz2', 191 '_dbm': 'dbm', 192 '_functools': 'functools', 193 '_gdbm': 'gdbm', 194 '_pickle': 'pickle', 195 }) 196 197 NAME_MAPPING.update({ 198 ('__builtin__', 'basestring'): ('builtins', 'str'), 199 ('exceptions', 'StandardError'): ('builtins', 'Exception'), 200 ('UserDict', 'UserDict'): ('collections', 'UserDict'), 201 ('socket', '_socketobject'): ('socket', 'SocketType'), 202 }) 203 204 REVERSE_NAME_MAPPING.update({ 205 ('_functools', 'reduce'): ('__builtin__', 'reduce'), 206 ('tkinter.filedialog', 'FileDialog'): ('FileDialog', 'FileDialog'), 207 ('tkinter.filedialog', 'LoadFileDialog'): ('FileDialog', 'LoadFileDialog'), 208 ('tkinter.filedialog', 'SaveFileDialog'): ('FileDialog', 'SaveFileDialog'), 209 ('tkinter.simpledialog', 'SimpleDialog'): ('SimpleDialog', 'SimpleDialog'), 210 ('xmlrpc.server', 'ServerHTMLDoc'): ('DocXMLRPCServer', 'ServerHTMLDoc'), 211 ('xmlrpc.server', 'XMLRPCDocGenerator'): 212 ('DocXMLRPCServer', 'XMLRPCDocGenerator'), 213 ('xmlrpc.server', 'DocXMLRPCRequestHandler'): 214 ('DocXMLRPCServer', 'DocXMLRPCRequestHandler'), 215 ('xmlrpc.server', 'DocXMLRPCServer'): 216 ('DocXMLRPCServer', 'DocXMLRPCServer'), 217 ('xmlrpc.server', 'DocCGIXMLRPCRequestHandler'): 218 ('DocXMLRPCServer', 'DocCGIXMLRPCRequestHandler'), 219 ('http.server', 'SimpleHTTPRequestHandler'): 220 ('SimpleHTTPServer', 'SimpleHTTPRequestHandler'), 221 ('http.server', 'CGIHTTPRequestHandler'): 222 ('CGIHTTPServer', 'CGIHTTPRequestHandler'), 223 ('_socket', 'socket'): ('socket', '_socketobject'), 224 }) 225 226 PYTHON3_OSERROR_EXCEPTIONS = ( 227 'BrokenPipeError', 228 'ChildProcessError', 229 'ConnectionAbortedError', 230 'ConnectionError', 231 'ConnectionRefusedError', 232 'ConnectionResetError', 233 'FileExistsError', 234 'FileNotFoundError', 235 'InterruptedError', 236 'IsADirectoryError', 237 'NotADirectoryError', 238 'PermissionError', 239 'ProcessLookupError', 240 'TimeoutError', 241 ) 242 243 for excname in PYTHON3_OSERROR_EXCEPTIONS: 244 REVERSE_NAME_MAPPING[('builtins', excname)] = ('exceptions', 'OSError') 245 246 PYTHON3_IMPORTERROR_EXCEPTIONS = ( 247 'ModuleNotFoundError', 248 ) 249 250 for excname in PYTHON3_IMPORTERROR_EXCEPTIONS: 251 REVERSE_NAME_MAPPING[('builtins', excname)] = ('exceptions', 'ImportError') 252
