comparison mercurial/util.py @ 38575:152f4822d210

pycompat: move rapply() from util

I want to use rapply() in utils.* modules, but that would introduce a
reference cycle util -> utils.* -> util. Moving rapply() to pycompat
should be okay since it mostly serves as a compatibility helper.

38574:f442c9494ec7

    pass

# Python compatibility

_notset = object()

def _rapply(f, xs):
    if xs is None:
        # assume None means non-value of optional data
        return xs
    if isinstance(xs, (list, set, tuple)):
        return type(xs)(_rapply(f, x) for x in xs)
    if isinstance(xs, dict):
        return type(xs)((_rapply(f, k), _rapply(f, v)) for k, v in xs.items())
    return f(xs)

def rapply(f, xs):
    """Apply function recursively to every item preserving the data structure

    >>> def f(x):
    ...     return 'f(%s)' % x
    >>> rapply(f, None) is None
    True
    >>> rapply(f, 'a')
    'f(a)'
    >>> rapply(f, {'a'}) == {'f(a)'}
    True
    >>> rapply(f, ['a', 'b', None, {'c': 'd'}, []])
    ['f(a)', 'f(b)', None, {'f(c)': 'f(d)'}, []]

    >>> xs = [object()]
    >>> rapply(pycompat.identity, xs) is xs
    True
    """
    if f is pycompat.identity:
        # fast path mainly for py2
        return xs
    return _rapply(f, xs)

def bitsfrom(container):
    bits = 0
    for bit in container:
        bits |= bit

38575:152f4822d210

    pass

# Python compatibility

_notset = object()

def bitsfrom(container):
    bits = 0
    for bit in container:
        bits |= bit