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()
|