1 import sys 2 import time 3 4 py3k = sys.version_info >= (3, 0) 5 py33 = sys.version_info >= (3, 3) 6 py2k = sys.version_info < (3,) 7 py26 = sys.version_info >= (2, 6) 8 jython = sys.platform.startswith('java') 9 win32 = sys.platform.startswith('win') 10 pypy = hasattr(sys, 'pypy_version_info') 11 12 if py3k: 13 from io import StringIO 14 import builtins as compat_builtins 15 from urllib.parse import quote_plus, unquote_plus 16 from html.entities import codepoint2name, name2codepoint 17 string_types = str, 18 binary_type = bytes 19 text_type = str 20 21 from io import BytesIO as byte_buffer 22 23 def u(s): 24 return s 25 26 def b(s): 27 return s.encode("latin-1") 28 29 def octal(lit): 30 return eval("0o" + lit) 31 32 else: 33 import __builtin__ as compat_builtins 34 try: 35 from cStringIO import StringIO 36 except: 37 from StringIO import StringIO 38 39 byte_buffer = StringIO 40 41 from urllib import quote_plus, unquote_plus 42 from htmlentitydefs import codepoint2name, name2codepoint 43 string_types = basestring, 44 binary_type = str 45 text_type = unicode 46 47 def u(s): 48 return unicode(s, "utf-8") 49 50 def b(s): 51 return s 52 53 def octal(lit): 54 return eval("0" + lit) 55 56 57 if py33: 58 from importlib import machinery 59 def load_module(module_id, path): 60 return machinery.SourceFileLoader(module_id, path).load_module() 61 else: 62 import imp 63 def load_module(module_id, path): 64 fp = open(path, 'rb') 65 try: 66 return imp.load_source(module_id, path, fp) 67 finally: 68 fp.close() 69 70 71 if py3k: 72 def reraise(tp, value, tb=None, cause=None): 73 if cause is not None: 74 value.__cause__ = cause 75 if value.__traceback__ is not tb: 76 raise value.with_traceback(tb) 77 raise value 78 else: 79 exec("def reraise(tp, value, tb=None, cause=None):\n" 80 " raise tp, value, tb\n") 81 82 83 def exception_as(): 84 return sys.exc_info()[1] 85 86 try: 87 import threading 88 if py3k: 89 import _thread as thread 90 else: 91 import thread 92 except ImportError: 93 import dummy_threading as threading 94 if py3k: 95 import _dummy_thread as thread 96 else: 97 import dummy_thread as thread 98 99 if win32 or jython: 100 time_func = time.clock 101 else: 102 time_func = time.time 103 104 try: 105 from functools import partial 106 except: 107 def partial(func, *args, **keywords): 108 def newfunc(*fargs, **fkeywords): 109 newkeywords = keywords.copy() 110 newkeywords.update(fkeywords) 111 return func(*(args + fargs), **newkeywords) 112 return newfunc 113 114 115 all = all 116 import json 117 118 def exception_name(exc): 119 return exc.__class__.__name__ 120 121 try: 122 from inspect import CO_VARKEYWORDS, CO_VARARGS 123 def inspect_func_args(fn): 124 if py3k: 125 co = fn.__code__ 126 else: 127 co = fn.func_code 128 129 nargs = co.co_argcount 130 names = co.co_varnames 131 args = list(names[:nargs]) 132 133 varargs = None 134 if co.co_flags & CO_VARARGS: 135 varargs = co.co_varnames[nargs] 136 nargs = nargs + 1 137 varkw = None 138 if co.co_flags & CO_VARKEYWORDS: 139 varkw = co.co_varnames[nargs] 140 141 if py3k: 142 return args, varargs, varkw, fn.__defaults__ 143 else: 144 return args, varargs, varkw, fn.func_defaults 145 except ImportError: 146 import inspect 147 def inspect_func_args(fn): 148 return inspect.getargspec(fn) 149 150 if py3k: 151 def callable(fn): 152 return hasattr(fn, '__call__') 153 else: 154 callable = callable 155 156 157 ################################################ 158 # cross-compatible metaclass implementation 159 # Copyright (c) 2010-2012 Benjamin Peterson 160 def with_metaclass(meta, base=object): 161 """Create a base class with a metaclass.""" 162 return meta("%sBase" % meta.__name__, (base,), {}) 163 ################################################ 164 165 166 def arg_stringname(func_arg): 167 """Gets the string name of a kwarg or vararg 168 In Python3.4 a function's args are 169 of _ast.arg type not _ast.name 170 """ 171 if hasattr(func_arg, 'arg'): 172 return func_arg.arg 173 else: 174 return str(func_arg) 175
