Project

General

Profile

1 187 aaronmk
#!/usr/bin/env python
2 926 aaronmk
# Combines two map spreadsheets A0->B and A1->C to A->B, with B overwriting C
3 187 aaronmk
4
import csv
5 738 aaronmk
import os.path
6 187 aaronmk
import sys
7
8 738 aaronmk
sys.path.append(os.path.dirname(__file__)+"/../lib")
9
10
import maps
11 1123 aaronmk
import opts
12 1769 aaronmk
import util
13 738 aaronmk
14 187 aaronmk
def main():
15 1123 aaronmk
    ignore = opts.env_flag('ignore')
16 1769 aaronmk
    header_num = util.cast(int, opts.get_env_var('header_num'))
17 1127 aaronmk
        # selects which map's header to use as the output header
18 187 aaronmk
    try: _prog_name, map_1_path = sys.argv
19
    except ValueError:
20 1127 aaronmk
        raise SystemExit('Usage: env [ignore=1] [header_num={0|1}] '+sys.argv[0]
21 1123 aaronmk
            +' <map_0 map_1 [| '+sys.argv[0]+' map_2]... >union_map')
22 187 aaronmk
23 1125 aaronmk
    headers = [None]*2
24 187 aaronmk
25 1125 aaronmk
    # Open map 0
26
    map_0_reader = csv.reader(sys.stdin)
27
    headers[0] = map_0_reader.next()
28
29
    # Open map 1
30 187 aaronmk
    stream = open(map_1_path, 'rb')
31 1125 aaronmk
    map_1_reader = csv.reader(stream)
32
    headers[1] = map_1_reader.next()
33 187 aaronmk
34 1125 aaronmk
    # Check col labels
35 1507 aaronmk
    combinable = maps.combinable(*headers)
36
    if not combinable and not ignore:
37 1502 aaronmk
        raise SystemExit('Map error: '
38 1125 aaronmk
        'Map 0 column 0 label doesn\'t contain map 1 column 0 label')
39 187 aaronmk
40 1125 aaronmk
    # Pass through map 0, storing which inputs it defines
41 187 aaronmk
    writer = csv.writer(sys.stdout)
42 1796 aaronmk
    writer.writerow(maps.merge_headers(*headers, **dict(prefer=header_num)))
43 1125 aaronmk
    inputs = set()
44
    for row in map_0_reader:
45
        if row[0] != '': inputs.add(row[0])
46
        writer.writerow(row)
47
48
    if combinable:
49
        # Add entries in map 1 that weren't already defined
50
        for row in map_1_reader:
51
            if row[0] != '' and row[0] not in inputs: writer.writerow(row)
52
53
    stream.close()
54 187 aaronmk
55
main()