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.
104 lines
3.4 KiB
104 lines
3.4 KiB
6 years ago
|
"""
|
||
|
|
To ensure compatibility from Python ``2.6`` - ``3.3``, a module has been
|
||
|
|
created. Clearly there is huge need to use conforming syntax.
|
||
|
|
"""
|
||
|
|
import sys
|
||
|
|
import platform
|
||
|
|
|
||
|
|
# Cannot use sys.version.major and minor names, because in Python 2.6 it's not
|
||
|
|
# a namedtuple.
|
||
|
|
py_version = int(str(sys.version_info[0]) + str(sys.version_info[1]))
|
||
|
|
|
||
|
|
# unicode function
|
||
|
|
try:
|
||
|
|
unicode = unicode
|
||
|
|
except NameError:
|
||
|
|
unicode = str
|
||
|
|
|
||
|
|
is_pypy = platform.python_implementation() == 'PyPy'
|
||
|
|
|
||
|
|
|
||
|
|
def use_metaclass(meta, *bases):
|
||
|
|
""" Create a class with a metaclass. """
|
||
|
|
if not bases:
|
||
|
|
bases = (object,)
|
||
|
|
return meta("HackClass", bases, {})
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
encoding = sys.stdout.encoding
|
||
|
|
if encoding is None:
|
||
|
|
encoding = 'utf-8'
|
||
|
|
except AttributeError:
|
||
|
|
encoding = 'ascii'
|
||
|
|
|
||
|
|
|
||
|
|
def u(string):
|
||
|
|
"""Cast to unicode DAMMIT!
|
||
|
|
Written because Python2 repr always implicitly casts to a string, so we
|
||
|
|
have to cast back to a unicode (and we know that we always deal with valid
|
||
|
|
unicode, because we check that in the beginning).
|
||
|
|
"""
|
||
|
|
if py_version >= 30:
|
||
|
|
return str(string)
|
||
|
|
|
||
|
|
if not isinstance(string, unicode):
|
||
|
|
return unicode(str(string), 'UTF-8')
|
||
|
|
return string
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
FileNotFoundError = FileNotFoundError
|
||
|
|
except NameError:
|
||
|
|
FileNotFoundError = IOError
|
||
|
|
|
||
|
|
|
||
|
|
def utf8_repr(func):
|
||
|
|
"""
|
||
|
|
``__repr__`` methods in Python 2 don't allow unicode objects to be
|
||
|
|
returned. Therefore cast them to utf-8 bytes in this decorator.
|
||
|
|
"""
|
||
|
|
def wrapper(self):
|
||
|
|
result = func(self)
|
||
|
|
if isinstance(result, unicode):
|
||
|
|
return result.encode('utf-8')
|
||
|
|
else:
|
||
|
|
return result
|
||
|
|
|
||
|
|
if py_version >= 30:
|
||
|
|
return func
|
||
|
|
else:
|
||
|
|
return wrapper
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
from functools import total_ordering
|
||
|
|
except ImportError:
|
||
|
|
# Python 2.6
|
||
|
|
def total_ordering(cls):
|
||
|
|
"""Class decorator that fills in missing ordering methods"""
|
||
|
|
convert = {
|
||
|
|
'__lt__': [('__gt__', lambda self, other: not (self < other or self == other)),
|
||
|
|
('__le__', lambda self, other: self < other or self == other),
|
||
|
|
('__ge__', lambda self, other: not self < other)],
|
||
|
|
'__le__': [('__ge__', lambda self, other: not self <= other or self == other),
|
||
|
|
('__lt__', lambda self, other: self <= other and not self == other),
|
||
|
|
('__gt__', lambda self, other: not self <= other)],
|
||
|
|
'__gt__': [('__lt__', lambda self, other: not (self > other or self == other)),
|
||
|
|
('__ge__', lambda self, other: self > other or self == other),
|
||
|
|
('__le__', lambda self, other: not self > other)],
|
||
|
|
'__ge__': [('__le__', lambda self, other: (not self >= other) or self == other),
|
||
|
|
('__gt__', lambda self, other: self >= other and not self == other),
|
||
|
|
('__lt__', lambda self, other: not self >= other)]
|
||
|
|
}
|
||
|
|
roots = set(dir(cls)) & set(convert)
|
||
|
|
if not roots:
|
||
|
|
raise ValueError('must define at least one ordering operation: < > <= >=')
|
||
|
|
root = max(roots) # prefer __lt__ to __le__ to __gt__ to __ge__
|
||
|
|
for opname, opfunc in convert[root]:
|
||
|
|
if opname not in roots:
|
||
|
|
opfunc.__name__ = opname
|
||
|
|
opfunc.__doc__ = getattr(int, opname).__doc__
|
||
|
|
setattr(cls, opname, opfunc)
|
||
|
|
return cls
|