Revision 59
Added by Aaron Marcuse-Kubitza about 13 years ago
join_maps | ||
---|---|---|
10 | 10 |
except ValueError: |
11 | 11 |
raise Exception('Usage: '+prog_name+' map_1 <map_0 >joined_map') |
12 | 12 |
|
13 |
csv_config = dict(delimiter=',', quotechar='"') |
|
14 |
|
|
15 | 13 |
# Get map 1 |
16 | 14 |
map_1 = {} |
17 | 15 |
stream = open(map_1_path, 'rb') |
18 |
reader = csv.reader(stream, **csv_config)
|
|
16 |
reader = csv.reader(stream) |
|
19 | 17 |
map_1_in, map_1_out = reader.next()[:2] |
20 | 18 |
for row in reader: |
21 | 19 |
if row[1] != '': map_1[row[0]] = row[1] |
22 | 20 |
stream.close() |
23 | 21 |
|
24 | 22 |
# Join map 1 to map 0 |
25 |
reader = csv.reader(sys.stdin, **csv_config)
|
|
23 |
reader = csv.reader(sys.stdin) |
|
26 | 24 |
map_0_in, map_0_out = reader.next()[:2] |
27 | 25 |
assert map_0_out == map_1_in |
28 |
writer = csv.writer(sys.stdout, **csv_config)
|
|
26 |
writer = csv.writer(sys.stdout) |
|
29 | 27 |
writer.writerow([map_0_in, map_1_out]) |
30 | 28 |
for row in reader: |
31 | 29 |
if row[1] in map_1: writer.writerow([row[0], map_1[row[1]]]) |
Also available in: Unified diff
map: Added support for custom fkeys to parent in db XML trees. Removed extraneous csv reader/writer config because Excel format is default. Improved documentation.