Revision 929
Added by Aaron Marcuse-Kubitza almost 13 years ago
bin/subtract | ||
---|---|---|
1 |
#!/usr/bin/env python |
|
2 |
# Subtracts map spreadsheet A1->C from A0->B to A->B |
|
3 |
|
|
4 |
import csv |
|
5 |
import sys |
|
6 |
|
|
7 |
def main(): |
|
8 |
try: _prog_name, map_1_path = sys.argv |
|
9 |
except ValueError: |
|
10 |
raise SystemExit('Usage: '+sys.argv[0]+' <map_0 map_1 [| '+sys.argv[0] |
|
11 |
+' map_2]... >subtraction_map') |
|
12 |
|
|
13 |
# Get map 1 |
|
14 |
map_1 = {} |
|
15 |
stream = open(map_1_path, 'rb') |
|
16 |
reader = csv.reader(stream) |
|
17 |
map_1_cols = reader.next() |
|
18 |
for row in reader: |
|
19 |
if row[0] != '': map_1[row[0]] = row |
|
20 |
stream.close() |
|
21 |
|
|
22 |
# Open map 0 |
|
23 |
reader = csv.reader(sys.stdin) |
|
24 |
map_0_cols = reader.next() |
|
25 |
if map_0_cols[0] != map_1_cols[0]: raise SystemExit('Map error: ' |
|
26 |
'Map 1 column 0 name doesn\'t match map 0 column 0 name') |
|
27 |
|
|
28 |
# Subtract map 1 from map 0 |
|
29 |
writer = csv.writer(sys.stdout) |
|
30 |
writer.writerow(map_0_cols) |
|
31 |
for row in reader: |
|
32 |
if row[0] not in map_1: writer.writerow(row) |
|
33 |
|
|
34 |
main() |
|
0 | 35 |
Also available in: Unified diff
Added subtract to subtract map spreadsheets