xref: /xnu-8792.61.2/tools/lldbmacros/core/compat.py (revision 42e220869062b56f8d7d0726fd4c88954f87902c)

"""
Comaptibility layer to support both Python 2 and Python 3 runtimes.
"""
from __future__ import absolute_import

from future.builtins import int
from future.utils import with_metaclass, PY3

if PY3:
    # Python 3 does not have long. Map it to int.
    long = int


class BaseValueInt(type):
    """ Metaclass for valueint.

        Allows to use valueint in places where long/int is expected.
    """

    def __instancecheck__(cls, instance):
        if cls == valueint:
            # Special case for Py2 short or long int
            return isinstance(instance, (int, long))

        return issubclass(instance.__class__, cls)


# The class below inherits form int on Python 3 (see the long = int above).
# In Python 2 mode it becames children of future's newint.
class valueint(with_metaclass(BaseValueInt, long)):
    """ Python 2/3 compatible integer that works with value class.

        The newint from future mostly works but does not implement all
        operators correctly so it breaks support for value class in Python 2.
    """

    def __floordiv__(self, other):
        """ Fix up // operator.

            newint class tries to construct newint even though the __floordiv__
            has returned NotImplemented. It is required to catch the exception
            and retry with __rfloordiv__.
        """
        try:
            result = super(valueint, self).__floordiv__(other)
        except TypeError:
            return other.__rfloordiv__(self)
        return result

    # The __rfloordiv__ operator has similar problem as __floordiv__ because it
    # does not forward to reverse operator. However it is not causing any extra
    # problems because expressions in form of valueint // value are always
    # handled by value class which converts second argument to int.