#!/usr/bin/env python

# Combines two map spreadsheets A0->B and A1->C to A->B, with B overwriting C

import csv

import os.path

import sys

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

import maps

import opts

def col_label(col_name): return col_name.partition(':')[0]

def main():

    ignore = opts.env_flag('ignore')

    try: _prog_name, map_1_path = sys.argv

    except ValueError:

        raise SystemExit('Usage: env [ignore=1] '+sys.argv[0]

            +' <map_0 map_1 [| '+sys.argv[0]+' map_2]... >union_map')

    map_ = {}

    def add_map(reader):

        for row in reader:

            if row[0] != '':

                try: new = map_[row[0]]

                except KeyError: new = row

                else: new = maps.merge_mappings(new, row)

                map_[row[0]] = new

    # Get map 1

    stream = open(map_1_path, 'rb')

    reader = csv.reader(stream)

    map_1_cols = reader.next()

    add_map(reader)

    stream.close()

    # Add map 0 to map 1, overwriting existing entries

    reader = csv.reader(sys.stdin)

    map_0_cols = reader.next()

    if not col_label(map_0_cols[0]).find(col_label(map_1_cols[0])) >= 0:

        if ignore: sys.exit()

        else: raise SystemExit('Map error: Map 0 column 0 label doesn\'t '

            'contain map 1 column 0 label')

    add_map(reader)

    # Write combined map

    writer = csv.writer(sys.stdout)

    writer.writerow(maps.merge_mappings(map_1_cols, map_0_cols))

    for row in map_.itervalues(): writer.writerow(row)

main()