1 """Utilities for writing code that runs on Python 2 and 3""" 2 3 # Copyright (c) 2010-2015 Benjamin Peterson 4 # 5 # Permission is hereby granted, free of charge, to any person obtaining a copy 6 # of this software and associated documentation files (the "Software"), to deal 7 # in the Software without restriction, including without limitation the rights 8 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 # copies of the Software, and to permit persons to whom the Software is 10 # furnished to do so, subject to the following conditions: 11 # 12 # The above copyright notice and this permission notice shall be included in all 13 # copies or substantial portions of the Software. 14 # 15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 # SOFTWARE. 22 23 from __future__ import absolute_import 24 25 import functools 26 import itertools 27 import operator 28 import sys 29 import types 30 31 __author__ = "Benjamin Peterson <benjamin (at] python.org>" 32 __version__ = "1.9.0" 33 34 35 # Useful for very coarse version differentiation. 36 PY2 = sys.version_info[0] == 2 37 PY3 = sys.version_info[0] == 3 38 39 if PY3: 40 string_types = str, 41 integer_types = int, 42 class_types = type, 43 text_type = str 44 binary_type = bytes 45 46 MAXSIZE = sys.maxsize 47 else: 48 string_types = basestring, 49 integer_types = (int, long) 50 class_types = (type, types.ClassType) 51 text_type = unicode 52 binary_type = str 53 54 if sys.platform.startswith("java"): 55 # Jython always uses 32 bits. 56 MAXSIZE = int((1 << 31) - 1) 57 else: 58 # It's possible to have sizeof(long) != sizeof(Py_ssize_t). 59 class X(object): 60 def __len__(self): 61 return 1 << 31 62 try: 63 len(X()) 64 except OverflowError: 65 # 32-bit 66 MAXSIZE = int((1 << 31) - 1) 67 else: 68 # 64-bit 69 MAXSIZE = int((1 << 63) - 1) 70 del X 71 72 73 def _add_doc(func, doc): 74 """Add documentation to a function.""" 75 func.__doc__ = doc 76 77 78 def _import_module(name): 79 """Import module, returning the module after the last dot.""" 80 __import__(name) 81 return sys.modules[name] 82 83 84 class _LazyDescr(object): 85 86 def __init__(self, name): 87 self.name = name 88 89 def __get__(self, obj, tp): 90 result = self._resolve() 91 setattr(obj, self.name, result) # Invokes __set__. 92 try: 93 # This is a bit ugly, but it avoids running this again by 94 # removing this descriptor. 95 delattr(obj.__class__, self.name) 96 except AttributeError: 97 pass 98 return result 99 100 101 class MovedModule(_LazyDescr): 102 103 def __init__(self, name, old, new=None): 104 super(MovedModule, self).__init__(name) 105 if PY3: 106 if new is None: 107 new = name 108 self.mod = new 109 else: 110 self.mod = old 111 112 def _resolve(self): 113 return _import_module(self.mod) 114 115 def __getattr__(self, attr): 116 _module = self._resolve() 117 value = getattr(_module, attr) 118 setattr(self, attr, value) 119 return value 120 121 122 class _LazyModule(types.ModuleType): 123 124 def __init__(self, name): 125 super(_LazyModule, self).__init__(name) 126 self.__doc__ = self.__class__.__doc__ 127 128 def __dir__(self): 129 attrs = ["__doc__", "__name__"] 130 attrs += [attr.name for attr in self._moved_attributes] 131 return attrs 132 133 # Subclasses should override this 134 _moved_attributes = [] 135 136 137 class MovedAttribute(_LazyDescr): 138 139 def __init__(self, name, old_mod, new_mod, old_attr=None, new_attr=None): 140 super(MovedAttribute, self).__init__(name) 141 if PY3: 142 if new_mod is None: 143 new_mod = name 144 self.mod = new_mod 145 if new_attr is None: 146 if old_attr is None: 147 new_attr = name 148 else: 149 new_attr = old_attr 150 self.attr = new_attr 151 else: 152 self.mod = old_mod 153 if old_attr is None: 154 old_attr = name 155 self.attr = old_attr 156 157 def _resolve(self): 158 module = _import_module(self.mod) 159 return getattr(module, self.attr) 160 161 162 class _SixMetaPathImporter(object): 163 """ 164 A meta path importer to import six.moves and its submodules. 165 166 This class implements a PEP302 finder and loader. It should be compatible 167 with Python 2.5 and all existing versions of Python3 168 """ 169 def __init__(self, six_module_name): 170 self.name = six_module_name 171 self.known_modules = {} 172 173 def _add_module(self, mod, *fullnames): 174 for fullname in fullnames: 175 self.known_modules[self.name + "." + fullname] = mod 176 177 def _get_module(self, fullname): 178 return self.known_modules[self.name + "." + fullname] 179 180 def find_module(self, fullname, path=None): 181 if fullname in self.known_modules: 182 return self 183 return None 184 185 def __get_module(self, fullname): 186 try: 187 return self.known_modules[fullname] 188 except KeyError: 189 raise ImportError("This loader does not know module " + fullname) 190 191 def load_module(self, fullname): 192 try: 193 # in case of a reload 194 return sys.modules[fullname] 195 except KeyError: 196 pass 197 mod = self.__get_module(fullname) 198 if isinstance(mod, MovedModule): 199 mod = mod._resolve() 200 else: 201 mod.__loader__ = self 202 sys.modules[fullname] = mod 203 return mod 204 205 def is_package(self, fullname): 206 """ 207 Return true, if the named module is a package. 208 209 We need this method to get correct spec objects with 210 Python 3.4 (see PEP451) 211 """ 212 return hasattr(self.__get_module(fullname), "__path__") 213 214 def get_code(self, fullname): 215 """Return None 216 217 Required, if is_package is implemented""" 218 self.__get_module(fullname) # eventually raises ImportError 219 return None 220 get_source = get_code # same as get_code 221 222 _importer = _SixMetaPathImporter(__name__) 223 224 225 class _MovedItems(_LazyModule): 226 """Lazy loading of moved objects""" 227 __path__ = [] # mark as package 228 229 230 _moved_attributes = [ 231 MovedAttribute("cStringIO", "cStringIO", "io", "StringIO"), 232 MovedAttribute("filter", "itertools", "builtins", "ifilter", "filter"), 233 MovedAttribute("filterfalse", "itertools", "itertools", "ifilterfalse", "filterfalse"), 234 MovedAttribute("input", "__builtin__", "builtins", "raw_input", "input"), 235 MovedAttribute("intern", "__builtin__", "sys"), 236 MovedAttribute("map", "itertools", "builtins", "imap", "map"), 237 MovedAttribute("range", "__builtin__", "builtins", "xrange", "range"), 238 MovedAttribute("reload_module", "__builtin__", "imp", "reload"), 239 MovedAttribute("reduce", "__builtin__", "functools"), 240 MovedAttribute("shlex_quote", "pipes", "shlex", "quote"), 241 MovedAttribute("StringIO", "StringIO", "io"), 242 MovedAttribute("UserDict", "UserDict", "collections"), 243 MovedAttribute("UserList", "UserList", "collections"), 244 MovedAttribute("UserString", "UserString", "collections"), 245 MovedAttribute("xrange", "__builtin__", "builtins", "xrange", "range"), 246 MovedAttribute("zip", "itertools", "builtins", "izip", "zip"), 247 MovedAttribute("zip_longest", "itertools", "itertools", "izip_longest", "zip_longest"), 248 249 MovedModule("builtins", "__builtin__"), 250 MovedModule("configparser", "ConfigParser"), 251 MovedModule("copyreg", "copy_reg"), 252 MovedModule("dbm_gnu", "gdbm", "dbm.gnu"), 253 MovedModule("_dummy_thread", "dummy_thread", "_dummy_thread"), 254 MovedModule("http_cookiejar", "cookielib", "http.cookiejar"), 255 MovedModule("http_cookies", "Cookie", "http.cookies"), 256 MovedModule("html_entities", "htmlentitydefs", "html.entities"), 257 MovedModule("html_parser", "HTMLParser", "html.parser"), 258 MovedModule("http_client", "httplib", "http.client"), 259 MovedModule("email_mime_multipart", "email.MIMEMultipart", "email.mime.multipart"), 260 MovedModule("email_mime_nonmultipart", "email.MIMENonMultipart", "email.mime.nonmultipart"), 261 MovedModule("email_mime_text", "email.MIMEText", "email.mime.text"), 262 MovedModule("email_mime_base", "email.MIMEBase", "email.mime.base"), 263 MovedModule("BaseHTTPServer", "BaseHTTPServer", "http.server"), 264 MovedModule("CGIHTTPServer", "CGIHTTPServer", "http.server"), 265 MovedModule("SimpleHTTPServer", "SimpleHTTPServer", "http.server"), 266 MovedModule("cPickle", "cPickle", "pickle"), 267 MovedModule("queue", "Queue"), 268 MovedModule("reprlib", "repr"), 269 MovedModule("socketserver", "SocketServer"), 270 MovedModule("_thread", "thread", "_thread"), 271 MovedModule("tkinter", "Tkinter"), 272 MovedModule("tkinter_dialog", "Dialog", "tkinter.dialog"), 273 MovedModule("tkinter_filedialog", "FileDialog", "tkinter.filedialog"), 274 MovedModule("tkinter_scrolledtext", "ScrolledText", "tkinter.scrolledtext"), 275 MovedModule("tkinter_simpledialog", "SimpleDialog", "tkinter.simpledialog"), 276 MovedModule("tkinter_tix", "Tix", "tkinter.tix"), 277 MovedModule("tkinter_ttk", "ttk", "tkinter.ttk"), 278 MovedModule("tkinter_constants", "Tkconstants", "tkinter.constants"), 279 MovedModule("tkinter_dnd", "Tkdnd", "tkinter.dnd"), 280 MovedModule("tkinter_colorchooser", "tkColorChooser", 281 "tkinter.colorchooser"), 282 MovedModule("tkinter_commondialog", "tkCommonDialog", 283 "tkinter.commondialog"), 284 MovedModule("tkinter_tkfiledialog", "tkFileDialog", "tkinter.filedialog"), 285 MovedModule("tkinter_font", "tkFont", "tkinter.font"), 286 MovedModule("tkinter_messagebox", "tkMessageBox", "tkinter.messagebox"), 287 MovedModule("tkinter_tksimpledialog", "tkSimpleDialog", 288 "tkinter.simpledialog"), 289 MovedModule("urllib_parse", __name__ + ".moves.urllib_parse", "urllib.parse"), 290 MovedModule("urllib_error", __name__ + ".moves.urllib_error", "urllib.error"), 291 MovedModule("urllib", __name__ + ".moves.urllib", __name__ + ".moves.urllib"), 292 MovedModule("urllib_robotparser", "robotparser", "urllib.robotparser"), 293 MovedModule("xmlrpc_client", "xmlrpclib", "xmlrpc.client"), 294 MovedModule("xmlrpc_server", "SimpleXMLRPCServer", "xmlrpc.server"), 295 MovedModule("winreg", "_winreg"), 296 ] 297 for attr in _moved_attributes: 298 setattr(_MovedItems, attr.name, attr) 299 if isinstance(attr, MovedModule): 300 _importer._add_module(attr, "moves." + attr.name) 301 del attr 302 303 _MovedItems._moved_attributes = _moved_attributes 304 305 moves = _MovedItems(__name__ + ".moves") 306 _importer._add_module(moves, "moves") 307 308 309 class Module_six_moves_urllib_parse(_LazyModule): 310 """Lazy loading of moved objects in six.moves.urllib_parse""" 311 312 313 _urllib_parse_moved_attributes = [ 314 MovedAttribute("ParseResult", "urlparse", "urllib.parse"), 315 MovedAttribute("SplitResult", "urlparse", "urllib.parse"), 316 MovedAttribute("parse_qs", "urlparse", "urllib.parse"), 317 MovedAttribute("parse_qsl", "urlparse", "urllib.parse"), 318 MovedAttribute("urldefrag", "urlparse", "urllib.parse"), 319 MovedAttribute("urljoin", "urlparse", "urllib.parse"), 320 MovedAttribute("urlparse", "urlparse", "urllib.parse"), 321 MovedAttribute("urlsplit", "urlparse", "urllib.parse"), 322 MovedAttribute("urlunparse", "urlparse", "urllib.parse"), 323 MovedAttribute("urlunsplit", "urlparse", "urllib.parse"), 324 MovedAttribute("quote", "urllib", "urllib.parse"), 325 MovedAttribute("quote_plus", "urllib", "urllib.parse"), 326 MovedAttribute("unquote", "urllib", "urllib.parse"), 327 MovedAttribute("unquote_plus", "urllib", "urllib.parse"), 328 MovedAttribute("urlencode", "urllib", "urllib.parse"), 329 MovedAttribute("splitquery", "urllib", "urllib.parse"), 330 MovedAttribute("splittag", "urllib", "urllib.parse"), 331 MovedAttribute("splituser", "urllib", "urllib.parse"), 332 MovedAttribute("uses_fragment", "urlparse", "urllib.parse"), 333 MovedAttribute("uses_netloc", "urlparse", "urllib.parse"), 334 MovedAttribute("uses_params", "urlparse", "urllib.parse"), 335 MovedAttribute("uses_query", "urlparse", "urllib.parse"), 336 MovedAttribute("uses_relative", "urlparse", "urllib.parse"), 337 ] 338 for attr in _urllib_parse_moved_attributes: 339 setattr(Module_six_moves_urllib_parse, attr.name, attr) 340 del attr 341 342 Module_six_moves_urllib_parse._moved_attributes = _urllib_parse_moved_attributes 343 344 _importer._add_module(Module_six_moves_urllib_parse(__name__ + ".moves.urllib_parse"), 345 "moves.urllib_parse", "moves.urllib.parse") 346 347 348 class Module_six_moves_urllib_error(_LazyModule): 349 """Lazy loading of moved objects in six.moves.urllib_error""" 350 351 352 _urllib_error_moved_attributes = [ 353 MovedAttribute("URLError", "urllib2", "urllib.error"), 354 MovedAttribute("HTTPError", "urllib2", "urllib.error"), 355 MovedAttribute("ContentTooShortError", "urllib", "urllib.error"), 356 ] 357 for attr in _urllib_error_moved_attributes: 358 setattr(Module_six_moves_urllib_error, attr.name, attr) 359 del attr 360 361 Module_six_moves_urllib_error._moved_attributes = _urllib_error_moved_attributes 362 363 _importer._add_module(Module_six_moves_urllib_error(__name__ + ".moves.urllib.error"), 364 "moves.urllib_error", "moves.urllib.error") 365 366 367 class Module_six_moves_urllib_request(_LazyModule): 368 """Lazy loading of moved objects in six.moves.urllib_request""" 369 370 371 _urllib_request_moved_attributes = [ 372 MovedAttribute("urlopen", "urllib2", "urllib.request"), 373 MovedAttribute("install_opener", "urllib2", "urllib.request"), 374 MovedAttribute("build_opener", "urllib2", "urllib.request"), 375 MovedAttribute("pathname2url", "urllib", "urllib.request"), 376 MovedAttribute("url2pathname", "urllib", "urllib.request"), 377 MovedAttribute("getproxies", "urllib", "urllib.request"), 378 MovedAttribute("Request", "urllib2", "urllib.request"), 379 MovedAttribute("OpenerDirector", "urllib2", "urllib.request"), 380 MovedAttribute("HTTPDefaultErrorHandler", "urllib2", "urllib.request"), 381 MovedAttribute("HTTPRedirectHandler", "urllib2", "urllib.request"), 382 MovedAttribute("HTTPCookieProcessor", "urllib2", "urllib.request"), 383 MovedAttribute("ProxyHandler", "urllib2", "urllib.request"), 384 MovedAttribute("BaseHandler", "urllib2", "urllib.request"), 385 MovedAttribute("HTTPPasswordMgr", "urllib2", "urllib.request"), 386 MovedAttribute("HTTPPasswordMgrWithDefaultRealm", "urllib2", "urllib.request"), 387 MovedAttribute("AbstractBasicAuthHandler", "urllib2", "urllib.request"), 388 MovedAttribute("HTTPBasicAuthHandler", "urllib2", "urllib.request"), 389 MovedAttribute("ProxyBasicAuthHandler", "urllib2", "urllib.request"), 390 MovedAttribute("AbstractDigestAuthHandler", "urllib2", "urllib.request"), 391 MovedAttribute("HTTPDigestAuthHandler", "urllib2", "urllib.request"), 392 MovedAttribute("ProxyDigestAuthHandler", "urllib2", "urllib.request"), 393 MovedAttribute("HTTPHandler", "urllib2", "urllib.request"), 394 MovedAttribute("HTTPSHandler", "urllib2", "urllib.request"), 395 MovedAttribute("FileHandler", "urllib2", "urllib.request"), 396 MovedAttribute("FTPHandler", "urllib2", "urllib.request"), 397 MovedAttribute("CacheFTPHandler", "urllib2", "urllib.request"), 398 MovedAttribute("UnknownHandler", "urllib2", "urllib.request"), 399 MovedAttribute("HTTPErrorProcessor", "urllib2", "urllib.request"), 400 MovedAttribute("urlretrieve", "urllib", "urllib.request"), 401 MovedAttribute("urlcleanup", "urllib", "urllib.request"), 402 MovedAttribute("URLopener", "urllib", "urllib.request"), 403 MovedAttribute("FancyURLopener", "urllib", "urllib.request"), 404 MovedAttribute("proxy_bypass", "urllib", "urllib.request"), 405 ] 406 for attr in _urllib_request_moved_attributes: 407 setattr(Module_six_moves_urllib_request, attr.name, attr) 408 del attr 409 410 Module_six_moves_urllib_request._moved_attributes = _urllib_request_moved_attributes 411 412 _importer._add_module(Module_six_moves_urllib_request(__name__ + ".moves.urllib.request"), 413 "moves.urllib_request", "moves.urllib.request") 414 415 416 class Module_six_moves_urllib_response(_LazyModule): 417 """Lazy loading of moved objects in six.moves.urllib_response""" 418 419 420 _urllib_response_moved_attributes = [ 421 MovedAttribute("addbase", "urllib", "urllib.response"), 422 MovedAttribute("addclosehook", "urllib", "urllib.response"), 423 MovedAttribute("addinfo", "urllib", "urllib.response"), 424 MovedAttribute("addinfourl", "urllib", "urllib.response"), 425 ] 426 for attr in _urllib_response_moved_attributes: 427 setattr(Module_six_moves_urllib_response, attr.name, attr) 428 del attr 429 430 Module_six_moves_urllib_response._moved_attributes = _urllib_response_moved_attributes 431 432 _importer._add_module(Module_six_moves_urllib_response(__name__ + ".moves.urllib.response"), 433 "moves.urllib_response", "moves.urllib.response") 434 435 436 class Module_six_moves_urllib_robotparser(_LazyModule): 437 """Lazy loading of moved objects in six.moves.urllib_robotparser""" 438 439 440 _urllib_robotparser_moved_attributes = [ 441 MovedAttribute("RobotFileParser", "robotparser", "urllib.robotparser"), 442 ] 443 for attr in _urllib_robotparser_moved_attributes: 444 setattr(Module_six_moves_urllib_robotparser, attr.name, attr) 445 del attr 446 447 Module_six_moves_urllib_robotparser._moved_attributes = _urllib_robotparser_moved_attributes 448 449 _importer._add_module(Module_six_moves_urllib_robotparser(__name__ + ".moves.urllib.robotparser"), 450 "moves.urllib_robotparser", "moves.urllib.robotparser") 451 452 453 class Module_six_moves_urllib(types.ModuleType): 454 """Create a six.moves.urllib namespace that resembles the Python 3 namespace""" 455 __path__ = [] # mark as package 456 parse = _importer._get_module("moves.urllib_parse") 457 error = _importer._get_module("moves.urllib_error") 458 request = _importer._get_module("moves.urllib_request") 459 response = _importer._get_module("moves.urllib_response") 460 robotparser = _importer._get_module("moves.urllib_robotparser") 461 462 def __dir__(self): 463 return ['parse', 'error', 'request', 'response', 'robotparser'] 464 465 _importer._add_module(Module_six_moves_urllib(__name__ + ".moves.urllib"), 466 "moves.urllib") 467 468 469 def add_move(move): 470 """Add an item to six.moves.""" 471 setattr(_MovedItems, move.name, move) 472 473 474 def remove_move(name): 475 """Remove item from six.moves.""" 476 try: 477 delattr(_MovedItems, name) 478 except AttributeError: 479 try: 480 del moves.__dict__[name] 481 except KeyError: 482 raise AttributeError("no such move, %r" % (name,)) 483 484 485 if PY3: 486 _meth_func = "__func__" 487 _meth_self = "__self__" 488 489 _func_closure = "__closure__" 490 _func_code = "__code__" 491 _func_defaults = "__defaults__" 492 _func_globals = "__globals__" 493 else: 494 _meth_func = "im_func" 495 _meth_self = "im_self" 496 497 _func_closure = "func_closure" 498 _func_code = "func_code" 499 _func_defaults = "func_defaults" 500 _func_globals = "func_globals" 501 502 503 try: 504 advance_iterator = next 505 except NameError: 506 def advance_iterator(it): 507 return it.next() 508 next = advance_iterator 509 510 511 try: 512 callable = callable 513 except NameError: 514 def callable(obj): 515 return any("__call__" in klass.__dict__ for klass in type(obj).__mro__) 516 517 518 if PY3: 519 def get_unbound_function(unbound): 520 return unbound 521 522 create_bound_method = types.MethodType 523 524 Iterator = object 525 else: 526 def get_unbound_function(unbound): 527 return unbound.im_func 528 529 def create_bound_method(func, obj): 530 return types.MethodType(func, obj, obj.__class__) 531 532 class Iterator(object): 533 534 def next(self): 535 return type(self).__next__(self) 536 537 callable = callable 538 _add_doc(get_unbound_function, 539 """Get the function out of a possibly unbound function""") 540 541 542 get_method_function = operator.attrgetter(_meth_func) 543 get_method_self = operator.attrgetter(_meth_self) 544 get_function_closure = operator.attrgetter(_func_closure) 545 get_function_code = operator.attrgetter(_func_code) 546 get_function_defaults = operator.attrgetter(_func_defaults) 547 get_function_globals = operator.attrgetter(_func_globals) 548 549 550 if PY3: 551 def iterkeys(d, **kw): 552 return iter(d.keys(**kw)) 553 554 def itervalues(d, **kw): 555 return iter(d.values(**kw)) 556 557 def iteritems(d, **kw): 558 return iter(d.items(**kw)) 559 560 def iterlists(d, **kw): 561 return iter(d.lists(**kw)) 562 563 viewkeys = operator.methodcaller("keys") 564 565 viewvalues = operator.methodcaller("values") 566 567 viewitems = operator.methodcaller("items") 568 else: 569 def iterkeys(d, **kw): 570 return iter(d.iterkeys(**kw)) 571 572 def itervalues(d, **kw): 573 return iter(d.itervalues(**kw)) 574 575 def iteritems(d, **kw): 576 return iter(d.iteritems(**kw)) 577 578 def iterlists(d, **kw): 579 return iter(d.iterlists(**kw)) 580 581 viewkeys = operator.methodcaller("viewkeys") 582 583 viewvalues = operator.methodcaller("viewvalues") 584 585 viewitems = operator.methodcaller("viewitems") 586 587 _add_doc(iterkeys, "Return an iterator over the keys of a dictionary.") 588 _add_doc(itervalues, "Return an iterator over the values of a dictionary.") 589 _add_doc(iteritems, 590 "Return an iterator over the (key, value) pairs of a dictionary.") 591 _add_doc(iterlists, 592 "Return an iterator over the (key, [values]) pairs of a dictionary.") 593 594 595 if PY3: 596 def b(s): 597 return s.encode("latin-1") 598 def u(s): 599 return s 600 unichr = chr 601 if sys.version_info[1] <= 1: 602 def int2byte(i): 603 return bytes((i,)) 604 else: 605 # This is about 2x faster than the implementation above on 3.2+ 606 int2byte = operator.methodcaller("to_bytes", 1, "big") 607 byte2int = operator.itemgetter(0) 608 indexbytes = operator.getitem 609 iterbytes = iter 610 import io 611 StringIO = io.StringIO 612 BytesIO = io.BytesIO 613 _assertCountEqual = "assertCountEqual" 614 _assertRaisesRegex = "assertRaisesRegex" 615 _assertRegex = "assertRegex" 616 else: 617 def b(s): 618 return s 619 # Workaround for standalone backslash 620 def u(s): 621 return unicode(s.replace(r'\\', r'\\\\'), "unicode_escape") 622 unichr = unichr 623 int2byte = chr 624 def byte2int(bs): 625 return ord(bs[0]) 626 def indexbytes(buf, i): 627 return ord(buf[i]) 628 iterbytes = functools.partial(itertools.imap, ord) 629 import StringIO 630 StringIO = BytesIO = StringIO.StringIO 631 _assertCountEqual = "assertItemsEqual" 632 _assertRaisesRegex = "assertRaisesRegexp" 633 _assertRegex = "assertRegexpMatches" 634 _add_doc(b, """Byte literal""") 635 _add_doc(u, """Text literal""") 636 637 638 def assertCountEqual(self, *args, **kwargs): 639 return getattr(self, _assertCountEqual)(*args, **kwargs) 640 641 642 def assertRaisesRegex(self, *args, **kwargs): 643 return getattr(self, _assertRaisesRegex)(*args, **kwargs) 644 645 646 def assertRegex(self, *args, **kwargs): 647 return getattr(self, _assertRegex)(*args, **kwargs) 648 649 650 if PY3: 651 exec_ = getattr(moves.builtins, "exec") 652 653 654 def reraise(tp, value, tb=None): 655 if value is None: 656 value = tp() 657 if value.__traceback__ is not tb: 658 raise value.with_traceback(tb) 659 raise value 660 661 else: 662 def exec_(_code_, _globs_=None, _locs_=None): 663 """Execute code in a namespace.""" 664 if _globs_ is None: 665 frame = sys._getframe(1) 666 _globs_ = frame.f_globals 667 if _locs_ is None: 668 _locs_ = frame.f_locals 669 del frame 670 elif _locs_ is None: 671 _locs_ = _globs_ 672 exec("""exec _code_ in _globs_, _locs_""") 673 674 675 exec_("""def reraise(tp, value, tb=None): 676 raise tp, value, tb 677 """) 678 679 680 if sys.version_info[:2] == (3, 2): 681 exec_("""def raise_from(value, from_value): 682 if from_value is None: 683 raise value 684 raise value from from_value 685 """) 686 elif sys.version_info[:2] > (3, 2): 687 exec_("""def raise_from(value, from_value): 688 raise value from from_value 689 """) 690 else: 691 def raise_from(value, from_value): 692 raise value 693 694 695 print_ = getattr(moves.builtins, "print", None) 696 if print_ is None: 697 def print_(*args, **kwargs): 698 """The new-style print function for Python 2.4 and 2.5.""" 699 fp = kwargs.pop("file", sys.stdout) 700 if fp is None: 701 return 702 def write(data): 703 if not isinstance(data, basestring): 704 data = str(data) 705 # If the file has an encoding, encode unicode with it. 706 if (isinstance(fp, file) and 707 isinstance(data, unicode) and 708 fp.encoding is not None): 709 errors = getattr(fp, "errors", None) 710 if errors is None: 711 errors = "strict" 712 data = data.encode(fp.encoding, errors) 713 fp.write(data) 714 want_unicode = False 715 sep = kwargs.pop("sep", None) 716 if sep is not None: 717 if isinstance(sep, unicode): 718 want_unicode = True 719 elif not isinstance(sep, str): 720 raise TypeError("sep must be None or a string") 721 end = kwargs.pop("end", None) 722 if end is not None: 723 if isinstance(end, unicode): 724 want_unicode = True 725 elif not isinstance(end, str): 726 raise TypeError("end must be None or a string") 727 if kwargs: 728 raise TypeError("invalid keyword arguments to print()") 729 if not want_unicode: 730 for arg in args: 731 if isinstance(arg, unicode): 732 want_unicode = True 733 break 734 if want_unicode: 735 newline = unicode("\n") 736 space = unicode(" ") 737 else: 738 newline = "\n" 739 space = " " 740 if sep is None: 741 sep = space 742 if end is None: 743 end = newline 744 for i, arg in enumerate(args): 745 if i: 746 write(sep) 747 write(arg) 748 write(end) 749 if sys.version_info[:2] < (3, 3): 750 _print = print_ 751 def print_(*args, **kwargs): 752 fp = kwargs.get("file", sys.stdout) 753 flush = kwargs.pop("flush", False) 754 _print(*args, **kwargs) 755 if flush and fp is not None: 756 fp.flush() 757 758 _add_doc(reraise, """Reraise an exception.""") 759 760 if sys.version_info[0:2] < (3, 4): 761 def wraps(wrapped, assigned=functools.WRAPPER_ASSIGNMENTS, 762 updated=functools.WRAPPER_UPDATES): 763 def wrapper(f): 764 f = functools.wraps(wrapped, assigned, updated)(f) 765 f.__wrapped__ = wrapped 766 return f 767 return wrapper 768 else: 769 wraps = functools.wraps 770 771 def with_metaclass(meta, *bases): 772 """Create a base class with a metaclass.""" 773 # This requires a bit of explanation: the basic idea is to make a dummy 774 # metaclass for one level of class instantiation that replaces itself with 775 # the actual metaclass. 776 class metaclass(meta): 777 def __new__(cls, name, this_bases, d): 778 return meta(name, bases, d) 779 return type.__new__(metaclass, 'temporary_class', (), {}) 780 781 782 def add_metaclass(metaclass): 783 """Class decorator for creating a class with a metaclass.""" 784 def wrapper(cls): 785 orig_vars = cls.__dict__.copy() 786 slots = orig_vars.get('__slots__') 787 if slots is not None: 788 if isinstance(slots, str): 789 slots = [slots] 790 for slots_var in slots: 791 orig_vars.pop(slots_var) 792 orig_vars.pop('__dict__', None) 793 orig_vars.pop('__weakref__', None) 794 return metaclass(cls.__name__, cls.__bases__, orig_vars) 795 return wrapper 796 797 798 def python_2_unicode_compatible(klass): 799 """ 800 A decorator that defines __unicode__ and __str__ methods under Python 2. 801 Under Python 3 it does nothing. 802 803 To support Python 2 and 3 with a single code base, define a __str__ method 804 returning text and apply this decorator to the class. 805 """ 806 if PY2: 807 if '__str__' not in klass.__dict__: 808 raise ValueError("@python_2_unicode_compatible cannot be applied " 809 "to %s because it doesn't define __str__()." % 810 klass.__name__) 811 klass.__unicode__ = klass.__str__ 812 klass.__str__ = lambda self: self.__unicode__().encode('utf-8') 813 return klass 814 815 816 # Complete the moves implementation. 817 # This code is at the end of this module to speed up module loading. 818 # Turn this module into a package. 819 __path__ = [] # required for PEP 302 and PEP 451 820 __package__ = __name__ # see PEP 366 @ReservedAssignment 821 if globals().get("__spec__") is not None: 822 __spec__.submodule_search_locations = [] # PEP 451 @UndefinedVariable 823 # Remove other six meta path importers, since they cause problems. This can 824 # happen if six is removed from sys.modules and then reloaded. (Setuptools does 825 # this for some reason.) 826 if sys.meta_path: 827 for i, importer in enumerate(sys.meta_path): 828 # Here's some real nastiness: Another "instance" of the six module might 829 # be floating around. Therefore, we can't use isinstance() to check for 830 # the six meta path importer, since the other six instance will have 831 # inserted an importer with different class. 832 if (type(importer).__name__ == "_SixMetaPathImporter" and 833 importer.name == __name__): 834 del sys.meta_path[i] 835 break 836 del i, importer 837 # Finally, add the importer to the meta path import hook. 838 sys.meta_path.append(_importer) 839
