|
|
- import sys
- import inspect
-
- PY2 = sys.version_info[0] == 2
-
-
- def with_metaclass(meta, *bases):
- # Taken from flask/six.
- class metaclass(meta):
- def __new__(cls, name, this_bases, d):
- return meta(name, bases, d)
- return type.__new__(metaclass, 'temporary_class', (), {})
-
-
- if PY2:
- text_type = unicode
- string_type = basestring
- from itertools import izip_longest as zip_longest
-
- def with_str_method(cls):
- """Class decorator that handles __str__ compat between py2 and py3."""
- # In python2, the __str__ should be __unicode__
- # and __str__ should return bytes.
- cls.__unicode__ = cls.__str__
- def __str__(self):
- return self.__unicode__().encode('utf-8')
- cls.__str__ = __str__
- return cls
-
- def with_repr_method(cls):
- """Class decorator that handle __repr__ with py2 and py3."""
- # This is almost the same thing as with_str_method *except*
- # it uses the unicode_escape encoding. This also means we need to be
- # careful encoding the input multiple times, so we only encode
- # if we get a unicode type.
- original_repr_method = cls.__repr__
- def __repr__(self):
- original_repr = original_repr_method(self)
- if isinstance(original_repr, text_type):
- original_repr = original_repr.encode('unicode_escape')
- return original_repr
- cls.__repr__ = __repr__
- return cls
-
- def get_methods(cls):
- for name, method in inspect.getmembers(cls,
- predicate=inspect.ismethod):
- yield name, method
-
- else:
- text_type = str
- string_type = str
- from itertools import zip_longest
-
- def with_str_method(cls):
- # In python3, we don't need to do anything, we return a str type.
- return cls
-
- def with_repr_method(cls):
- return cls
-
- def get_methods(cls):
- for name, method in inspect.getmembers(cls,
- predicate=inspect.isfunction):
- yield name, method
|
