spyro/util.py at master · ebmoon/spyro · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
def find_linenum_starts_with(lines, target, start=0):
    for i, line in enumerate(lines[start:]):
        if 0 == line.find(target):
            return start+i

    return -1

def find_linenum_with(lines, target, start=0):
    for i, line in enumerate(lines[start:]):
        if line.find(target) >= 0:
            return start+i

    return -1

def sum_dict(d1, d2):
    d = d1.copy()
    for k, v in d2.items():
        if k in d.keys():
            d[k] += v
        else:
            d[k] = v

    return d

def max_dict(d1, d2):
    d = d1.copy()
    for k, v in d2.items():
        if k in d.keys():
            d[k] = max(d[k], v)
        else:
            d[k] = v

    return d

def union_dict(d1, d2):
    d = d1.copy()
    for k, v in d2.items():
        if k not in d.keys():
            d[k] = v

    return d