Project

General

Profile

1 11 aaronmk
# Useful functions and classes
2
3 1068 aaronmk
import collections
4
5 931 aaronmk
#### Function wrappers for statements
6
7
def noop(*args, **kw_args): pass
8
9
def and_(a, b): return a and b
10
11 791 aaronmk
#### Object metadata
12
13 341 aaronmk
def type_name(value): return type(value).__name__
14
15 135 aaronmk
def module(value): return type(value).__module__.split('.')
16
17
def root_module(value): return module(value)[0]
18
19 1233 aaronmk
#### Type checking and conversion
20 791 aaronmk
21
class ConstraintError(ValueError):
22
    def __init__(self, check_func, value):
23 795 aaronmk
        ValueError.__init__(self, str(value)+' must satisfy constraint '
24
            +check_func.__name__)
25 791 aaronmk
26 836 aaronmk
def cast(type_, val):
27 1046 aaronmk
    '''Passes None through. Does not cast a subclass to a superclass (which
28
    doesn't make sense in a dynamically-typed language).'''
29
    if val != None and not isinstance(val, type_): val = type_(val)
30 836 aaronmk
    return val
31
32 1233 aaronmk
def all_not_none(vals): return reduce(and_, map(lambda v: v != None, vals))
33
34 791 aaronmk
def is_str(val): return isinstance(val, basestring)
35
36 1007 aaronmk
def is_list(val): return isinstance(val, list)
37
38 1233 aaronmk
def bool2str(val):
39
    if val: return '1'
40
    else: return ''
41
42 1046 aaronmk
#### Basic types
43
44 1359 aaronmk
def do_ignore_none(do, val):
45
    if val == None: return val
46
    else: return do(val)
47
48 1483 aaronmk
def coalesce(*vals):
49
    for val in vals:
50
        if val != None: return val
51
    return None
52
53 1373 aaronmk
def none_if(val, *none_vals):
54
    for none_val in none_vals:
55
        if cast(type(none_val), val) == none_val: return None
56
    return val
57 1046 aaronmk
58 1445 aaronmk
#### Classes
59
60
def classes_eq(val0, val1): return val0() == val1() # compare instances
61
62 791 aaronmk
#### Iterables
63
64 135 aaronmk
def first(iter_): return iter_.next()
65
66 133 aaronmk
def skip(iter_, func):
67
    # Advance iter while func is True
68
    try:
69
        while func(iter_.curr()): iter_.next()
70
    except StopIteration: pass # nothing after the matching elements
71
72 1147 aaronmk
class WrapIter:
73
    def __init__(self, wrap_func, iterable):
74
        self.wrap_func = wrap_func
75
        self.iter_ = iterable.__iter__()
76
77
    def __iter__(self): return self
78
79
    def next(self): return self.wrap_func(self.iter_.next())
80 934 aaronmk
81 791 aaronmk
class CheckedIter:
82
    def __init__(self, check_func, iterable):
83
        self.check_func = check_func
84
        self.iter_ = iterable.__iter__()
85
86
    def __iter__(self): return self
87
88
    def next(self):
89
        entry = self.iter_.next()
90
        if self.check_func(entry): return entry
91
        else: raise ConstraintError(self.check_func, entry)
92
93 1008 aaronmk
#### Lists
94
95
def list_get(list_, idx, default=None):
96
    try: return list_[idx]
97
    except IndexError: return default
98
99 1414 aaronmk
def list_set_length(list_, len_, fill=None, expand_only=False):
100 1415 aaronmk
    '''@param list_ Must be a list, not a tuple'''
101 1414 aaronmk
    extra_needed = len_ - len(list_)
102
    if extra_needed > 0: list_ += extra_needed*[fill]
103
    elif not expand_only: del list_[len_:]
104
105 1415 aaronmk
def list_as_length(list_, len_, fill=None):
106
    list_ = list(list_) # will also convert a tuple to a list
107
    list_set_length(list_, len_, fill)
108
    return list_
109
110 1281 aaronmk
def list_set(list_, idx, value, default=None):
111 1414 aaronmk
    list_set_length(list_, idx+1, default, expand_only=True)
112 1281 aaronmk
    list_[idx] = value
113
114
def list_setdefault(list_, idx, default=None):
115
    try: return list_[idx]
116
    except IndexError:
117
        list_set(list_, idx, default, default)
118
        return default
119
120 1147 aaronmk
def list_subset(list_, idxs):
121 1494 aaronmk
    '''@param idxs list|None If None, returns entire list'''
122
    if idxs == None: return list_
123 1147 aaronmk
    subset = []
124
    for idx in idxs:
125
        try: subset.append(list_[idx])
126
        except IndexError: pass
127
    return subset
128
129 1012 aaronmk
def list_eq_is(list0, list1):
130
    '''Compares two lists using is'''
131
    if len(list0) != len(list1): return False
132
    for i in xrange(len(list0)):
133
        if list0[i] is not list1[i]: return False
134
    return True
135
136 1344 aaronmk
def list_replace(list_, repl_elem, with_slice):
137
    list_ = list_[:] # don't modify input!
138
    while True:
139
        try: idx = list_.index(repl_elem)
140
        except ValueError: break
141
        list_[idx:idx+1] = with_slice
142
    return list_
143
144 1145 aaronmk
def list_flip(list_): return dict((v, i) for i, v in enumerate(list_))
145
146 1767 aaronmk
def sort_by_len(list_of_lists): list_of_lists.sort(lambda *a: cmp(*map(len, a)))
147
148
def shortest(*lists):
149
    lists = list(lists)
150
    sort_by_len(lists)
151
    return lists[0]
152
153
def longest(*lists):
154
    lists = list(lists)
155
    sort_by_len(lists)
156
    return lists[-1]
157
158 791 aaronmk
#### Dicts
159
160 330 aaronmk
def rename_key(dict_, orig, new): dict_[new] = dict_.pop(orig)
161 466 aaronmk
162 467 aaronmk
def dict_subset(dict_, keys):
163
    subset = dict()
164
    for key in keys:
165
        try: subset[key] = dict_[key]
166
        except KeyError: pass
167
    return subset
168 1068 aaronmk
169
class DefaultDict(collections.defaultdict):
170
    def __init__(self, dict_, default=None):
171
        collections.defaultdict.__init__(self, lambda: default, dict_)
172
173
def dict_subset_right_join(dict_, keys):
174
    '''Gets a subset of a dict, using None for subset keys that don't exist'''
175
    return dict_subset(DefaultDict(dict_), keys)
176 1147 aaronmk
177
class ListDict:
178 1415 aaronmk
    '''Views a list as a dict, given a key->index mapping
179
    @pre len(list_) == len(keys)'''
180 1147 aaronmk
    def __init__(self, list_, keys, key_idxs=None):
181 1415 aaronmk
        assert len(list_) == len(keys)
182 1147 aaronmk
        if key_idxs == None: key_idxs = list_flip(keys)
183
        self.list = list_
184
        self.keys = keys
185
        self.key_idxs = key_idxs
186
187
    def __getitem__(self, key): return self.list[self.key_idxs[key]]
188
189 1178 aaronmk
    def __str__(self): return '{\n'+(''.join(str(k)+': '+str(v)+'\n'
190
        for k, v in zip(self.keys, self.list)))+'}\n'
191 1443 aaronmk
192
#### Named tuples
193
194
class NamedTuple:
195
    def __init__(self, **attrs): self.__dict__ = attrs