Project

General

Profile

1
#!/usr/bin/env python
2
# Inner-joins two map spreadsheets A->B and B->C to A->C
3

    
4
import csv
5
import os.path
6
import sys
7
import warnings
8

    
9
sys.path.append(os.path.dirname(__file__)+"/../lib")
10

    
11
import maps
12
import util
13

    
14
def main():
15
    try: _prog_name, map_1_path = sys.argv
16
    except ValueError:
17
        raise SystemExit('Usage: '+sys.argv[0]+' <map_0 map_1 [| '+sys.argv[0]
18
            +' map_2]... >joined_map')
19
    
20
    # Get map 1
21
    map_1 = {}
22
    stream = open(map_1_path, 'rb')
23
    reader = csv.reader(stream)
24
    map_1_cols = reader.next()
25
    for row in reader:
26
        if row[0] != '': map_1[row[0]] = row
27
    stream.close()
28
    
29
    # Join map 1 to map 0
30
    reader = csv.reader(sys.stdin)
31
    writer = csv.writer(sys.stdout)
32
    map_0_cols = reader.next()
33
    if not maps.join_combinable(map_0_cols, map_1_cols):
34
        raise SystemExit('Map error: Map 0 output column name "'+map_0_cols[1]
35
        +'" doesn\'t match map 1 input column name "'+map_1_cols[0]+'"')
36
    writer.writerow(maps.merge_mappings(map_0_cols, map_1_cols))
37
    for row in reader:
38
        def set_error(msg):
39
            warnings.warn(UserWarning(msg))
40
            row[2] = '** '+msg+' ** '+util.list_setdefault(row, 2, '')
41
            row[1] = ''
42
        
43
        if row[1] != '':
44
            out_orig = row[1] # used in "No join mapping" error msg
45
            
46
            # Look for a match
47
            out_row = None
48
            suffix = ''
49
            while True:
50
                try:
51
                    out_row = map_1[row[1]]
52
                    break
53
                except KeyError:
54
                    # Heuristically look for a match on a parent path.
55
                    # If this produces a syntactically invalid parent path (e.g.
56
                    # there is a / within []), it will be ignored since there
57
                    # wouldn't be a an entry in map_1.
58
                    row[1], sep, new_suffix = row[1].rpartition('/')
59
                    if sep == '': break
60
                    suffix = sep+new_suffix+suffix # prepend new suffix
61
            
62
            # Write new mapping
63
            if out_row != None:
64
                row = maps.merge_mappings(row, out_row)
65
                if row[1] != '': row[1] += suffix # don't modify out_row!
66
            else: set_error('No join mapping for '+out_orig)
67
        elif row[2] == '': # also no comment explaining why no input mapping
68
            set_error('No input mapping for '+row[0])
69
        
70
        writer.writerow(row)
71

    
72
main()
(18-18/41)