#!/usr/bin/env python

# Inner-joins two map spreadsheets A->B and B->C to A->C

import csv

import sys

def merge_rows(*rows):

    '''e.g. ['a','b'] + ['','y','z'] = ['a','b; y','z']'''

    def get(row, i):

        try: return row[i]

        except IndexError: return ''

    return ['; '.join(filter(lambda v: v != '', [get(row, i) for row in rows]))

    for i in xrange(max(map(len, rows)))]

def main():

    try: _prog_name, map_1_path = sys.argv

    except ValueError:

        raise SystemExit('Usage: '+sys.argv[0]+' <map_0 map_1 [| '+sys.argv[0]

            +' map_2]... >joined_map')

    # Get map 1

    map_1 = {}

    stream = open(map_1_path, 'rb')

    reader = csv.reader(stream)

    map_1_in, map_1_out = reader.next()[:2]

    for row in reader:

        if row[0] != '': map_1[row[0]] = row

    stream.close()

    # Join map 1 to map 0

    reader = csv.reader(sys.stdin)

    writer = csv.writer(sys.stdout)

    cols = reader.next()

    if not cols[1] == map_1_in: raise SystemExit('Map error: '

        'Map 0 output column name doesn\'t match map 1 input column name')

    cols[1] = map_1_out

    writer.writerow(cols)

    for row in reader:

        if row[1] != '':

            try:

                new = map_1[row[1]]

                row[1] = new[1] # mapping

                row[2:] = merge_rows(row[2:], new[2:]) # comments

            except KeyError:

                row[2] = '** No join mapping for '+row[1]+' ** '+row[2]

                row[1] = ''

        writer.writerow(row)

main()