You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

839 lines
29 KiB

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