Project

General

Profile

1
# Data structures
2

    
3
import lists
4
import util
5

    
6
def rmap(func, value, levels=None):
7
    '''Recursively applies func to all members of value
8
    @param func(value, is_leaf):value
9
    '''
10
    expand = util.coalesce(levels, 1) > 0 and (lists.is_seq(value)
11
        or isinstance(value, dict)) # whether value would be expanded
12
    value = func(value, not expand)
13
    
14
    if expand: # does nothing if func() returned a type that won't be expanded
15
        levels = util.do_ignore_none(lambda v: v-1, levels)
16
        rmap_ = lambda v: rmap(func, v, levels)
17
        if lists.is_seq(value): value = map(rmap_, value)
18
        elif isinstance(value, dict):
19
            value = dict(((k, rmap_(v)) for k, v in value.iteritems()))
20
    
21
    return value
(8-8/52)