#!/usr/bin/env python

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

import csv

import sys

def main():

    prog_name = sys.argv[0]

    try: prog_name, map_1_path = sys.argv

    except ValueError:

        raise Exception('Usage: '+prog_name+' map_1 <map_0 >joined_map')

    csv_config = dict(delimiter=',', quotechar='"')

    # Get map 1

    map_1 = {}

    stream = open(map_1_path, 'rb')

    reader = csv.reader(stream, **csv_config)

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

    for row in reader:

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

    stream.close()

    # Join map 1 to map 0

    reader = csv.reader(sys.stdin, **csv_config)

    map_0_in, map_0_out = reader.next()[:2]

    assert map_0_out == map_1_in

    writer = csv.writer(sys.stdout, **csv_config)

    writer.writerow([map_0_in, map_1_out])

    for row in reader:

        if row[1] in map_1: writer.writerow([row[0], map_1[row[1]]])

main()