six.py 28 KB

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