comparison mercurial/util.py @ 35900:72de5c504833

py3: factor out helpers to apply string conversion recursively

35899:d5457d94e1c9

# libraries, and sure enough Mercurial is not a library.)
os.stat_float_times(False)

def safehasattr(thing, attr):
    return getattr(thing, attr, _notset) is not _notset

def bytesinput(fin, fout, *args, **kwargs):
    sin, sout = sys.stdin, sys.stdout
    try:
        sys.stdin, sys.stdout = encoding.strio(fin), encoding.strio(fout)

35900:72de5c504833

# libraries, and sure enough Mercurial is not a library.)
os.stat_float_times(False)

def safehasattr(thing, attr):
    return getattr(thing, attr, _notset) is not _notset

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 bytesinput(fin, fout, *args, **kwargs):
    sin, sout = sys.stdin, sys.stdout
    try:
        sys.stdin, sys.stdout = encoding.strio(fin), encoding.strio(fout)